Add test using digest-method for aes-ctr

Reproduces #3
3stack-software · Feb 21, 2017 · 3ac534f · 3ac534f
1 parent f3d4bfe
commit 3ac534f
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/tests/test_crypto.py b/tests/test_crypto.py
@@ -31,6 +31,16 @@ def test_aes_ctr_legacy():
     )
     assert plaintext == recovered_plaintext
 
+    material = aes_ctr.seal_aes_ctr_legacy(
+        key_service,
+        plaintext,
+        digest_method='SHA512'
+    )
+    recovered_plaintext = aes_ctr.open_aes_ctr_legacy(
+        key_service, material
+    )
+    assert plaintext == recovered_plaintext
+
 
 def test_aes_ctr():
     key_service = DummyKeyService()
@@ -45,6 +55,16 @@ def test_aes_ctr():
     )
     assert plaintext == recovered_plaintext
 
+    material = aes_ctr.seal_aes_ctr(
+        key_service,
+        plaintext,
+        digest_method='SHA512'
+    )
+    recovered_plaintext = aes_ctr.open_aes_ctr(
+        key_service, material
+    )
+    assert plaintext == recovered_plaintext
+
 
 def test_aes_gcm():
     key_service = DummyKeyService()