commit a94775925595bf21c849af6eca1a833e51d12e4e
author Alex Gaynor <alex.gaynor@gmail.com> Mon Jun 30 10:03:22 2014 -0700
committer Alex Gaynor <alex.gaynor@gmail.com> Mon Jun 30 10:03:22 2014 -0700
tree 7c46e24eeff71166b2119829fc668cdef6772296
parent cc5224f973de58ddd298d94d8966ccddb7f761a8

Simplify code and add test

tests/hazmat/primitives/test_aes.py

diff --git a/tests/hazmat/primitives/test_aes.py b/tests/hazmat/primitives/test_aes.py
index 003b3ba..5bde7d3 100644
--- a/tests/hazmat/primitives/test_aes.py
+++ b/tests/hazmat/primitives/test_aes.py
@@ -225,6 +225,6 @@
             "gcmEncryptExtIV192.rsp",
             "gcmEncryptExtIV256.rsp",
         ],
-        lambda key: algorithms.AES(key),
-        lambda iv, tag, min_tag_length=16: modes.GCM(iv, tag, min_tag_length),
+        algorithms.AES,
+        modes.GCM,
     )

tests/hazmat/primitives/utils.py

diff --git a/tests/hazmat/primitives/utils.py b/tests/hazmat/primitives/utils.py
index 4640c2e..6428b03 100644
--- a/tests/hazmat/primitives/utils.py
+++ b/tests/hazmat/primitives/utils.py
@@ -311,6 +311,9 @@
     with pytest.raises(ValueError):
         mode_factory(binascii.unhexlify(b"0" * 24), b"000")
 
+    with pytest.raises(ValueError):
+        mode_factory(binascii.unhexlify(b"0" * 24), b"000000", 2)
+
     cipher = Cipher(
         cipher_factory(binascii.unhexlify(b"0" * 32)),
         mode_factory(binascii.unhexlify(b"0" * 24), b"0" * 16),