Unit Test

gsutil-mirrors · Dec 8, 2021 · 00e616e · 00e616e
1 parent 6edde83
commit 00e616e
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/tests/contrib/test_gce.py b/tests/contrib/test_gce.py
@@ -173,3 +173,19 @@ def test_custom_metadata_root_from_env(self):
         self.assertEqual(http.requests, 1)
         expected_uri = 'http://{}/computeMetadata/v1/'.format(fake_metadata_root)
         self.assertEqual(http.uri, expected_uri)
+
+    def test_new_custom_metadata_host_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_HOST'] = fake_metadata_root
+        reload_module(_metadata)
+        try:
+            _metadata.get(http, '')
+        finally:
+            del os.environ['GCE_METADATA_HOST']
+            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)