def test_register_retry_handlers(self): service_data = { 'metadata': { 'endpointPrefix': 'foo' }, } session = mock.Mock() loader = mock.Mock() session.get_component.return_value = loader loader.load_data.return_value = { 'retry': { '__default__': { 'max_attempts': 10, 'delay': { 'type': 'exponential', 'base': 2, 'growth_factor': 5, }, }, }, } handlers.register_retries_for_service(service_data=service_data, session=session, service_name='foo') session.register.assert_called_with('needs-retry.foo', mock.ANY, unique_id='retry-config-foo')
def test_register_retry_for_handlers_with_no_endpoint_prefix(self): no_endpoint_prefix = {'metadata': {}} session = mock.Mock() handlers.register_retries_for_service(service_data=no_endpoint_prefix, session=mock.Mock(), service_name='foo') self.assertFalse(session.register.called)
def test_register_retry_for_handlers_with_no_endpoint_prefix(self): no_endpoint_prefix = {'metadata': {}} session = mock.Mock() handlers.register_retries_for_service(service_data=no_endpoint_prefix, session=mock.Mock(), service_name='foo') self.assertFalse(session.register.called)
def test_register_retry_handlers(self): service_data = {"metadata": {"endpointPrefix": "foo"}} session = mock.Mock() loader = mock.Mock() session.get_component.return_value = loader loader.load_data.return_value = { "retry": { "__default__": {"max_attempts": 10, "delay": {"type": "exponential", "base": 2, "growth_factor": 5}} } } handlers.register_retries_for_service(service_data=service_data, session=session, service_name="foo") session.register.assert_called_with("needs-retry.foo", mock.ANY, unique_id="retry-config-foo")
def test_register_retry_handlers(self): service_data = { 'metadata': {'endpointPrefix': 'foo'}, } session = mock.Mock() loader = mock.Mock() session.get_component.return_value = loader loader.load_data.return_value = { 'retry': { '__default__': { 'max_attempts': 10, 'delay': { 'type': 'exponential', 'base': 2, 'growth_factor': 5, }, }, }, } handlers.register_retries_for_service(service_data=service_data, session=session, service_name='foo') session.register.assert_called_with('needs-retry.foo', mock.ANY, unique_id='retry-config-foo')