Ejemplo n.º 1
0
 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)
Ejemplo n.º 2
0
 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")
Ejemplo n.º 3
0
    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)
Ejemplo n.º 4
0
    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'
            "}",
        )
Ejemplo n.º 5
0
 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)
Ejemplo n.º 6
0
 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)
Ejemplo n.º 7
0
 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)
Ejemplo n.º 8
0
 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)
Ejemplo n.º 9
0
 def test_parse_metadata_to_json_with_none(self):
     mock_metadata = None
     result = Interceptor.parse_metadata_to_json(mock_metadata)
     self.assertEqual(result, "{}")
Ejemplo n.º 10
0
 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)
Ejemplo n.º 11
0
 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")