1 files changed, 12 insertions, 0 deletions
diff --git a/src/test/model/pki/cert/TbsCertificateTest.java b/src/test/model/pki/cert/TbsCertificateTest.java
index ae92ace..be26eb9 100644
--- a/src/test/model/pki/cert/TbsCertificateTest.java
+++ b/src/test/model/pki/cert/TbsCertificateTest.java
@@ -35,6 +35,18 @@ public class TbsCertificateTest {
     }
 
     @Test
+    void testGetExtension() throws ParseException {
+        TbsCertificate parsed = new TbsCertificate(new BytesReader(trimToTbs(TestConstants.CERT_L1_ECC)),
+                false);
+        assertNull(parsed.getExtension(ObjectIdentifier.OID_EXTENSION_REQUEST));
+        assertNotNull(parsed.getExtension(ObjectIdentifier.OID_BASIC_CONSTRAINTS));
+        parsed = new TbsCertificate(new BytesReader(trimToTbs(TestConstants.CERT_V1)),
+                false);
+        assertNull(parsed.getExtension(ObjectIdentifier.OID_EXTENSION_REQUEST));
+        assertNull(parsed.getExtension(ObjectIdentifier.OID_BASIC_CONSTRAINTS));
+    }
+
+    @Test
     void testParse() throws ParseException {
         TbsCertificate parsed = new TbsCertificate(new BytesReader(trimToTbs(TestConstants.CERT_L1_ECC)),
                 false);