def test_fetch_service_config_failed(self):
mock_response = mock.MagicMock()
mock_response.status = 403
mock_http_client = mock.MagicMock()
mock_http_client.request.return_value = mock_response
ServiceConfigFetchTest._get_http_client.return_value = mock_http_client
with self.assertRaisesRegexp(Exception, "status code 403"):
service_config.fetch_service_config()
def test_fetch_service_config_failed(self):
mock_response = mock.MagicMock()
mock_response.status = 403
mock_http_client = mock.MagicMock()
mock_http_client.request.return_value = mock_response
ServiceConfigFetchTest._get_http_client.return_value = mock_http_client
with self.assertRaisesRegexp(Exception, "status code 403"):
service_config.fetch_service_config()
def test_fetch_service_config_with_wrong_service_version(self):
mock_response = mock.MagicMock()
mock_response.status = 200
config = copy.deepcopy(ServiceConfigFetchTest._SERVICE_CONFIG_JSON)
config["id"] = "incorrect_service_version"
mock_response.data = json.dumps(config)
mock_http_client = mock.MagicMock()
mock_http_client.request.return_value = mock_response
ServiceConfigFetchTest._get_http_client.return_value = mock_http_client
message = ("Unexpected service version in service config: " +
config["id"])
with self.assertRaisesRegexp(ValueError, message):
service_config.fetch_service_config()
def test_fetch_service_config_with_wrong_service_name(self):
mock_response = mock.MagicMock()
mock_response.status = 200
config = copy.deepcopy(ServiceConfigFetchTest._SERVICE_CONFIG_JSON)
config["name"] = "incorrect_service_name"
mock_response.data = json.dumps(config)
mock_http_client = mock.MagicMock()
mock_http_client.request.return_value = mock_response
ServiceConfigFetchTest._get_http_client.return_value = mock_http_client
message = ("Unexpected service name in service config: " +
config["name"])
with self.assertRaisesRegexp(ValueError, message):
service_config.fetch_service_config()
def test_fetch_service_config(self):
mock_response = mock.MagicMock()
mock_response.status = 200
mock_response.data = json.dumps(ServiceConfigFetchTest._SERVICE_CONFIG_JSON)
mock_http_client = mock.MagicMock()
mock_http_client.request.return_value = mock_response
ServiceConfigFetchTest._get_http_client.return_value = mock_http_client
service = encoding.JsonToMessage(messages.Service,
json.dumps(self._SERVICE_CONFIG_JSON))
self.assertEqual(service, service_config.fetch_service_config())
template = service_config._SERVICE_MGMT_URL_TEMPLATE
url = template.format(ServiceConfigFetchTest._SERVICE_NAME,
ServiceConfigFetchTest._SERVICE_VERSION)
headers={"Authorization": "Bearer " + ServiceConfigFetchTest._ACCESS_TOKEN}
mock_http_client.request.assert_called_once_with("GET", url,
headers=headers)
def test_fetch_service_config(self):
mock_response = mock.MagicMock()
mock_response.status = 200
mock_response.data = json.dumps(
ServiceConfigFetchTest._SERVICE_CONFIG_JSON)
mock_http_client = mock.MagicMock()
mock_http_client.request.return_value = mock_response
ServiceConfigFetchTest._get_http_client.return_value = mock_http_client
service = encoding.JsonToMessage(messages.Service,
json.dumps(self._SERVICE_CONFIG_JSON))
self.assertEqual(service, service_config.fetch_service_config())
template = service_config._SERVICE_MGMT_URL_TEMPLATE
url = template.format(ServiceConfigFetchTest._SERVICE_NAME,
ServiceConfigFetchTest._SERVICE_VERSION)
headers = {
"Authorization": "Bearer " + ServiceConfigFetchTest._ACCESS_TOKEN
}
mock_http_client.request.assert_called_once_with("GET",
url,
headers=headers)
def test_no_service_version(self):
del os.environ["ENDPOINTS_SERVICE_VERSION"]
message = 'The "ENDPOINTS_SERVICE_VERSION" environment variable is not set'
with self.assertRaisesRegexp(ValueError, message):
service_config.fetch_service_config()
def test_no_service_version(self):
del os.environ["ENDPOINTS_SERVICE_VERSION"]
message = 'The "ENDPOINTS_SERVICE_VERSION" environment variable is not set'
with self.assertRaisesRegexp(ValueError, message):
service_config.fetch_service_config()