38 lines
1.4 KiB
Python
38 lines
1.4 KiB
Python
|
from lemur import app
|
||
|
from lemur.tests import LemurTestCase
|
||
|
from lemur.tests.constants import TEST_CERT, TEST_KEY
|
||
|
|
||
|
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')
|
||
|
|
||
|
|