From 85be46d74706a46b6407685610066506d23a49e8 Mon Sep 17 00:00:00 2001 From: Alex Gaynor Date: Mon, 4 Nov 2013 19:23:17 -0800 Subject: Replace assumption with some fakes --- tests/hazmat/bindings/test_openssl.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) (limited to 'tests/hazmat') diff --git a/tests/hazmat/bindings/test_openssl.py b/tests/hazmat/bindings/test_openssl.py index 3523fa4e..4e8dfa58 100644 --- a/tests/hazmat/bindings/test_openssl.py +++ b/tests/hazmat/bindings/test_openssl.py @@ -11,15 +11,21 @@ # See the License for the specific language governing permissions and # limitations under the License. -import binascii - import pytest from cryptography.exceptions import UnsupportedAlgorithm from cryptography.hazmat.bindings.openssl.backend import backend, Backend from cryptography.hazmat.primitives.block import BlockCipher -from cryptography.hazmat.primitives.block.ciphers import AES, TripleDES -from cryptography.hazmat.primitives.block.modes import CBC, ECB +from cryptography.hazmat.primitives.block.ciphers import AES +from cryptography.hazmat.primitives.block.modes import CBC + + +class FakeMode(object): + pass + + +class FakeCipher(object): + pass class TestOpenSSL(object): @@ -51,12 +57,11 @@ class TestOpenSSL(object): def test_nonexistent_cipher(self): b = Backend() - # TODO: this test assumes that 3DES-ECB doesn't exist b.ciphers.register_cipher_adapter( - TripleDES, ECB, lambda backend, cipher, mode: backend.ffi.NULL + FakeCipher, FakeMode, lambda backend, cipher, mode: backend.ffi.NULL ) cipher = BlockCipher( - TripleDES(binascii.unhexlify(b"0" * 16)), ECB(), backend=b + FakeCipher(), FakeMode(), backend=b, ) with pytest.raises(UnsupportedAlgorithm): cipher.encryptor() -- cgit v1.2.3