From: godog Date: Tue, 7 Oct 2014 22:49:04 +0000 (+0100) Subject: add ca.verify tests X-Git-Url: https://v.licheni.net/stack/cam.git/commitdiff_plain/HEAD add ca.verify tests --- diff --git a/cam/tests/test_ca.py b/cam/tests/test_ca.py index fdcd238..39a90d1 100644 --- a/cam/tests/test_ca.py +++ b/cam/tests/test_ca.py @@ -29,18 +29,27 @@ class CertStub(object): return os.path.exists(self.public_key_file) +class CAStub(object): + def __init__(self, *args, **kwargs): + self.tmpdir = tempfile.mkdtemp() + self.ca = ca.CA(self.tmpdir, *args, **kwargs) + + def destroy(self): + self.ca.close() + shutil.rmtree(self.tmpdir) + + class CATest(unittest.TestCase): def setUp(self): - self.tmpdir = tempfile.mkdtemp() - self.ca = ca.CA(self.tmpdir, - {'cn': 'test ca', 'org': 'test', + self.ca_stub = CAStub({'cn': 'test ca', 'org': 'test', 'bits': '1024', 'email': 'test@test.com'}, - password='testpw') + password='testpw') + self.ca = self.ca_stub.ca + self.tmpdir = self.ca_stub.tmpdir def tearDown(self): - self.ca.close() - shutil.rmtree(self.tmpdir) + self.ca_stub.destroy() def test_create(self): self.ca.create() @@ -70,3 +79,22 @@ class CATest(unittest.TestCase): self.ca.generate(cert) self.ca.revoke(cert) self.assertTrue(os.path.exists(os.path.join(self.tmpdir, 'public/ca.crl'))) + + def test_verify(self): + orig_cert = CertStub('test', 'www.test.com', self.tmpdir) + self.ca.create() + self.ca.generate(orig_cert) + + alt_ca = CAStub({'cn': 'test ca', 'org': 'test', + 'bits': '1024', 'email': 'test@test.com'}, + password='testpw') + alt_ca.ca.create() + alt_cert = CertStub('test', 'www.test.com', alt_ca.tmpdir) + alt_ca.ca.generate(alt_cert) + + self.assertTrue(self.ca.verify(orig_cert.public_key_file)) + self.assertTrue(alt_ca.ca.verify(alt_cert.public_key_file)) + self.assertFalse(self.ca.verify(alt_cert.public_key_file)) + self.assertFalse(alt_ca.ca.verify(orig_cert.public_key_file)) + + alt_ca.destroy()