def test_set_limit_overrides_ta(self): limits = sample_limits() self.mock_svc1.get_limits.return_value = limits["SvcFoo"] self.mock_svc2.get_limits.return_value = limits["SvcBar"] overrides = {"SvcBar": {"bar limit2": 100}, "SvcFoo": {"foo limit3": 3}} self.cls.set_limit_overrides(overrides, override_ta=False) assert self.mock_svc1.mock_calls == [call.set_limit_override("foo limit3", 3, override_ta=False)] assert self.mock_svc2.mock_calls == [call.set_limit_override("bar limit2", 100, override_ta=False)]
def test_set_limit_overrides(self): overrides = { 'EC2/Foo bar': "2", 'ElastiCache/Cache cluster subnet groups': "100", } mock_checker = Mock(spec_set=AwsLimitChecker) self.cls.checker = mock_checker self.cls.set_limit_overrides(overrides) assert mock_checker.mock_calls == [ call.set_limit_override('EC2', 'Foo bar', 2), call.set_limit_override('ElastiCache', 'Cache cluster subnet groups', 100) ]
def test_set_limit_override(self): limits = sample_limits() self.mock_svc1.get_limits.return_value = limits['SvcFoo'] self.cls.set_limit_override('SvcFoo', 'foo limit3', 99) assert self.mock_svc1.mock_calls == [ call.set_limit_override('foo limit3', 99, override_ta=True) ]
def test_set_limit_overrides(self): overrides = { 'EC2/Foo bar': "2", 'ElastiCache/Cache cluster subnet groups': "100", } mock_checker = Mock(spec_set=AwsLimitChecker) self.cls.checker = mock_checker self.cls.set_limit_overrides(overrides) assert mock_checker.mock_calls == [ call.set_limit_override('EC2', 'Foo bar', 2), call.set_limit_override( 'ElastiCache', 'Cache cluster subnet groups', 100 ) ]
def test_set_limit_override(self): limits = sample_limits() self.mock_svc1.get_limits.return_value = limits['SvcFoo'] self.cls.set_limit_override('SvcFoo', 'foo limit3', 99) assert self.mock_svc1.mock_calls == [ call.set_limit_override('foo limit3', 99, override_ta=True) ]
def test_set_limit_override(self): mock_limit = Mock(spec_set=AwsLimit) type(mock_limit).default_limit = 5 cls = AwsServiceTester(1, 2) cls.limits['foo'] = mock_limit cls.set_limit_override('foo', 10) assert mock_limit.mock_calls == [ call.set_limit_override(10, override_ta=True) ]
def test_set_limit_override(self): mock_limit = Mock(spec_set=AwsLimit) type(mock_limit).default_limit = 5 cls = AwsServiceTester(1, 2) cls.limits['foo'] = mock_limit cls.set_limit_override('foo', 10) assert mock_limit.mock_calls == [ call.set_limit_override(10, override_ta=True) ]
def test_set_limit_overrides_ta(self): limits = sample_limits() self.mock_svc1.get_limits.return_value = limits['SvcFoo'] self.mock_svc2.get_limits.return_value = limits['SvcBar'] overrides = { 'SvcBar': { 'bar limit2': 100, }, 'SvcFoo': { 'foo limit3': 3, }, } self.cls.set_limit_overrides(overrides, override_ta=False) assert self.mock_svc1.mock_calls == [ call.set_limit_override('foo limit3', 3, override_ta=False) ] assert self.mock_svc2.mock_calls == [ call.set_limit_override('bar limit2', 100, override_ta=False) ]
def test_set_limit_overrides_ta(self): limits = sample_limits() self.mock_svc1.get_limits.return_value = limits['SvcFoo'] self.mock_svc2.get_limits.return_value = limits['SvcBar'] overrides = { 'SvcBar': { 'bar limit2': 100, }, 'SvcFoo': { 'foo limit3': 3, }, } self.cls.set_limit_overrides(overrides, override_ta=False) assert self.mock_svc1.mock_calls == [ call.set_limit_override('foo limit3', 3, override_ta=False) ] assert self.mock_svc2.mock_calls == [ call.set_limit_override('bar limit2', 100, override_ta=False) ]
def test_set_limit_override_ta(self): limits = sample_limits() self.mock_svc1.get_limits.return_value = limits["SvcFoo"] self.cls.set_limit_override("SvcFoo", "foo limit3", 99, override_ta=False) assert self.mock_svc1.mock_calls == [call.set_limit_override("foo limit3", 99, override_ta=False)]