def test_get_limits_again(self):
"""test that existing limits dict is returned on subsequent calls"""
mock_limits = Mock()
cls = _RedshiftService(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()
cls = _RedshiftService(21, 43, {}, None)
cls.limits = mock_limits
res = cls.get_limits()
assert res == mock_limits
def test_init(self):
"""test __init__()"""
cls = _RedshiftService(21, 43, {}, None)
assert cls.service_name == 'Redshift'
assert cls.api_name == 'redshift'
assert cls.conn is None
assert cls.warning_threshold == 21
assert cls.critical_threshold == 43
def test_init(self):
"""test __init__()"""
cls = _RedshiftService(21, 43)
assert cls.service_name == 'Redshift'
assert cls.api_name == 'redshift'
assert cls.conn is None
assert cls.warning_threshold == 21
assert cls.critical_threshold == 43
def test_get_limits(self):
cls = _RedshiftService(21, 43, {}, None)
cls.limits = {}
res = cls.get_limits()
assert sorted(res.keys()) == sorted([
'Redshift manual snapshots',
'Redshift subnet groups',
])
for name, limit in res.items():
assert limit.service == cls
assert limit.def_warning_threshold == 21
assert limit.def_critical_threshold == 43
def test_get_limits(self):
cls = _RedshiftService(21, 43)
cls.limits = {}
res = cls.get_limits()
assert sorted(res.keys()) == sorted([
'Redshift manual snapshots',
'Redshift subnet groups',
])
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_subnet_groups(self):
response = result_fixtures.Redshift.test_describe_cluster_subnet_groups
limit_key = 'Redshift subnet groups'
mock_conn = Mock()
mock_conn.describe_cluster_subnet_groups.return_value = response
cls = _RedshiftService(21, 43, {'region_name': 'us-west-2'}, None)
cls.conn = mock_conn
cls._find_cluster_subnet_groups()
assert mock_conn.mock_calls == [call.describe_cluster_subnet_groups()]
assert len(cls.limits[limit_key].get_current_usage()) == 1
assert cls.limits[limit_key].get_current_usage()[0].get_value() == 3
def test_find_usage_subnet_groups(self):
response = result_fixtures.Redshift.test_describe_cluster_subnet_groups
limit_key = 'Redshift subnet groups'
mock_conn = Mock()
mock_conn.describe_cluster_subnet_groups.return_value = response
cls = _RedshiftService(21, 43, {'region_name': 'us-west-2'})
cls.conn = mock_conn
cls._find_cluster_subnet_groups()
assert mock_conn.mock_calls == [call.describe_cluster_subnet_groups()]
assert len(cls.limits[limit_key].get_current_usage()) == 1
assert cls.limits[limit_key].get_current_usage()[
0].get_value() == 3
def test_find_usage_manual_snapshots(self):
response = result_fixtures.Redshift.test_describe_cluster_snapshots
limit_key = 'Redshift manual snapshots'
mock_conn = Mock()
mock_conn.describe_cluster_snapshots.return_value = response
cls = _RedshiftService(21, 43, {'region_name': 'us-west-2'})
cls.conn = mock_conn
cls._find_cluster_manual_snapshots()
assert mock_conn.mock_calls == [
call.describe_cluster_snapshots(SnapshotType='manual')
]
assert len(cls.limits[limit_key].get_current_usage()) == 1
assert cls.limits[limit_key].get_current_usage()[0].get_value() == 2
def test_find_usage_manual_snapshots(self):
response = result_fixtures.Redshift.test_describe_cluster_snapshots
limit_key = 'Redshift manual snapshots'
mock_conn = Mock()
mock_conn.describe_cluster_snapshots.return_value = response
cls = _RedshiftService(21, 43, {'region_name': 'us-west-2'})
cls.conn = mock_conn
cls._find_cluster_manual_snapshots()
assert mock_conn.mock_calls == [
call.describe_cluster_snapshots(SnapshotType='manual')
]
assert len(cls.limits[limit_key].get_current_usage()) == 1
assert cls.limits[limit_key].get_current_usage()[
0].get_value() == 2
def test_find_usage(self):
"""test find usage method calls other methods"""
mock_conn = Mock()
with patch('%s.connect' % pb) as mock_connect:
with patch.multiple(
pb,
_find_cluster_manual_snapshots=DEFAULT,
_find_cluster_subnet_groups=DEFAULT,
) as mocks:
cls = _RedshiftService(21, 43, {}, None)
cls.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_conn.mock_calls == []
for x in [
'_find_cluster_manual_snapshots',
'_find_cluster_subnet_groups',
]:
assert mocks[x].mock_calls == [call()]
def test_find_usage(self):
"""test find usage method calls other methods"""
mock_conn = Mock()
with patch('%s.connect' % pb) as mock_connect:
with patch.multiple(
pb,
_find_cluster_manual_snapshots=DEFAULT,
_find_cluster_subnet_groups=DEFAULT,
) as mocks:
cls = _RedshiftService(21, 43)
cls.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_conn.mock_calls == []
for x in [
'_find_cluster_manual_snapshots',
'_find_cluster_subnet_groups',
]:
assert mocks[x].mock_calls == [call()]
def test_required_iam_permissions(self):
cls = _RedshiftService(21, 43, {}, None)
assert cls.required_iam_permissions() == [
"redshift:DescribeClusterSnapshots",
"redshift:DescribeClusterSubnetGroups",
]
def test_required_iam_permissions(self):
cls = _RedshiftService(21, 43)
assert cls.required_iam_permissions() == [
"redshift:DescribeClusterSnapshots",
"redshift:DescribeClusterSubnetGroups",
]