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)