def test_simple(self): limit = AwsLimit( 'limitname', self.mock_svc, 3, 1, 2 ) limit._current_usage = 2 assert limit.get_current_usage() == 2
def test_get_current_usage(self): limit = AwsLimit( 'limitname', self.mock_svc, 3, 1, 2 ) limit._current_usage = 2 assert limit.get_current_usage() == 2
def test_check_thresholds_pct_warn_crit(self): limit = AwsLimit('limitname', self.mock_svc, 100, 1, 2) u1 = AwsLimitUsage(limit, 50, resource_id='foo4bar') u2 = AwsLimitUsage(limit, 3, resource_id='foo3bar') u3 = AwsLimitUsage(limit, 95, resource_id='foo2bar') limit._current_usage = [u1, u2, u3] with patch('awslimitchecker.limit.AwsLimit.' '_get_thresholds') as mock_get_thresh: with patch('awslimitchecker.limit.AwsLimit.get_' 'limit') as mock_get_limit: mock_get_thresh.return_value = (None, 40, None, 80) mock_get_limit.return_value = 100 res = limit.check_thresholds() assert res is False assert limit._warnings == [u1] assert limit._criticals == [u3] assert mock_get_thresh.mock_calls == [call()] assert mock_get_limit.mock_calls == [call()]
def test_ta_zero(self): limit = AwsLimit('limitname', self.mock_svc, 3, 1, 2) u1 = AwsLimitUsage(limit, 4, resource_id='foo4bar') u2 = AwsLimitUsage(limit, 3, resource_id='foo3bar') u3 = AwsLimitUsage(limit, 2, resource_id='foo2bar') limit._current_usage = [u1, u2, u3] limit._set_ta_unlimited() with patch('awslimitchecker.limit.AwsLimit.' '_get_thresholds') as mock_get_thresh: with patch('awslimitchecker.limit.AwsLimit.get_' 'limit') as mock_get_limit: mock_get_thresh.return_value = (None, 40, None, 80) mock_get_limit.return_value = 0 res = limit.check_thresholds() assert res is True assert limit._warnings == [] assert limit._criticals == [] assert mock_get_thresh.mock_calls == [call()] assert mock_get_limit.mock_calls == [call(), call(), call()]
def test_check_thresholds_ta_unlimited(self): limit = AwsLimit('limitname', self.mock_svc, 3, 1, 2) u1 = AwsLimitUsage(limit, 4, resource_id='foo4bar') u2 = AwsLimitUsage(limit, 3, resource_id='foo3bar') u3 = AwsLimitUsage(limit, 2, resource_id='foo2bar') limit._current_usage = [u1, u2, u3] limit._set_ta_unlimited() with patch('awslimitchecker.limit.AwsLimit.' '_get_thresholds') as mock_get_thresh: with patch('awslimitchecker.limit.AwsLimit.get_' 'limit') as mock_get_limit: mock_get_thresh.return_value = (None, 40, None, 80) mock_get_limit.return_value = None res = limit.check_thresholds() assert res is True assert limit._warnings == [] assert limit._criticals == [] assert mock_get_thresh.mock_calls == [] assert mock_get_limit.mock_calls == [call()]