Refactored code

src_old/tests/utils_test/credential_manager_test.py

@@ -0,0 +1,37 @@
+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)