adding unit test

OWASP · Oct 23, 2023 · 2d123f3 · 2d123f3
1 parent 5cc2da6
commit 2d123f3
Show file tree

Hide file tree

Showing 3 changed files with 25 additions and 1 deletion.
diff --git a/tests/data/ssfs_npl_dat b/tests/data/ssfs_npl_dat
diff --git a/tests/data/ssfs_npl_key b/tests/data/ssfs_npl_key
diff --git a/tests/sapssfs_test.py b/tests/sapssfs_test.py
@@ -21,7 +21,7 @@
 # External imports
 # Custom imports
 from tests.utils import data_filename
-from pysap.SAPSSFS import (SAPSSFSKey, SAPSSFSData, SAPSSFSLock)
+from pysap.SAPSSFS import (SAPSSFSKey, SAPSSFSKeyE, SAPSSFSData, SAPSSFSLock)
 
 
 class PySAPSSFSKeyTest(unittest.TestCase):
@@ -145,6 +145,30 @@ def test_ssfs_data_record_decrypt(self):
             self.assertFalse(record.is_stored_as_plaintext)
             self.assertTrue(record.valid)
 
+class PySAPSSFSDataDecryptETest(unittest.TestCase):
+
+    ENCRYPTED_VALUES = {"RS/CIPHERTEXT_VAL": "hellow0rld"}
+
+    def test_ssfs_data_record_decrypt(self):
+        """Test decrypting a record with a given key in a SSFS Data file."""
+
+        with open(data_filename("ssfs_npl_key"), "rb") as fd:
+            s = fd.read()
+        key = SAPSSFSKeyE(s)
+
+        with open(data_filename("ssfs_npl_dat"), "rb") as fd:
+            s = fd.read()
+        data = SAPSSFSData(s)
+
+        for name, value in self.ENCRYPTED_VALUES.items():
+            self.assertTrue(data.has_record(name))
+            self.assertIsNotNone(data.get_record(name))
+            self.assertEqual(data.get_value(name, key), value)
+
+            record = data.get_record(name)
+            self.assertFalse(record.is_stored_as_plaintext)
+            self.assertTrue(record.valid)
+
 
 def test_suite():
     loader = unittest.TestLoader()