def prepare_discovery_request(self, response_body):
self._response = test_utils.MockConnectionResponse(200, response_body)
discovery = discovery_service.DiscoveryService(
self.api_config_manager)
discovery._discovery_proxy = self.mox.CreateMock(
discovery_api_proxy.DiscoveryApiProxy)
return discovery
def prepare_discovery_request(self, status_code, body):
self.mox.StubOutWithMock(http.client.HTTPSConnection, 'request')
self.mox.StubOutWithMock(http.client.HTTPSConnection, 'getresponse')
self.mox.StubOutWithMock(http.client.HTTPSConnection, 'close')
http.client.HTTPSConnection.request(mox.IsA(str), mox.IsA(str),
mox.IgnoreArg(), mox.IsA(dict))
http.client.HTTPSConnection.getresponse().AndReturn(
test_utils.MockConnectionResponse(status_code, body))
http.client.HTTPSConnection.close()
def test_generate_discovery_doc_invalid_format(self):
self._response = test_utils.MockConnectionResponse(400, 'Error')
self.assertRaises(ValueError,
self.discovery_api.generate_discovery_doc,
self.api_config_dict, 'blah')