diff --git a/tests/test_security.py b/tests/test_security.py new file mode 100644 index 000000000..72ca3959d --- /dev/null +++ b/tests/test_security.py @@ -0,0 +1,20 @@ +import unittest +from security import SecurityManager # Assuming you have a SecurityManager class + +class TestSecurityManager(unittest.TestCase): + def setUp(self): + self.security_manager = SecurityManager() + + def test_encryption(self): + original_data = "Sensitive Data" + encrypted_data = self.security_manager.encrypt(original_data) + self.assertNotEqual(original_data, encrypted_data) + + def test_decryption(self): + original_data = "Sensitive Data" + encrypted_data = self.security_manager.encrypt(original_data) + decrypted_data = self.security_manager.decrypt(encrypted_data) + self.assertEqual(original_data, decrypted_data) + +if __name__ == "__main__": + unittest.main()