86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
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)
|
|
##
|
|
##
|
|
##
|