Add roundtrip encrypt-decrypt test using hypothesis.

Ayrx · Mar 21, 2015 · c7ef488 · c7ef488
1 parent b746eb6
commit c7ef488
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/test_with_hypothesis.py b/tests/test_with_hypothesis.py
@@ -0,0 +1,10 @@
+from hypothesis import given
+
+from aead import AEAD
+
+
+@given(bytes, bytes)
+def test_round_trip_encrypt_decrypt(plaintext, associated_data):
+    cryptor = AEAD(AEAD.generate_key())
+    ct = cryptor.encrypt(plaintext, associated_data)
+    assert plaintext == cryptor.decrypt(ct, associated_data)