def test_get_limits_again(self):
"""test that existing limits dict is returned on subsequent calls"""
mock_limits = Mock()
cls = _CertificatemanagerService(21, 43, {}, None)
cls.limits = mock_limits
res = cls.get_limits()
assert res == mock_limits
def test_init(self):
"""test __init__()"""
cls = _CertificatemanagerService(21, 43, {}, None)
assert cls.service_name == 'CertificateManager'
assert cls.api_name == 'acm'
assert cls.conn is None
assert cls.warning_threshold == 21
assert cls.critical_threshold == 43
def test_get_limits(self):
cls = _CertificatemanagerService(21, 43, {}, None)
cls.limits = {}
res = cls.get_limits()
assert sorted(res.keys()) == sorted([
'ACM certificates',
])
for name, limit in res.items():
assert limit.service == cls
assert limit.def_warning_threshold == 21
assert limit.def_critical_threshold == 43
def test_find_usage(self):
"""
Test overall find_usage method
Check that find_usage() method calls the other methods.
"""
with patch.multiple(pb,
connect=DEFAULT,
_find_usage_certificates=DEFAULT,
autospec=True) as mocks:
cls = _CertificatemanagerService(21, 43, {}, None)
assert cls._have_usage is False
cls.find_usage()
assert cls._have_usage is True
assert len(mocks) == 2
# the other methods should have been called
for x in ["_find_usage_certificates"]:
assert mocks[x].mock_calls == [call(cls)]
def test_find_usage_certificates(self):
"""
Verify the correctness of usage
This test mocks the AWS list_certificates response (after pagination).
"""
# Setup the mock and call the tested function
resp = result_fixtures.CertificateManager.test_find_usage_certificates
mock_conn = Mock()
with patch("%s.paginate_dict" % pbm) as mock_paginate:
cls = _CertificatemanagerService(21, 43, {}, None)
cls.conn = mock_conn
mock_paginate.return_value = resp
cls._find_usage_certificates()
# Check that usage values are correctly set
assert len(cls.limits["ACM certificates"].get_current_usage()) == 1
assert (cls.limits["ACM certificates"].get_current_usage()
[0].get_value() == 3)
assert (
cls.limits["ACM certificates"].get_current_usage()[0].resource_id
is None)
def test_required_iam_permissions(self):
cls = _CertificatemanagerService(21, 43, {}, None)
assert cls.required_iam_permissions() == ["acm:ListCertificates"]