def test_get_limits_again(self):
"""test that existing limits dict is returned on subsequent calls"""
mock_limits = Mock(spec_set=AwsLimit)
cls = _S3Service(21, 43)
cls.limits = mock_limits
res = cls.get_limits()
assert res == mock_limits
def test_get_limits_again(self):
"""test that existing limits dict is returned on subsequent calls"""
mock_limits = Mock(spec_set=AwsLimit)
cls = _S3Service(21, 43)
cls.limits = mock_limits
res = cls.get_limits()
assert res == mock_limits
def test_init(self):
"""test __init__()"""
cls = _S3Service(21, 43)
assert cls.service_name == 'S3'
assert cls.api_name == 's3'
assert cls.conn is None
assert cls.warning_threshold == 21
assert cls.critical_threshold == 43
def test_init(self):
"""test __init__()"""
cls = _S3Service(21, 43)
assert cls.service_name == 'S3'
assert cls.api_name == 's3'
assert cls.conn is None
assert cls.warning_threshold == 21
assert cls.critical_threshold == 43
def test_get_limits(self):
cls = _S3Service(21, 43)
cls.limits = {}
res = cls.get_limits()
assert sorted(res.keys()) == sorted([
'Buckets',
])
assert res['Buckets'].service == cls
assert res['Buckets'].def_warning_threshold == 21
assert res['Buckets'].def_critical_threshold == 43
assert res['Buckets'].default_limit == 100
def test_get_limits(self):
cls = _S3Service(21, 43)
cls.limits = {}
res = cls.get_limits()
assert sorted(res.keys()) == sorted([
'Buckets',
])
assert res['Buckets'].service == cls
assert res['Buckets'].def_warning_threshold == 21
assert res['Buckets'].def_critical_threshold == 43
assert res['Buckets'].default_limit == 100
def test_find_usage(self):
mock_buckets = Mock()
mock_buckets.all.return_value = ['a', 'b', 'c']
mock_conn = Mock(buckets=mock_buckets)
with patch('%s.connect_resource' % pb) as mock_connect:
cls = _S3Service(21, 43)
cls.resource_conn = mock_conn
assert cls._have_usage is False
cls.find_usage()
assert mock_connect.mock_calls == [call()]
assert cls._have_usage is True
assert mock_buckets.mock_calls == [call.all()]
assert len(cls.limits['Buckets'].get_current_usage()) == 1
assert cls.limits['Buckets'].get_current_usage()[0].get_value() == 3
def test_find_usage(self):
mock_buckets = Mock()
mock_buckets.all.return_value = ['a', 'b', 'c']
mock_conn = Mock(buckets=mock_buckets)
with patch('%s.connect_resource' % pb) as mock_connect:
cls = _S3Service(21, 43)
cls.resource_conn = mock_conn
assert cls._have_usage is False
cls.find_usage()
assert mock_connect.mock_calls == [call()]
assert cls._have_usage is True
assert mock_buckets.mock_calls == [call.all()]
assert len(cls.limits['Buckets'].get_current_usage()) == 1
assert cls.limits['Buckets'].get_current_usage()[0].get_value() == 3
def test_required_iam_permissions(self):
cls = _S3Service(21, 43)
assert cls.required_iam_permissions() == ['s3:ListAllMyBuckets']
def test_required_iam_permissions(self):
cls = _S3Service(21, 43)
assert cls.required_iam_permissions() == [
's3:ListAllMyBuckets'
]
