add handler tests

Author: VipulMascarenhas

tests/unitary/with_extras/aqua/test_deployment_handler.py

@@ -29,6 +29,8 @@ class TestDataset:
         "model_id": "ocid1.datasciencemodel.oc1.iad.<OCID>",
         "instance_shape": "VM.GPU.A10.1",
         "display_name": "test-deployment-name",
+        "freeform_tags": {"ftag1": "fvalue1", "ftag2": "fvalue2"},
+        "defined_tags": {"dtag1": "dvalue1", "dtag2": "dvalue2"},
     }
     inference_request = {
         "prompt": "What is 1+1?",
@@ -156,6 +158,8 @@ def test_post(self, mock_create):
             private_endpoint_id=None,
             container_image_uri=None,
             cmd_var=None,
+            freeform_tags=TestDataset.deployment_request["freeform_tags"],
+            defined_tags=TestDataset.deployment_request["defined_tags"],
         )
 
 

tests/unitary/with_extras/aqua/test_model_handler.py

@@ -132,10 +132,38 @@ def test_list(self, mock_list):
 
     @parameterized.expand(
         [
-            (None, None, False, None, None, None),
-            ("odsc-llm-fine-tuning", None, False, None, None, ["test.json"]),
-            (None, "test.gguf", True, None, ["*.json"], None),
-            (None, None, True, "iad.ocir.io/<namespace>/<image>:<tag>", ["*.json"], ["test.json"]),
+            (None, None, False, None, None, None, None, None),
+            (
+                "odsc-llm-fine-tuning",
+                None,
+                False,
+                None,
+                None,
+                ["test.json"],
+                None,
+                None,
+            ),
+            (None, "test.gguf", True, None, ["*.json"], None, None, None),
+            (
+                None,
+                None,
+                True,
+                "iad.ocir.io/<namespace>/<image>:<tag>",
+                ["*.json"],
+                ["test.json"],
+                None,
+                None,
+            ),
+            (
+                None,
+                None,
+                False,
+                None,
+                None,
+                None,
+                {"ftag1": "fvalue1"},
+                {"dtag1": "dvalue1"},
+            ),
         ],
     )
     @patch("notebook.base.handlers.APIHandler.finish")
@@ -148,6 +176,8 @@ def test_register(
         inference_container_uri,
         allow_patterns,
         ignore_patterns,
+        freeform_tags,
+        defined_tags,
         mock_register,
         mock_finish,
     ):
@@ -168,7 +198,9 @@ def test_register(
                 download_from_hf=download_from_hf,
                 inference_container_uri=inference_container_uri,
                 allow_patterns=allow_patterns,
-                ignore_patterns=ignore_patterns
+                ignore_patterns=ignore_patterns,
+                freeform_tags=freeform_tags,
+                defined_tags=defined_tags,
             )
         )
         result = self.model_handler.post()
@@ -183,7 +215,9 @@ def test_register(
             download_from_hf=download_from_hf,
             inference_container_uri=inference_container_uri,
             allow_patterns=allow_patterns,
-            ignore_patterns=ignore_patterns
+            ignore_patterns=ignore_patterns,
+            freeform_tags=freeform_tags,
+            defined_tags=defined_tags,
         )
         assert result["id"] == "test_id"
         assert result["inference_container"] == "odsc-tgi-serving"