add some test cases for certmaster stuff

test_certmaster.py:
	just test cases for generating a CA key, and a slave key

1 files changed, 36 insertions, 0 deletions

diff --git a/test/unittest/test_certmaster.py b/test/unittest/test_certmaster.py
new file mode 100644
index 0000000..acc4510
--- /dev/null
+++ b/test/unittest/test_certmaster.py
@@ -0,0 +1,36 @@
+#!/usr/bin/python
+
+
+# unit test for certmaster
+
+import unittest
+
+from certmaster import certs
+
+
+class BaseTest:
+    def __init__(self):
+        pass
+
+
+class TestCa(BaseTest):
+    def __init__(self):
+        self.test_dir = "/tmp/test_certs"
+        
+        self.ca_key_file = "%s/test.key" % self.test_dir
+        self.ca_cert_file = "%s/test.crt" % self.test_dir
+        self.CN = "Test-Cert-CN"
+        
+    def test_create_ca(self):
+        certs.create_ca(self.CN, self.ca_key_file, self.ca_cert_file)
+
+
+    def test_create_slave(self):
+        certs.create_ca(self.CN, self.ca_key_file, self.ca_cert_file)
+        pkey = certs.make_keypair("%s/test.pkey" % self.test_dir)
+        csr = certs.make_csr(pkey, "%s/test.csr" % self.test_dir)
+        certs.create_ca(self.CN, self.ca_key_file, self.ca_cert_file)
+        ca_cert = certs.retrieve_cert_from_file(self.ca_cert_file)
+        ca_key = certs.retrieve_key_from_file(self.ca_key_file)
+        certs.create_slave_certificate(csr, ca_key, ca_cert, self.test_dir, "%s/test_signed.crt" % self.test_dir)
+