def test_get_failure(self):
http_request = request_mock(
http_client.NOT_FOUND, 'text/html', '<p>Error</p>')
with self.assertRaises(httplib2.HttpLib2Error):
_metadata.get(http_request, PATH)
http_request.assert_called_once_with(EXPECTED_URL, **EXPECTED_KWARGS)
def test_get_failure(self):
http_request = request_mock(http_client.NOT_FOUND, 'text/html',
'<p>Error</p>')
with self.assertRaises(http_client.HTTPException):
_metadata.get(http_request, PATH)
http_request.assert_called_once_with(EXPECTED_URL, **EXPECTED_KWARGS)
def test_get_failure(self):
http = request_mock(http_client.NOT_FOUND, 'text/html', '<p>Error</p>')
with self.assertRaises(http_client.HTTPException):
_metadata.get(http, PATH)
# Verify mocks.
self.assertEqual(http.requests, 1)
self.assertEqual(http.uri, EXPECTED_URL)
self.assertEqual(http.method, 'GET')
self.assertIsNone(http.body)
self.assertEqual(http.headers, _metadata.METADATA_HEADERS)
def test_get_failure(self):
http = request_mock(http_client.NOT_FOUND, "text/html", "<p>Error</p>")
with self.assertRaises(http_client.HTTPException):
_metadata.get(http, PATH)
# Verify mocks.
self.assertEqual(http.requests, 1)
self.assertEqual(http.uri, EXPECTED_URL)
self.assertEqual(http.method, "GET")
self.assertIsNone(http.body)
self.assertEqual(http.headers, _metadata.METADATA_HEADERS)
def test_custom_metadata_root_from_env(self):
headers = {'content-type': 'application/json'}
http = http_mock.HttpMock(headers=headers, data='{}')
fake_metadata_root = 'another.metadata.service'
os.environ['GCE_METADATA_ROOT'] = fake_metadata_root
reload_module(_metadata)
try:
_metadata.get(http, '')
finally:
del os.environ['GCE_METADATA_ROOT']
reload_module(_metadata)
# Verify mock.
self.assertEqual(http.requests, 1)
expected_uri = 'http://{}/computeMetadata/v1/'.format(fake_metadata_root)
self.assertEqual(http.uri, expected_uri)
def test_get_success_string(self):
http_request = request_mock(
http_client.OK, 'text/html', '<p>Hello World!</p>')
self.assertEqual(
_metadata.get(http_request, PATH),
'<p>Hello World!</p>'
)
http_request.assert_called_once_with(EXPECTED_URL, **EXPECTED_KWARGS)
def test_get_success_json(self):
http_request = request_mock(
http_client.OK, 'application/json', json.dumps(DATA))
self.assertEqual(
_metadata.get(http_request, PATH),
DATA
)
http_request.assert_called_once_with(EXPECTED_URL, **EXPECTED_KWARGS)
def test_get_success_string(self):
http = request_mock(http_client.OK, 'text/html', '<p>Hello World!</p>')
self.assertEqual(_metadata.get(http, PATH), '<p>Hello World!</p>')
# Verify mocks.
self.assertEqual(http.requests, 1)
self.assertEqual(http.uri, EXPECTED_URL)
self.assertEqual(http.method, 'GET')
self.assertIsNone(http.body)
self.assertEqual(http.headers, _metadata.METADATA_HEADERS)
def test_get_success_string(self):
http = request_mock(http_client.OK, "text/html", "<p>Hello World!</p>")
self.assertEqual(_metadata.get(http, PATH), "<p>Hello World!</p>")
# Verify mocks.
self.assertEqual(http.requests, 1)
self.assertEqual(http.uri, EXPECTED_URL)
self.assertEqual(http.method, "GET")
self.assertIsNone(http.body)
self.assertEqual(http.headers, _metadata.METADATA_HEADERS)
def test_get_success_json(self):
http = request_mock(http_client.OK, "application/json", json.dumps(DATA))
self.assertEqual(_metadata.get(http, PATH), DATA)
# Verify mocks.
self.assertEqual(http.requests, 1)
self.assertEqual(http.uri, EXPECTED_URL)
self.assertEqual(http.method, "GET")
self.assertIsNone(http.body)
self.assertEqual(http.headers, _metadata.METADATA_HEADERS)
def test_get_success_json(self):
http = request_mock(http_client.OK, 'application/json',
json.dumps(DATA))
self.assertEqual(_metadata.get(http, PATH), DATA)
# Verify mocks.
self.assertEqual(http.requests, 1)
self.assertEqual(http.uri, EXPECTED_URL)
self.assertEqual(http.method, 'GET')
self.assertIsNone(http.body)
self.assertEqual(http.headers, _metadata.METADATA_HEADERS)
def test_get_success_string(self):
http_request = request_mock(http_client.OK, 'text/html',
'<p>Hello World!</p>')
self.assertEqual(_metadata.get(http_request, PATH),
'<p>Hello World!</p>')
http_request.assert_called_once_with(EXPECTED_URL, **EXPECTED_KWARGS)
def test_get_success_json(self):
http_request = request_mock(http_client.OK, 'application/json',
json.dumps(DATA))
self.assertEqual(_metadata.get(http_request, PATH), DATA)
http_request.assert_called_once_with(EXPECTED_URL, **EXPECTED_KWARGS)
