|
4 | 4 | import copy |
5 | 5 | import unittest |
6 | 6 | import xmlsec |
7 | | -import pkg_resources |
8 | 7 | from xmlsec.test.case import load_test_data |
9 | 8 | from xmlsec import constants, utils |
10 | 9 | import six |
11 | 10 | import base64 |
12 | 11 | from . import find_alts, run_cmd |
13 | 12 | import tempfile |
| 13 | +from importlib_resources import files |
14 | 14 |
|
15 | 15 | __author__ = 'ft' |
16 | 16 |
|
@@ -39,9 +39,9 @@ def _get_all_signatures(t): |
39 | 39 |
|
40 | 40 | class TestSignVerifyXmlSec1(unittest.TestCase): |
41 | 41 | def setUp(self): |
42 | | - self.datadir = pkg_resources.resource_filename(__name__, 'data') |
43 | | - self.private_keyspec = os.path.join(self.datadir, 'test.key') |
44 | | - self.public_keyspec = os.path.join(self.datadir, 'test.pem') |
| 42 | + self.datadir = files(__name__).joinpath('data') |
| 43 | + self.private_keyspec = str(self.datadir.joinpath('test.key')) |
| 44 | + self.public_keyspec = str(self.datadir.joinpath('test.pem')) |
45 | 45 | self.cases = load_test_data('data/verifyxmlsec1') |
46 | 46 | self.tmpf = tempfile.NamedTemporaryFile(delete=False) |
47 | 47 |
|
@@ -80,38 +80,39 @@ def tearDown(self): |
80 | 80 | class TestVerify(unittest.TestCase): |
81 | 81 |
|
82 | 82 | def setUp(self): |
83 | | - self.datadir = pkg_resources.resource_filename(__name__, 'data') |
84 | | - self.resource_dir = pkg_resources.resource_filename(__name__, '') |
| 83 | + self.datadir = files(__name__).joinpath('data') |
| 84 | + self.resource_dir = self.datadir.joinpath('verify') |
85 | 85 | self.cases = load_test_data('data/verify') |
86 | 86 |
|
87 | 87 | def test_verify_all(self): |
88 | 88 | for case in self.cases.values(): |
89 | | - print(str(case)) |
90 | | - public_keyspec = os.path.join(self.resource_dir, case.name, "signer.crt") |
| 89 | + public_keyspec = str(self.resource_dir.joinpath(case.name).joinpath("signer.crt")) |
91 | 90 | res = xmlsec.verify(case.as_etree("in.xml"), public_keyspec) |
92 | 91 | self.assertTrue(res) |
93 | 92 |
|
94 | 93 | def test_verify_one(self): |
95 | 94 | case = self.cases['it-connector'] |
96 | | - public_keyspec = os.path.join(self.resource_dir, case.name, "signer.crt") |
| 95 | + public_keyspec = str(self.resource_dir.joinpath('it-connector').joinpath("signer.crt")) |
97 | 96 | res = xmlsec.verify(case.as_etree("in.xml"), public_keyspec) |
98 | 97 | self.assertTrue(res) |
99 | 98 |
|
100 | 99 | def test_verify_pss_mgf1(self): |
101 | 100 | case = self.cases['eesti'] |
102 | | - public_keyspec = os.path.join(self.resource_dir, case.name, "signer.crt") |
| 101 | + public_keyspec = str(self.resource_dir.joinpath('eesti').joinpath("signer.crt")) |
103 | 102 | res = xmlsec.verify(case.as_etree("in.xml"), public_keyspec) |
104 | 103 | self.assertTrue(res) |
105 | 104 |
|
106 | 105 |
|
107 | 106 | class TestSignVerify(unittest.TestCase): |
108 | 107 |
|
109 | 108 | def setUp(self): |
110 | | - self.datadir = pkg_resources.resource_filename(__name__, 'data') |
111 | | - self.private_keyspec = os.path.join(self.datadir, 'test.key') |
112 | | - self.public_keyspec = os.path.join(self.datadir, 'test.pem') |
| 109 | + self.datadir = files(__name__).joinpath('data') |
| 110 | + self.resource_dir = self.datadir.joinpath('signverify') |
| 111 | + self.private_keyspec = str(self.datadir.joinpath('test.key')) |
| 112 | + self.public_keyspec = str(self.datadir.joinpath('test.pem')) |
113 | 113 | self.cases = load_test_data('data/signverify') |
114 | 114 |
|
| 115 | + |
115 | 116 | def test_sign_verify_SAML_assertion1(self): |
116 | 117 | """ |
117 | 118 | Test signing a SAML assertion, and making sure we can verify it. |
@@ -431,7 +432,7 @@ def test_mm_with_inner_signature(self): |
431 | 432 | print("Signed SignatureValue: %s" % (repr(signed_sv))) |
432 | 433 | print("Expected SignatureValue: %s" % (repr(expected_sv))) |
433 | 434 |
|
434 | | - self.assertEquals(digest, expected_digest) |
| 435 | + self.assertEqual(digest, expected_digest) |
435 | 436 | self.assertEqual(signed_sv, expected_sv) |
436 | 437 |
|
437 | 438 | def test_verify_href(self): |
|
0 commit comments