* Fixed IndexError if a non existent field gets queried for a value. Using dict.get to either get a value or provide a reasonable default * Removed list construction around new_key value * Clean up source code with 80x25 terminal and vim (complies with PEP-8) * Added some docstrings
30 lines
877 B
Python
30 lines
877 B
Python
"""
|
|
This file demonstrates writing tests using the unittest module. These will pass
|
|
when you run "manage.py test".
|
|
|
|
Replace this with more appropriate tests for your application.
|
|
"""
|
|
|
|
from django.test import TestCase
|
|
from account.password_encryption import encrypt_ldap_password, \
|
|
decrypt_ldap_password
|
|
|
|
|
|
class PasswordEncryptionTest(TestCase):
|
|
"""
|
|
Test for the cbmi apps.
|
|
"""
|
|
TEST_LDAP_PASSWD = 'correcthorsebatterystaple'
|
|
|
|
def encrypt_it(self):
|
|
return encrypt_ldap_password(self.TEST_LDAP_PASSWD)
|
|
|
|
def test_encrypt_ldap_password(self):
|
|
message, key = self.encrypt_it()
|
|
print('key:', key)
|
|
print('message:', message)
|
|
|
|
def test_decrypt_ldap_password(self):
|
|
message, key = self.encrypt_it()
|
|
decrypted = decrypt_ldap_password(message, key)
|
|
self.assertEqual(self.TEST_LDAP_PASSWD, decrypted)
|