def test_can_create_default_retry_handler():
mock_client = mock.Mock()
mock_client.meta.service_model.service_id = model.ServiceId('my-service')
assert_is_instance(standard.register_retry_handler(mock_client),
standard.RetryHandler)
call_args_list = mock_client.meta.events.register.call_args_list
# We should have registered the retry quota to after-calls
first_call = call_args_list[0][0]
second_call = call_args_list[1][0]
# Not sure if there's a way to verify the class associated with the
# bound method matches what we expect.
assert_equal(first_call[0], 'after-call.my-service')
assert_equal(second_call[0], 'needs-retry.my-service')
def test_hyphenize_lower_cases(self):
self.assertEqual(model.ServiceId('MyService').hyphenize(), 'myservice')
def test_hypenize_replaces_spaces(self):
self.assertEqual(
model.ServiceId('my service').hyphenize(), 'my-service')