From a83913fd5b54cfc04c3f2b1e4a4adee5ee601ac1 Mon Sep 17 00:00:00 2001 From: Paul Kehrer Date: Sat, 30 Jan 2016 20:44:00 -0600 Subject: support rsa key generation testing when key is not serializable --- tests/hazmat/primitives/test_rsa.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) (limited to 'tests') diff --git a/tests/hazmat/primitives/test_rsa.py b/tests/hazmat/primitives/test_rsa.py index c432db82..c0b8923f 100644 --- a/tests/hazmat/primitives/test_rsa.py +++ b/tests/hazmat/primitives/test_rsa.py @@ -60,6 +60,15 @@ class DummyHashAlgorithm(object): block_size = 64 +def _check_rsa_private_numbers_if_serializable(key): + if isinstance(key, rsa.RSAPrivateKeyWithSerialization): + _check_rsa_private_numbers(key.private_numbers()) + + +def test_check_rsa_private_numbers_if_serializable(): + _check_rsa_private_numbers_if_serializable("notserializable") + + def _flatten_pkcs1_examples(vectors): flattened_vectors = [] for vector in vectors: @@ -123,7 +132,7 @@ class TestRSA(object): skey = rsa.generate_private_key(public_exponent, key_size, backend) assert skey.key_size == key_size - _check_rsa_private_numbers(skey.private_numbers()) + _check_rsa_private_numbers_if_serializable(skey) pkey = skey.public_key() assert isinstance(pkey.public_numbers(), rsa.RSAPublicNumbers) -- cgit v1.2.3