1# 2# This file is part of pyasn1-modules software. 3# 4# Created by Russ Housley 5# Copyright (c) 2019, Vigil Security, LLC 6# License: http://snmplabs.com/pyasn1/license.html 7# 8 9import sys 10 11from pyasn1.codec.der import decoder as der_decoder 12from pyasn1.codec.der import encoder as der_encoder 13 14from pyasn1_modules import pem 15from pyasn1_modules import rfc3565 16 17try: 18 import unittest2 as unittest 19except ImportError: 20 import unittest 21 22 23class AESKeyWrapTestCase(unittest.TestCase): 24 kw_alg_id_pem_text = "MAsGCWCGSAFlAwQBLQ==" 25 26 def setUp(self): 27 self.asn1Spec = rfc3565.AlgorithmIdentifier() 28 29 def testDerCodec(self): 30 substrate = pem.readBase64fromText(self.kw_alg_id_pem_text) 31 asn1Object, rest = der_decoder.decode(substrate, asn1Spec=self.asn1Spec) 32 assert not rest 33 assert asn1Object.prettyPrint() 34 assert asn1Object[0] == rfc3565.id_aes256_wrap 35 assert der_encoder.encode(asn1Object) == substrate 36 37 38class AESCBCTestCase(unittest.TestCase): 39 aes_alg_id_pem_text = "MB0GCWCGSAFlAwQBKgQQEImWuoUOPwM5mTu1h4oONw==" 40 41 def setUp(self): 42 self.asn1Spec = rfc3565.AlgorithmIdentifier() 43 44 def testDerCodec(self): 45 substrate = pem.readBase64fromText(self.aes_alg_id_pem_text) 46 asn1Object, rest = der_decoder.decode(substrate, asn1Spec=self.asn1Spec) 47 assert not rest 48 assert asn1Object.prettyPrint() 49 assert asn1Object[0] == rfc3565.id_aes256_CBC 50 assert asn1Object[1].isValue 51 assert der_encoder.encode(asn1Object) == substrate 52 53 54suite = unittest.TestLoader().loadTestsFromModule(sys.modules[__name__]) 55 56if __name__ == '__main__': 57 import sys 58 59 result = unittest.TextTestRunner(verbosity=2).run(suite) 60 sys.exit(not result.wasSuccessful()) 61