Index: tools/telemetry/third_party/gsutil/third_party/apitools/apitools/base/py/base_api_test.py |
diff --git a/tools/telemetry/third_party/gsutil/third_party/apitools/apitools/base/py/base_api_test.py b/tools/telemetry/third_party/gsutil/third_party/apitools/apitools/base/py/base_api_test.py |
deleted file mode 100644 |
index 86141a3a71a9acb7a8f58b5a6b366772508bc314..0000000000000000000000000000000000000000 |
--- a/tools/telemetry/third_party/gsutil/third_party/apitools/apitools/base/py/base_api_test.py |
+++ /dev/null |
@@ -1,179 +0,0 @@ |
-import datetime |
-import sys |
- |
-from protorpc import message_types |
-from protorpc import messages |
-from six.moves import urllib_parse |
-import unittest2 |
- |
-from apitools.base.py import base_api |
-from apitools.base.py import encoding |
-from apitools.base.py import http_wrapper |
- |
- |
-class SimpleMessage(messages.Message): |
- field = messages.StringField(1) |
- |
- |
-class MessageWithTime(messages.Message): |
- timestamp = message_types.DateTimeField(1) |
- |
- |
-class MessageWithRemappings(messages.Message): |
- |
- class AnEnum(messages.Enum): |
- value_one = 1 |
- value_two = 2 |
- |
- str_field = messages.StringField(1) |
- enum_field = messages.EnumField('AnEnum', 2) |
- |
- |
-encoding.AddCustomJsonFieldMapping( |
- MessageWithRemappings, 'str_field', 'remapped_field') |
-encoding.AddCustomJsonEnumMapping( |
- MessageWithRemappings.AnEnum, 'value_one', 'ONE/TWO') |
- |
- |
-class StandardQueryParameters(messages.Message): |
- field = messages.StringField(1) |
- prettyPrint = messages.BooleanField( |
- 5, default=True) # pylint: disable=invalid-name |
- pp = messages.BooleanField(6, default=True) |
- |
- |
-class FakeCredentials(object): |
- |
- def authorize(self, _): # pylint: disable=invalid-name |
- return None |
- |
- |
-class FakeClient(base_api.BaseApiClient): |
- MESSAGES_MODULE = sys.modules[__name__] |
- _PACKAGE = 'package' |
- _SCOPES = ['scope1'] |
- _CLIENT_ID = 'client_id' |
- _CLIENT_SECRET = 'client_secret' |
- |
- |
-class FakeService(base_api.BaseApiService): |
- |
- def __init__(self, client=None): |
- client = client or FakeClient( |
- 'http://www.example.com/', credentials=FakeCredentials()) |
- super(FakeService, self).__init__(client) |
- |
- |
-class BaseApiTest(unittest2.TestCase): |
- |
- def __GetFakeClient(self): |
- return FakeClient('', credentials=FakeCredentials()) |
- |
- def testUrlNormalization(self): |
- client = FakeClient('http://www.googleapis.com', get_credentials=False) |
- self.assertTrue(client.url.endswith('/')) |
- |
- def testNoCredentials(self): |
- client = FakeClient('', get_credentials=False) |
- self.assertIsNotNone(client) |
- self.assertIsNone(client._credentials) |
- |
- def testIncludeEmptyFieldsClient(self): |
- msg = SimpleMessage() |
- client = self.__GetFakeClient() |
- self.assertEqual('{}', client.SerializeMessage(msg)) |
- with client.IncludeFields(('field',)): |
- self.assertEqual('{"field": null}', client.SerializeMessage(msg)) |
- |
- def testJsonResponse(self): |
- method_config = base_api.ApiMethodInfo( |
- response_type_name='SimpleMessage') |
- service = FakeService() |
- http_response = http_wrapper.Response( |
- info={'status': '200'}, content='{"field": "abc"}', |
- request_url='http://www.google.com') |
- response_message = SimpleMessage(field='abc') |
- self.assertEqual(response_message, service.ProcessHttpResponse( |
- method_config, http_response)) |
- with service.client.JsonResponseModel(): |
- self.assertEqual( |
- http_response.content, |
- service.ProcessHttpResponse(method_config, http_response)) |
- |
- def testAdditionalHeaders(self): |
- additional_headers = {'Request-Is-Awesome': '1'} |
- client = self.__GetFakeClient() |
- |
- # No headers to start |
- http_request = http_wrapper.Request('http://www.example.com') |
- new_request = client.ProcessHttpRequest(http_request) |
- self.assertFalse('Request-Is-Awesome' in new_request.headers) |
- |
- # Add a new header and ensure it's added to the request. |
- client.additional_http_headers = additional_headers |
- http_request = http_wrapper.Request('http://www.example.com') |
- new_request = client.ProcessHttpRequest(http_request) |
- self.assertTrue('Request-Is-Awesome' in new_request.headers) |
- |
- def testQueryEncoding(self): |
- method_config = base_api.ApiMethodInfo( |
- request_type_name='MessageWithTime', query_params=['timestamp']) |
- service = FakeService() |
- request = MessageWithTime( |
- timestamp=datetime.datetime(2014, 10, 0o7, 12, 53, 13)) |
- http_request = service.PrepareHttpRequest(method_config, request) |
- |
- url_timestamp = urllib_parse.quote(request.timestamp.isoformat()) |
- self.assertTrue(http_request.url.endswith(url_timestamp)) |
- |
- def testPrettyPrintEncoding(self): |
- method_config = base_api.ApiMethodInfo( |
- request_type_name='MessageWithTime', query_params=['timestamp']) |
- service = FakeService() |
- request = MessageWithTime( |
- timestamp=datetime.datetime(2014, 10, 0o7, 12, 53, 13)) |
- |
- global_params = StandardQueryParameters() |
- http_request = service.PrepareHttpRequest(method_config, request, |
- global_params=global_params) |
- self.assertFalse('prettyPrint' in http_request.url) |
- self.assertFalse('pp' in http_request.url) |
- |
- global_params.prettyPrint = False # pylint: disable=invalid-name |
- global_params.pp = False |
- |
- http_request = service.PrepareHttpRequest(method_config, request, |
- global_params=global_params) |
- self.assertTrue('prettyPrint=0' in http_request.url) |
- self.assertTrue('pp=0' in http_request.url) |
- |
- def testQueryRemapping(self): |
- method_config = base_api.ApiMethodInfo( |
- request_type_name='MessageWithRemappings', |
- query_params=['remapped_field', 'enum_field']) |
- request = MessageWithRemappings( |
- str_field='foo', enum_field=MessageWithRemappings.AnEnum.value_one) |
- http_request = FakeService().PrepareHttpRequest(method_config, request) |
- result_params = urllib_parse.parse_qs( |
- urllib_parse.urlparse(http_request.url).query) |
- expected_params = {'enum_field': 'ONE%2FTWO', 'remapped_field': 'foo'} |
- self.assertTrue(expected_params, result_params) |
- |
- def testPathRemapping(self): |
- method_config = base_api.ApiMethodInfo( |
- relative_path='parameters/{remapped_field}/remap/{enum_field}', |
- request_type_name='MessageWithRemappings', |
- path_params=['remapped_field', 'enum_field']) |
- request = MessageWithRemappings( |
- str_field='gonna', |
- enum_field=MessageWithRemappings.AnEnum.value_one) |
- service = FakeService() |
- expected_url = service.client.url + 'parameters/gonna/remap/ONE%2FTWO' |
- http_request = service.PrepareHttpRequest(method_config, request) |
- self.assertEqual(expected_url, http_request.url) |
- |
- method_config.relative_path = ( |
- 'parameters/{+remapped_field}/remap/{+enum_field}') |
- expected_url = service.client.url + 'parameters/gonna/remap/ONE/TWO' |
- http_request = service.PrepareHttpRequest(method_config, request) |
- self.assertEqual(expected_url, http_request.url) |