Fix failing tests

caraml-dev · Nov 28, 2023 · ebcba3d · ebcba3d
1 parent 073064b
commit ebcba3d
Showing 1 changed file with 49 additions and 0 deletions.
diff --git a/python/sdk/test/model_test.py b/python/sdk/test/model_test.py
@@ -361,6 +361,13 @@ def test_deploy(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/1234",
+            body=json.dumps(ep1.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",
@@ -404,6 +411,13 @@ def test_deploy_upiv1(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/1234",
+            body=json.dumps(upi_ep.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",
@@ -447,6 +461,13 @@ def test_deploy_using_raw_deployment_mode(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/1234",
+            body=json.dumps(ep3.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",
@@ -490,6 +511,13 @@ def test_deploy_with_autoscaling_policy(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/1234",
+            body=json.dumps(ep4.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",
@@ -559,6 +587,13 @@ def test_deploy_default_env(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/1234",
+            body=json.dumps(ep1.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",
@@ -599,6 +634,13 @@ def test_redeploy_model(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/1234",
+            body=json.dumps(ep4.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",
@@ -645,6 +687,13 @@ def test_deploy_with_gpu(self, version):
             status=200,
             content_type="application/json",
         )
+        responses.add(
+            "GET",
+            "/v1/models/1/versions/1/endpoint/789",
+            body=json.dumps(ep5.to_dict()),
+            status=200,
+            content_type="application/json",
+        )
         responses.add(
             "GET",
             "/v1/models/1/versions/1/endpoint",