Create test_crypto.py

KOSASIH · Sep 18, 2024 · ae4d128 · ae4d128
1 parent 4e1c172
commit ae4d128
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/crypto/test/test_crypto.py b/crypto/test/test_crypto.py
@@ -0,0 +1,19 @@
+import unittest
+from crypto_guard import CryptoGuard
+
+class TestCryptoGuard(unittest.TestCase):
+    def test_encrypt(self):
+        # Test the encrypt method
+        crypto_guard = CryptoGuard()
+        encrypted_data = crypto_guard.encrypt("Hello, World!")
+        self.assertNotEqual(encrypted_data, "Hello, World!")
+
+    def test_decrypt(self):
+        # Test the decrypt method
+        crypto_guard = CryptoGuard()
+        encrypted_data = crypto_guard.encrypt("Hello, World!")
+        decrypted_data = crypto_guard.decrypt(encrypted_data)
+        self.assertEqual(decrypted_data, "Hello, World!")
+
+if __name__ == "__main__":
+    unittest.main()