def test_failure(self):
http_request = mock.MagicMock()
response = httplib2.Response({'status': http_client.NOT_FOUND})
content = b'Not found'
http_request.return_value = (response, content)
with self.assertRaises(MetadataServerHttpError):
_get_metadata(http_request)
http_request.assert_called_once_with(
_METADATA_ROOT + '/?recursive=true',
headers={'Metadata-Flavor': 'Google'})
def test_success_not_json(self):
http_request = mock.MagicMock()
data = '12345'.encode('utf-8')
http_request.return_value = (
httplib2.Response({'status': http_client.OK,
'content-type': 'text/plain'}), data)
result = _get_metadata(http_request, recursive=False)
self.assertEqual(result, data.decode('utf-8'))
http_request.assert_called_once_with(
_METADATA_ROOT,
headers={'Metadata-Flavor': 'Google'})
def test_success(self):
http_request = mock.MagicMock()
data = json.dumps(A_SERVICE_ACCOUNT).encode('utf-8')
http_request.return_value = (
httplib2.Response({'status': http_client.OK,
'content-type': 'application/json'}), data)
result = _get_metadata(http_request)
self.assertEqual(result, json.loads(data.decode('utf-8')))
http_request.assert_called_once_with(
_METADATA_ROOT + '/?recursive=true',
headers={'Metadata-Flavor': 'Google'})