verify that encryptor/decryptor returns CipherContext compliant interface
diff --git a/tests/primitives/test_block.py b/tests/primitives/test_block.py
index 8e42908..5e147a7 100644
--- a/tests/primitives/test_block.py
+++ b/tests/primitives/test_block.py
@@ -17,6 +17,7 @@
import pytest
+from cryptography.primitives import interfaces
from cryptography.primitives.block import BlockCipher, ciphers, modes
@@ -32,14 +33,14 @@
ciphers.AES(binascii.unhexlify(b"0" * 32)),
modes.CBC(binascii.unhexlify(b"0" * 32))
)
- assert cipher.encryptor() is not None
+ assert isinstance(cipher.encryptor(), interfaces.CipherContext)
def test_creates_decryptor(self):
cipher = BlockCipher(
ciphers.AES(binascii.unhexlify(b"0" * 32)),
modes.CBC(binascii.unhexlify(b"0" * 32))
)
- assert cipher.decryptor() is not None
+ assert isinstance(cipher.decryptor(), interfaces.CipherContext)
class TestBlockCipherContext(object):