wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 1 | # coding: utf-8 |
wbond | 284814a | 2015-08-24 09:36:27 -0400 | [diff] [blame] | 2 | from __future__ import unicode_literals, division, absolute_import, print_function |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 3 | |
| 4 | import unittest |
| 5 | import sys |
| 6 | import os |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 7 | |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 8 | from asn1crypto import csr, util |
wbond | 9d65e68 | 2015-08-24 09:38:56 -0400 | [diff] [blame] | 9 | from ._unittest_compat import patch |
| 10 | |
| 11 | patch() |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 12 | |
| 13 | if sys.version_info < (3,): |
| 14 | byte_cls = str |
wbond | a26664f | 2015-10-07 11:57:35 -0400 | [diff] [blame] | 15 | num_cls = long # noqa |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 16 | else: |
| 17 | byte_cls = bytes |
| 18 | num_cls = int |
| 19 | |
| 20 | |
| 21 | tests_root = os.path.dirname(__file__) |
| 22 | fixtures_dir = os.path.join(tests_root, 'fixtures') |
| 23 | |
| 24 | |
| 25 | class CSRTests(unittest.TestCase): |
| 26 | |
| 27 | def test_parse_csr(self): |
| 28 | with open(os.path.join(fixtures_dir, 'test-inter-der.csr'), 'rb') as f: |
| 29 | certification_request = csr.CertificationRequest.load(f.read()) |
| 30 | |
| 31 | cri = certification_request['certification_request_info'] |
| 32 | |
| 33 | self.assertEqual( |
| 34 | 'v1', |
| 35 | cri['version'].native |
| 36 | ) |
| 37 | |
| 38 | self.assertEqual( |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 39 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 40 | ('country_name', 'US'), |
| 41 | ('state_or_province_name', 'Massachusetts'), |
| 42 | ('locality_name', 'Newbury'), |
| 43 | ('organization_name', 'Codex Non Sufficit LC'), |
| 44 | ('organizational_unit_name', 'Testing Intermediate'), |
| 45 | ('common_name', 'Will Bond'), |
| 46 | ('email_address', 'will@codexns.io'), |
| 47 | ]), |
| 48 | cri['subject'].native |
| 49 | ) |
| 50 | self.assertEqual( |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 51 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 52 | ('algorithm', 'rsa'), |
| 53 | ('parameters', None), |
| 54 | ]), |
| 55 | cri['subject_pk_info']['algorithm'].native |
| 56 | ) |
| 57 | self.assertEqual( |
wbond | a26664f | 2015-10-07 11:57:35 -0400 | [diff] [blame] | 58 | 24141757533938720807477509823483015516687050697622322097001928034085434547050399731881871694642845241206788286795830006142635608141713689209738431462004600429798152826994774062467402648660593454536565119527837471261495586474194846971065722669734666949739228862107500673350843489920495869942508240779131331715037662761414997889327943217889802893638175792326783316531272170879284118280173511200768884738639370318760377047837471530387161553030663446359575963736475504659902898072137674205021477968813148345198711103071746476009234601299344030395455052526948041544669303473529511160643491569274897838845918784633403435929, # noqa |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 59 | cri['subject_pk_info']['public_key'].parsed['modulus'].native |
| 60 | ) |
| 61 | self.assertEqual( |
| 62 | 65537, |
| 63 | cri['subject_pk_info']['public_key'].parsed['public_exponent'].native |
| 64 | ) |
| 65 | self.assertEqual( |
| 66 | [], |
| 67 | cri['attributes'].native |
| 68 | ) |
| 69 | |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 70 | def test_parse_csr2(self): |
| 71 | with open(os.path.join(fixtures_dir, 'test-third-der.csr'), 'rb') as f: |
| 72 | certification_request = csr.CertificationRequest.load(f.read()) |
| 73 | |
| 74 | cri = certification_request['certification_request_info'] |
| 75 | |
| 76 | self.assertEqual( |
| 77 | 'v1', |
| 78 | cri['version'].native |
| 79 | ) |
| 80 | |
| 81 | self.assertEqual( |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 82 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 83 | ('country_name', 'US'), |
| 84 | ('state_or_province_name', 'Massachusetts'), |
| 85 | ('locality_name', 'Newbury'), |
| 86 | ('organization_name', 'Codex Non Sufficit LC'), |
| 87 | ('organizational_unit_name', 'Test Third-Level Certificate'), |
| 88 | ('common_name', 'Will Bond'), |
| 89 | ('email_address', 'will@codexns.io'), |
| 90 | ]), |
| 91 | cri['subject'].native |
| 92 | ) |
| 93 | self.assertEqual( |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 94 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 95 | ('algorithm', 'rsa'), |
| 96 | ('parameters', None), |
| 97 | ]), |
| 98 | cri['subject_pk_info']['algorithm'].native |
| 99 | ) |
| 100 | self.assertEqual( |
wbond | a26664f | 2015-10-07 11:57:35 -0400 | [diff] [blame] | 101 | 24242772097421005542208203320016703216069397492249392798445262959177221203301502279838173203064357049006693856302147277901773700963054800321566171864477088538775137040886151390015408166478059887940234405152693144166884492162723776487601158833605063151869850475289834250129252480954724818505034734280077580919995584375189497366089269712298471489896645221362055822887892887126082288043106492130176555423739906252380437817155678204772878611148787130925042126257401487070141904017757131876614711613405231164930930771261221451019736883391322299033324412671768599041417705072563016759224152503535867541947310239343903761461, # noqa |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 102 | cri['subject_pk_info']['public_key'].parsed['modulus'].native |
| 103 | ) |
| 104 | self.assertEqual( |
| 105 | 65537, |
| 106 | cri['subject_pk_info']['public_key'].parsed['public_exponent'].native |
| 107 | ) |
| 108 | self.assertEqual( |
| 109 | [ |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 110 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 111 | ('type', 'extension_request'), |
| 112 | ( |
| 113 | 'values', |
| 114 | [ |
| 115 | [ |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 116 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 117 | ('extn_id', 'basic_constraints'), |
| 118 | ('critical', False), |
| 119 | ( |
| 120 | 'extn_value', |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 121 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 122 | ('ca', False), |
| 123 | ('path_len_constraint', None), |
| 124 | ]) |
| 125 | ), |
| 126 | ]), |
wbond | 44b8919 | 2015-08-24 09:34:01 -0400 | [diff] [blame] | 127 | util.OrderedDict([ |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 128 | ('extn_id', 'key_usage'), |
| 129 | ('critical', False), |
| 130 | ( |
| 131 | 'extn_value', |
wbond | 407e9e3 | 2015-08-24 09:35:28 -0400 | [diff] [blame] | 132 | set(['digital_signature', 'non_repudiation', 'key_encipherment']), |
wbond | 3855a15 | 2015-07-13 10:47:32 -0400 | [diff] [blame] | 133 | ), |
| 134 | ]) |
| 135 | ] |
| 136 | ] |
| 137 | ), |
| 138 | ]), |
| 139 | ], |
| 140 | cri['attributes'].native |
| 141 | ) |