# from lemur.tests import LemurTestCase # from lemur.certificates.models import Certificate # from moto import mock_iam, mock_sts # class IAMTestCase(LemurTestCase): # @mock_sts # @mock_iam # def test_get_all_server_certs(self): # from lemur.common.services.aws.iam import upload_cert, get_all_server_certs # cert = Certificate(TEST_CERT) # upload_cert('1111', cert, TEST_KEY) # certs = get_all_server_certs('1111') # self.assertEquals(len(certs), 1) # # @mock_sts # @mock_iam # def test_get_server_cert(self): # from lemur.common.services.aws.iam import upload_cert, get_cert_from_arn # cert = Certificate(TEST_CERT) # upload_cert('1111', cert, TEST_KEY) # body, chain = get_cert_from_arn('arn:aws:iam::123456789012:server-certificate/AHB-dfdsflkj.net-NetflixInc-20140525-20150525') # self.assertTrue(body) # # @mock_sts # @mock_iam # def test_upload_server_cert(self): # from lemur.common.services.aws.iam import upload_cert # cert = Certificate(TEST_CERT) # response = upload_cert('1111', cert, TEST_KEY) # self.assertEquals(response['upload_server_certificate_response']['upload_server_certificate_result']['server_certificate_metadata']['server_certificate_name'], 'AHB-dfdsflkj.net-NetflixInc-20140525-20150525') # #