Removed duplicate tests, added tests + fix for use after finalize

1 files changed, 9 insertions, 15 deletions

diff --git a/tests/primitives/test_block.py b/tests/primitives/test_block.py
index 5e342d2f..79d65a0f 100644
--- a/tests/primitives/test_block.py
+++ b/tests/primitives/test_block.py
@@ -4,22 +4,16 @@ import pytest
 
 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()