From e69c5fe0649adcfb1c540c39e2b296ad44cde244 Mon Sep 17 00:00:00 2001 From: Paul Kehrer Date: Wed, 30 Dec 2015 21:03:26 -0600 Subject: don't allow UnrecognizedExtension with get_extension_for_class --- tests/test_x509_ext.py | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'tests') diff --git a/tests/test_x509_ext.py b/tests/test_x509_ext.py index 7c5ca5f2..df796e0a 100644 --- a/tests/test_x509_ext.py +++ b/tests/test_x509_ext.py @@ -1057,6 +1057,11 @@ class TestExtensions(object): exts.get_extension_for_class(x509.IssuerAlternativeName) assert exc.value.oid == ExtensionOID.ISSUER_ALTERNATIVE_NAME + def test_unrecognized_extension_for_class(self): + exts = x509.Extensions([]) + with pytest.raises(TypeError): + exts.get_extension_for_class(x509.UnrecognizedExtension) + def test_indexing(self, backend): cert = _load_cert( os.path.join("x509", "cryptography.io.pem"), -- cgit v1.2.3