update unit tests

Author: Fiona-Waters

Diff for: src/codeflare_sdk/utils/generate_yaml.py

@@ -309,6 +309,7 @@ def get_default_kueue_name(namespace: str):
 
 
 def local_queue_exists(namespace, local_queue_name: str):
+    # get all local queues in the namespace
     try:
         config_check()
         api_instance = client.CustomObjectsApi(api_config_handler())
@@ -320,6 +321,7 @@ def local_queue_exists(namespace, local_queue_name: str):
         )
     except Exception as e:  # pragma: no cover
         return _kube_api_error_handling(e)
+    # check if local queue with the name provided in cluster config exists
     for lq in local_queues["items"]:
         if lq["metadata"]["name"] == local_queue_name:
             return True

Diff for: tests/unit_test.py

@@ -344,6 +344,10 @@ def test_cluster_creation_no_mcad_local_queue(mocker):
         "kubernetes.client.CustomObjectsApi.get_cluster_custom_object",
         return_value={"spec": {"domain": "apps.cluster.awsroute.org"}},
     )
+    mocker.patch(
+        "kubernetes.client.CustomObjectsApi.list_namespaced_custom_object",
+        return_value=get_local_queue("kueue.x-k8s.io", "v1beta1", "ns", "localqueues"),
+    )
     config = createClusterConfig()
     config.name = "unit-test-cluster-ray"
     config.mcad = False
@@ -3015,6 +3019,10 @@ def test_cluster_throw_for_no_raycluster(mocker: MockerFixture):
         "codeflare_sdk.utils.generate_yaml.get_default_kueue_name",
         return_value="default",
     )
+    mocker.patch(
+        "codeflare_sdk.utils.generate_yaml.local_queue_exists",
+        return_value="true",
+    )
 
     def throw_if_getting_raycluster(group, version, namespace, plural):
         if plural == "rayclusters":