commit e69c5fe0649adcfb1c540c39e2b296ad44cde244
author Paul Kehrer <paul.l.kehrer@gmail.com> Wed Dec 30 21:03:26 2015 -0600
committer Paul Kehrer <paul.l.kehrer@gmail.com> Wed Dec 30 21:03:26 2015 -0600
tree 04baa988849f11c1b58ace4371864a1b9b2472ff
parent 7640889dbbc379fe1f164cbd3094b2189aa655ba

don't allow UnrecognizedExtension with get_extension_for_class

tests/test_x509_ext.py

diff --git a/tests/test_x509_ext.py b/tests/test_x509_ext.py
index 7c5ca5f..df796e0 100644
--- a/tests/test_x509_ext.py
+++ b/tests/test_x509_ext.py
@@ -1057,6 +1057,11 @@
             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"),