Removed duplicate tests, added tests + fix for use after finalize
diff --git a/tests/primitives/test_block.py b/tests/primitives/test_block.py
index 5e342d2..79d65a0 100644
--- a/tests/primitives/test_block.py
+++ b/tests/primitives/test_block.py
@@ -4,22 +4,16 @@
from cryptography.primitives.block import BlockCipher, ciphers, modes, padding
-from ..utils import load_nist_vectors_from_file
-
class TestBlockCipher(object):
- @pytest.mark.parametrize(("key", "iv", "plaintext", "ciphertext"),
- load_nist_vectors_from_file(
- "AES/KAT/CBCGFSbox256.rsp",
- "ENCRYPT",
- ["key", "iv", "plaintext", "ciphertext"]
- )
- )
- def test_aes_cbc_nopadding(self, key, iv, plaintext, ciphertext):
+ def test_use_after_finalize(self):
cipher = BlockCipher(
- ciphers.AES(binascii.unhexlify(key)),
- modes.CBC(binascii.unhexlify(iv), padding.NoPadding())
+ ciphers.AES(binascii.unhexlify(b"0" * 32)),
+ modes.CBC(binascii.unhexlify(b"0" * 32), padding.NoPadding())
)
- actual_ciphertext = cipher.encrypt(binascii.unhexlify(plaintext))
- actual_ciphertext += cipher.finalize()
- assert binascii.hexlify(actual_ciphertext)
+ cipher.encrypt(b"a" * 16)
+ cipher.finalize()
+ with pytest.raises(ValueError):
+ cipher.encrypt(b"b" * 16)
+ with pytest.raises(ValueError):
+ cipher.finalize()