| Index: tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/__init__.py
|
| diff --git a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/__init__.py b/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/__init__.py
|
| deleted file mode 100644
|
| index 3332965701a380a6991a57d70f4174f5bfe2dc4d..0000000000000000000000000000000000000000
|
| --- a/tools/telemetry/third_party/gsutilz/third_party/boto/tests/unit/__init__.py
|
| +++ /dev/null
|
| @@ -1,110 +0,0 @@
|
| -from boto.compat import http_client
|
| -from tests.compat import mock, unittest
|
| -
|
| -
|
| -class AWSMockServiceTestCase(unittest.TestCase):
|
| - """Base class for mocking aws services."""
|
| - # This param is used by the unittest module to display a full
|
| - # diff when assert*Equal methods produce an error message.
|
| - maxDiff = None
|
| - connection_class = None
|
| -
|
| - def setUp(self):
|
| - self.https_connection = mock.Mock(spec=http_client.HTTPSConnection)
|
| - self.https_connection.debuglevel = 0
|
| - self.https_connection_factory = (
|
| - mock.Mock(return_value=self.https_connection), ())
|
| - self.service_connection = self.create_service_connection(
|
| - https_connection_factory=self.https_connection_factory,
|
| - aws_access_key_id='aws_access_key_id',
|
| - aws_secret_access_key='aws_secret_access_key')
|
| - self.initialize_service_connection()
|
| -
|
| - def initialize_service_connection(self):
|
| - self.actual_request = None
|
| - self.original_mexe = self.service_connection._mexe
|
| - self.service_connection._mexe = self._mexe_spy
|
| - self.proxy = None
|
| - self.use_proxy = False
|
| -
|
| - def create_service_connection(self, **kwargs):
|
| - if self.connection_class is None:
|
| - raise ValueError("The connection_class class attribute must be "
|
| - "set to a non-None value.")
|
| - return self.connection_class(**kwargs)
|
| -
|
| - def _mexe_spy(self, request, *args, **kwargs):
|
| - self.actual_request = request
|
| - return self.original_mexe(request, *args, **kwargs)
|
| -
|
| - def create_response(self, status_code, reason='', header=[], body=None):
|
| - if body is None:
|
| - body = self.default_body()
|
| - response = mock.Mock(spec=http_client.HTTPResponse)
|
| - response.status = status_code
|
| - response.read.return_value = body
|
| - response.reason = reason
|
| -
|
| - response.getheaders.return_value = header
|
| - response.msg = dict(header)
|
| -
|
| - def overwrite_header(arg, default=None):
|
| - header_dict = dict(header)
|
| - if arg in header_dict:
|
| - return header_dict[arg]
|
| - else:
|
| - return default
|
| - response.getheader.side_effect = overwrite_header
|
| -
|
| - return response
|
| -
|
| - def assert_request_parameters(self, params, ignore_params_values=None):
|
| - """Verify the actual parameters sent to the service API."""
|
| - request_params = self.actual_request.params.copy()
|
| - if ignore_params_values is not None:
|
| - for param in ignore_params_values:
|
| - try:
|
| - del request_params[param]
|
| - except KeyError:
|
| - pass
|
| - self.assertDictEqual(request_params, params)
|
| -
|
| - def set_http_response(self, status_code, reason='', header=[], body=None):
|
| - http_response = self.create_response(status_code, reason, header, body)
|
| - self.https_connection.getresponse.return_value = http_response
|
| -
|
| - def default_body(self):
|
| - return ''
|
| -
|
| -
|
| -class MockServiceWithConfigTestCase(AWSMockServiceTestCase):
|
| - def setUp(self):
|
| - super(MockServiceWithConfigTestCase, self).setUp()
|
| - self.environ = {}
|
| - self.config = {}
|
| - self.config_patch = mock.patch('boto.provider.config.get',
|
| - self.get_config)
|
| - self.has_config_patch = mock.patch('boto.provider.config.has_option',
|
| - self.has_config)
|
| - self.environ_patch = mock.patch('os.environ', self.environ)
|
| - self.config_patch.start()
|
| - self.has_config_patch.start()
|
| - self.environ_patch.start()
|
| -
|
| - def tearDown(self):
|
| - self.config_patch.stop()
|
| - self.has_config_patch.stop()
|
| - self.environ_patch.stop()
|
| -
|
| - def has_config(self, section_name, key):
|
| - try:
|
| - self.config[section_name][key]
|
| - return True
|
| - except KeyError:
|
| - return False
|
| -
|
| - def get_config(self, section_name, key, default=None):
|
| - try:
|
| - return self.config[section_name][key]
|
| - except KeyError:
|
| - return None
|
|
|