def test_get_google_ads_failure_decode_error(self): """Returns none if the google ads failure cannot be decoded.""" interceptor = Interceptor(default_version) mock_failure_value = _MOCK_FAILURE_VALUE + b"1234" mock_metadata = ((interceptor._failure_key, mock_failure_value), ) result = interceptor._get_google_ads_failure(mock_metadata) self.assertEqual(result, None)
def test_deferred_error_proto_module_load(self): """Tests that import_module is called when an API error is received.""" with mock.patch("google.ads.googleads.interceptors." "interceptor.import_module") as import_mock: interceptor = Interceptor(default_version) mock_metadata = ((interceptor._failure_key, _MOCK_FAILURE_VALUE), ) interceptor._get_google_ads_failure(mock_metadata) import_mock.assert_called_once_with( f"google.ads.googleads.{default_version}.errors.types.errors")
def test_get_error_from_response_does_not_cache_error(self): """Ensures errors are not cached across requests..""" interceptor = Interceptor(default_version) class MockRpcErrorResponse(grpc.RpcError): def code(self): return grpc.StatusCode.INVALID_ARGUMENT def trailing_metadata(self): return (( interceptor._failure_key, b"\n \n\x02\x08\x10\x12\x1aInvalid customer ID '123'.", ), ) def exception(self): return self response = MockRpcErrorResponse() first_result = interceptor._get_error_from_response(response) second_result = interceptor._get_error_from_response(response) self.assertIsNot(first_result, second_result)
def test_parse_metadata_to_json(self): mock_metadata = [ ("x-goog-api-client", "gl-python/123 grpc/123 gax/123"), ("developer-token", "0000000000"), ("login-customer-id", "9999999999"), ] result = Interceptor.parse_metadata_to_json(mock_metadata) self.assertEqual( result, "{\n" ' "developer-token": "REDACTED",\n' ' "login-customer-id": "9999999999",\n' ' "x-goog-api-client": "gl-python/123 ' 'grpc/123 gax/123"\n' "}", )
def test_init_no_error_proto_load(self): """Ensures that error proto modules are not loaded on init.""" interceptor = Interceptor(default_version) self.assertEqual(interceptor._error_protos, None)
def test_get_google_ads_failure_with_None(self): """Returns None if None is passed.""" interceptor = Interceptor(default_version) result = interceptor._get_google_ads_failure(None) self.assertEqual(result, None)
def test_get_google_ads_failure_no_failure_key(self): """Returns None if an error cannot be found in metadata.""" mock_metadata = (("another-key", "another-val"), ) interceptor = Interceptor(default_version) result = interceptor._get_google_ads_failure(mock_metadata) self.assertEqual(result, None)
def test_get_google_ads_failure(self): """Obtains the content of a google ads failure from metadata.""" interceptor = Interceptor(default_version) mock_metadata = ((interceptor._failure_key, _MOCK_FAILURE_VALUE), ) result = interceptor._get_google_ads_failure(mock_metadata) self.assertIsInstance(result, error_protos.GoogleAdsFailure)
def test_parse_metadata_to_json_with_none(self): mock_metadata = None result = Interceptor.parse_metadata_to_json(mock_metadata) self.assertEqual(result, "{}")
def test_get_request_id_no_id(self): """Ensures None is returned if metadata does't contain a request ID.""" mock_metadata = (("another-key", "another-val"), ) result = Interceptor.get_request_id_from_metadata(mock_metadata) self.assertEqual(result, None)
def test_get_request_id_from_metadata(self): """Ensures request-id is retrieved from metadata tuple.""" mock_metadata = (("request-id", "123456"), ) result = Interceptor.get_request_id_from_metadata(mock_metadata) self.assertEqual(result, "123456")