diff options
Diffstat (limited to 'tests/test_rfc7030.py')
-rw-r--r-- | tests/test_rfc7030.py | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/tests/test_rfc7030.py b/tests/test_rfc7030.py new file mode 100644 index 0000000..c351922 --- /dev/null +++ b/tests/test_rfc7030.py @@ -0,0 +1,91 @@ +# +# This file is part of pyasn1-modules software. +# +# Created by Russ Housley +# Copyright (c) 2019, Vigil Security, LLC +# License: http://snmplabs.com/pyasn1/license.html +# + +import sys + +from pyasn1.codec.der.decoder import decode as der_decode +from pyasn1.codec.der.encoder import encode as der_encode + +from pyasn1_modules import pem +from pyasn1_modules import rfc5652 +from pyasn1_modules import rfc7030 + +from pyasn1.type import univ + +try: + import unittest2 as unittest +except ImportError: + import unittest + + +class CSRAttrsTestCase(unittest.TestCase): + pem_text = """\ +MEEGCSqGSIb3DQEJBzASBgcqhkjOPQIBMQcGBSuBBAAiMBYGCSqGSIb3DQEJDjEJ +BgcrBgEBAQEWBggqhkjOPQQDAw== +""" + + the_oids = (univ.ObjectIdentifier('1.2.840.113549.1.9.7'), + univ.ObjectIdentifier('1.2.840.10045.4.3.3'), + ) + + the_attrTypes = (univ.ObjectIdentifier('1.2.840.10045.2.1'), + univ.ObjectIdentifier('1.2.840.113549.1.9.14'), + ) + + the_attrVals = ('1.3.132.0.34', + '1.3.6.1.1.1.1.22', + ) + + + def setUp(self): + self.asn1Spec = rfc7030.CsrAttrs() + + def testDerCodec(self): + substrate = pem.readBase64fromText(self.pem_text) + asn1Object, rest = der_decode(substrate, asn1Spec=self.asn1Spec) + assert not rest + assert asn1Object.prettyPrint() + assert der_encode(asn1Object) == substrate + + for attr_or_oid in asn1Object: + if attr_or_oid.getName() == 'oid': + assert attr_or_oid['oid'] in self.the_oids + + if attr_or_oid.getName() == 'attribute': + assert attr_or_oid['attribute']['attrType'] in self.the_attrTypes + + def testOpenTypes(self): + openTypesMap = { } + openTypesMap.update(rfc5652.cmsAttributesMap) + for at in self.the_attrTypes: + openTypesMap.update({ at: univ.ObjectIdentifier(), }) + + substrate = pem.readBase64fromText(self.pem_text) + asn1Object, rest = der_decode(substrate, + asn1Spec=self.asn1Spec, + openTypes=openTypesMap, + decodeOpenTypes=True) + assert not rest + assert asn1Object.prettyPrint() + assert der_encode(asn1Object) == substrate + + for attr_or_oid in asn1Object: + if attr_or_oid.getName() == 'attribute': + valString = attr_or_oid['attribute']['attrValues'][0].prettyPrint() + + if attr_or_oid['attribute']['attrType'] == self.the_attrTypes[0]: + assert valString == self.the_attrVals[0] + + if attr_or_oid['attribute']['attrType'] == self.the_attrTypes[1]: + assert valString == self.the_attrVals[1] + + +suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__]) + +if __name__ == '__main__': + unittest.TextTestRunner(verbosity=2).run(suite) |