38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
|
import base64
|
||
|
|
import unittest
|
||
|
|
|
||
|
|
from sh_edraft.utils import CredentialManager
|
||
|
|
|
||
|
|
|
||
|
|
class CredentialManagerTest(unittest.TestCase):
|
||
|
|
|
||
|
|
def setUp(self): pass
|
||
|
|
|
||
|
|
def test_encode(self):
|
||
|
|
test_string = 'Hello World'
|
||
|
|
expected_test_result = base64.b64encode(test_string.encode('utf-8')).decode('utf-8')
|
||
|
|
|
||
|
|
test_result = CredentialManager.encrypt(test_string)
|
||
|
|
|
||
|
|
self.assertIsNotNone(test_result)
|
||
|
|
self.assertEqual(expected_test_result, test_result)
|
||
|
|
|
||
|
|
def test_decode(self):
|
||
|
|
test_string = 'SGVsbG8gV29ybGQ='
|
||
|
|
expected_test_result = base64.b64decode(test_string).decode('utf-8')
|
||
|
|
|
||
|
|
test_result = CredentialManager.decrypt(test_string)
|
||
|
|
|
||
|
|
self.assertIsNotNone(test_result)
|
||
|
|
self.assertEqual(expected_test_result, test_result)
|
||
|
|
|
||
|
|
def test_build_string(self):
|
||
|
|
test_string = 'String is $credentials'
|
||
|
|
test_credentials = 'SGVsbG8gV29ybGQ='
|
||
|
|
expected_test_result = test_string.replace('$credentials', base64.b64decode(test_credentials).decode('utf-8'))
|
||
|
|
|
||
|
|
test_result = CredentialManager.build_string(test_string, test_credentials)
|
||
|
|
|
||
|
|
self.assertIsNotNone(test_result)
|
||
|
|
self.assertEqual(expected_test_result, test_result)
|