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", ]