# vim:fileencoding=utf8:et:ts=4:sts=4:sw=4:ft=python

from django.conf import settings
from django.test import TestCase

from base64 import b64encode
from Crypto import Random
from mockldap import MockLdap
from passlib.hash import ldap_md5_crypt

from okupy.common.ldap_helpers import (set_secondary_password,
                                       remove_secondary_password)
from okupy.common.test_helpers import set_request, ldap_users
from okupy.crypto.ciphers import cipher
from okupy.tests import vars


class SecondaryPassword(TestCase):
    @classmethod
    def setUpClass(cls):
        cls.mockldap = MockLdap(vars.DIRECTORY)

    @classmethod
    def tearDownClass(cls):
        del cls.mockldap

    def setUp(self):
        self.mockldap.start()
        self.ldapobj = self.mockldap[settings.AUTH_LDAP_SERVER_URI]

    def tearDown(self):
        self.mockldap.stop()
        del self.ldapobj

    def test_secondary_password_gets_added_in_session(self):
        request = set_request(uri='/', user=vars.USER_ALICE)
        set_secondary_password(request, 'ldaptest')
        self.assertEqual(len(request.session['secondary_password']), 48)

    def test_secondary_password_gets_added_in_ldap(self):
        request = set_request(uri='/', user=vars.USER_ALICE)
        self.assertEqual(len(ldap_users('alice')[1]['userPassword']), 1)
        set_secondary_password(request, 'ldaptest')
        self.assertEqual(len(ldap_users(
            'alice',
            directory=self.ldapobj.directory)[1]['userPassword']), 2)

    def test_remove_leftovers_before_adding_secondary_password(self):
        leftover = ldap_md5_crypt.encrypt('leftover_password')
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(leftover)
        request = set_request(uri='/', user=vars.USER_ALICE)
        set_secondary_password(request, 'ldaptest')
        self.assertNotIn(leftover, ldap_users(
            'alice', directory=self.ldapobj.directory)[1]['userPassword'])

    def test_dont_remove_primary_password_while_cleaning_leftovers(self):
        leftover = ldap_md5_crypt.encrypt('leftover_password')
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(leftover)
        request = set_request(uri='/', user=vars.USER_ALICE)
        set_secondary_password(request, 'ldaptest')
        self.assertTrue(ldap_md5_crypt.verify(
            'ldaptest', ldap_users(
                'alice',
                directory=self.ldapobj.directory)[1]['userPassword'][0]))

    def test_dont_remove_unknown_hashes_while_cleaning_leftovers(self):
        leftover = ldap_md5_crypt.encrypt('leftover_password')
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(leftover)
        leftover2 = 'plain_leftover2'
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(leftover2)
        request = set_request(uri='/', user=vars.USER_ALICE)
        set_secondary_password(request, 'ldaptest')
        self.assertIn(leftover2, ldap_users(
            'alice', directory=self.ldapobj.directory)[1]['userPassword'])

    def test_session_and_ldap_secondary_passwords_match(self):
        request = set_request(uri='/', user=vars.USER_ALICE)
        set_secondary_password(request, 'ldaptest')
        self.assertTrue(ldap_md5_crypt.verify(b64encode(cipher.decrypt(
            request.session['secondary_password'], 48)),
            ldap_users(
                'alice',
                directory=self.ldapobj.directory)[1]['userPassword'][1]))

    def test_remove_secondary_password_from_ldap(self):
        secondary_password = Random.get_random_bytes(48)
        secondary_password_crypt = ldap_md5_crypt.encrypt(b64encode(
            secondary_password))
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(secondary_password_crypt)
        request = set_request(uri='/', user=vars.USER_ALICE)
        request.session['secondary_password'] = cipher.encrypt(
            secondary_password)
        remove_secondary_password(request)
        self.assertNotIn(secondary_password_crypt, ldap_users(
            'alice', directory=self.ldapobj.directory)[1]['userPassword'])

    def test_dont_remove_primary_password_when_removing_secondary_passwd(self):
        secondary_password = Random.get_random_bytes(48)
        secondary_password_crypt = ldap_md5_crypt.encrypt(b64encode(
            secondary_password))
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(secondary_password_crypt)
        request = set_request(uri='/', user=vars.USER_ALICE)
        request.session['secondary_password'] = cipher.encrypt(
            secondary_password)
        remove_secondary_password(request)
        self.assertTrue(ldap_md5_crypt.verify('ldaptest', ldap_users(
            'alice',
            directory=self.ldapobj.directory)[1]['userPassword'][0]))

    def test_dont_remove_unknown_hashes_when_removing_secondary_password(self):
        secondary_password = Random.get_random_bytes(48)
        secondary_password_crypt = ldap_md5_crypt.encrypt(b64encode(
            secondary_password))
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append('unknown_hash')
        self.ldapobj.directory[ldap_users('alice')[0]][
            'userPassword'].append(secondary_password_crypt)
        request = set_request(uri='/', user=vars.USER_ALICE)
        request.session['secondary_password'] = cipher.encrypt(
            secondary_password)
        remove_secondary_password(request)
        self.assertIn('unknown_hash', ldap_users(
            'alice', directory=self.ldapobj.directory)[1]['userPassword'])