Danyil/marqo url resolver use endpoint field (#122)

* replace load_balancer_dns_name with endpoint * replace load_balancer_dns_name with endpoint in missed test
marqo-ai · Jul 14, 2023 · 9f1658d · 9f1658d
1 parent 81d702f
commit 9f1658d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/v0_tests/test_config.py b/tests/v0_tests/test_config.py
@@ -31,8 +31,8 @@ def test_get_url_when_cluster_is_marqo_and_no_index_name_specified(self):
     @mock.patch("requests.get")
     def test_get_url_when_cluster_is_marqo_and_index_name_specified(self, mock_get):
         mock_get.return_value.json.return_value = {"indices": [
-            {"index_name": "index1", "load_balancer_dns_name": "example.com", "index_status": "READY"},
-            {"index_name": "index2", "load_balancer_dns_name": "example2.com", "index_status": "READY"}
+            {"index_name": "index1", "endpoint": "example.com", "index_status": "READY"},
+            {"index_name": "index2", "endpoint": "example2.com", "index_status": "READY"}
         ]}
         c = config.Config(url="https://api.marqo.ai")
         print(c.marqo_url_resolver._urls_mapping)