Adding in some initial tests

lemur/tests/test_certificates.py

@@ -0,0 +1,85 @@
+import os
+import shutil
+
+import boto
+from moto import mock_iam, mock_sts, mock_s3
+
+from lemur.tests import LemurTestCase
+
+
+#class CertificateTestCase(LemurTestCase):
+#    def test_create_challenge(self):
+#        from lemur.certificates.service import create_challenge
+#        self.assertTrue(len(create_challenge()) >= 24)
+#
+#    def test_hash_domains(self):
+#        from lemur.certificates.service import hash_domains
+#        h = hash_domains(['netflix.com', 'www.netflix.com', 'movies.netflix.com'])
+#        self.assertEqual('c9c83253b46c7c1245c100ed3f7045eb', h)
+#
+#    def test_create_csr(self):
+#        from lemur.certificates.service import create_csr
+#        from lemur.tests.test_csr import TEST_CSR
+#        path = create_csr(['netflix.com'], TEST_CSR)
+#        files = len(os.listdir(path))
+#        self.assertEqual(files, 4)
+#        shutil.rmtree(path)
+#
+#    def test_create_san_csr(self):
+#        from lemur.certificates.service import create_csr
+#        from lemur.tests.test_csr import TEST_CSR
+#        path = create_csr(['netflix.com', 'www.netflix.com'], TEST_CSR)
+#        files = len(os.listdir(path))
+#        self.assertEqual(files, 4)
+#        shutil.rmtree(path)
+#
+#    def test_create_path(self):
+#        from lemur.certificates.service import create_path
+#        path = create_path("blah")
+#        self.assertIn('blah', path)
+#        shutil.rmtree(path)
+#
+#    @mock_s3
+#    @mock_sts
+#    @mock_iam
+#    def test_save_cert(self):
+#        from lemur.certificates.service import save_cert
+#        from lemur.common.services.aws.iam import get_all_server_certs
+#        conn = boto.connect_s3()
+#        bucket = conn.create_bucket(app.config.get('S3_BUCKET'))
+#        cert = save_cert(TEST_CERT, TEST_KEY, None, "blah", "blah", [1])
+#        count = 0
+#        for key in bucket.list():
+#            count += 1
+#
+#        self.assertEqual(count, 4)
+#        certs = get_all_server_certs('1111')
+#        self.assertEqual(len(certs), 1)
+#
+##    @mock_s3
+##    @mock_sts
+##    @mock_iam
+##    def test_upload_cert(self):
+##        from lemur.certificates.service import upload
+##        from lemur.common.services.aws.iam import get_all_server_certs
+##        conn = boto.connect_s3()
+##        bucket = conn.create_bucket(app.config.get('S3_BUCKET'))
+##
+##        cert_up = {"public_cert": TEST_CERT, "private_key": TEST_KEY, "owner": "test@example.com", "accounts_ids": ['1111']}
+##
+##        cert_name = upload(**cert_up)
+##        valid_name = 'AHB-dfdsflkj.net-NetflixInc-20140525-20150525'
+##        self.assertEqual(cert_name, valid_name)
+##
+##        app.logger.debug(cert_name)
+##        count = 0
+##
+##        for key in bucket.list():
+##            count += 1
+##
+##        self.assertEqual(count, 2)
+##        certs = get_all_server_certs('179727101194')
+##        self.assertEqual(len(certs), 1)
+##
+##
+##