def test_wrap_method_basic():
method = mock.Mock(spec=['__call__'], return_value=42)
wrapped_method = google.api.core.gapic_v1.method.wrap_method(method,
metadata=None)
result = wrapped_method(1, 2, meep='moop')
assert result == 42
method.assert_called_once_with(1, 2, meep='moop')
def test_wrap_method_with_overriding_timeout_as_a_number():
method = mock.Mock(spec=['__call__'], return_value=42)
default_retry = retry.Retry()
default_timeout = timeout.ConstantTimeout(60)
wrapped_method = google.api.core.gapic_v1.method.wrap_method(
method, default_retry, default_timeout)
result = wrapped_method(timeout=22)
assert result == 42
method.assert_called_once_with(timeout=22, metadata=mock.ANY)
def test_wrap_method_with_custom_metadata():
method = mock.Mock(spec=['__call__'])
wrapped_method = google.api.core.gapic_v1.method.wrap_method(
method, metadata={'foo': 'bar'})
wrapped_method(1, 2, meep='moop')
method.assert_called_once_with(1, 2, meep='moop', metadata=mock.ANY)
metadata = method.call_args[1]['metadata']
assert len(metadata) == 2
assert ('foo', 'bar') in metadata
def test_wrap_method_with_default_metadata():
method = mock.Mock(spec=['__call__'])
wrapped_method = google.api.core.gapic_v1.method.wrap_method(method)
wrapped_method(1, 2, meep='moop')
method.assert_called_once_with(1, 2, meep='moop', metadata=mock.ANY)
metadata = method.call_args[1]['metadata']
assert len(metadata) == 1
assert metadata[0][0] == 'x-goog-api-client'
assert 'api-core' in metadata[0][1]
def test_wrap_method_with_merged_metadata():
method = mock.Mock(spec=['__call__'])
wrapped_method = google.api.core.gapic_v1.method.wrap_method(
method, metadata={'x-goog-api-client': 'foo/1.2.3'})
wrapped_method(1, 2, meep='moop')
method.assert_called_once_with(1, 2, meep='moop', metadata=mock.ANY)
metadata = method.call_args[1]['metadata']
assert len(metadata) == 1
assert metadata[0][0] == 'x-goog-api-client'
assert metadata[0][1].endswith(' foo/1.2.3')
