Fix client test setup bug (#275)

* skip client tests * re-add client tests, fix marqo cloud URL
marqo-ai · Feb 10, 2025 · 692ee88 · 692ee88
1 parent 42fc95a
commit 692ee88
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/v2_tests/test_client.py b/tests/v2_tests/test_client.py
@@ -68,8 +68,12 @@ def test_non_cloud_endpoint_client_is_not_marqo_cloud(self):
 
 @mark.fixed
 class TestClientMethods(MarqoTestCase):
+    def setUp(self) -> None:
+        self.initial_marqo_cloud_url = os.environ.get("MARQO_CLOUD_URL", constants.CLOUD_AWS_API_ENDPOINT)
+
     def tearDown(self) -> None:
-        os.environ["MARQO_CLOUD_URL"] = constants.CLOUD_AWS_API_ENDPOINT
+        os.environ["MARQO_CLOUD_URL"] = self.initial_marqo_cloud_url
+
     def test_is_cloud_api_endpoint(self):
         # Custom Marqo Cloud URL
         os.environ["MARQO_CLOUD_URL"] = "https://arandomsite.ai"