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')