From b2c94fd22e0da4159be8c98dd32917bdf9cfb504 Mon Sep 17 00:00:00 2001 From: Paul Kehrer Date: Tue, 22 Oct 2013 12:45:07 -0500 Subject: verify that encryptor/decryptor returns CipherContext compliant interface --- tests/primitives/test_block.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) (limited to 'tests/primitives/test_block.py') diff --git a/tests/primitives/test_block.py b/tests/primitives/test_block.py index 8e429085..5e147a79 100644 --- a/tests/primitives/test_block.py +++ b/tests/primitives/test_block.py @@ -17,6 +17,7 @@ import binascii import pytest +from cryptography.primitives import interfaces from cryptography.primitives.block import BlockCipher, ciphers, modes @@ -32,14 +33,14 @@ class TestBlockCipher(object): 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): -- cgit v1.2.3