raise error if raycluster crd not available

Signed-off-by: Kevin <[email protected]>

Author: KPostOffice

Diff for: src/codeflare_sdk/cluster/cluster.py

@@ -51,6 +51,7 @@
 import requests
 
 from kubernetes import config
+from kubernetes.client.rest import ApiException
 
 
 class Cluster:
@@ -225,6 +226,10 @@ def up(self):
         Applies the AppWrapper yaml, pushing the resource request onto
         the MCAD queue.
         """
+
+        # check if RayCluster CustomResourceDefinition exists if not throw RuntimeError
+        self._throw_for_no_raycluster()
+
         namespace = self.config.namespace
         if self.config.openshift_oauth:
             create_openshift_oauth_objects(
@@ -259,12 +264,32 @@ def up(self):
         except Exception as e:  # pragma: no cover
             return _kube_api_error_handling(e)
 
+    def _throw_for_no_raycluster(self):
+        api_instance = client.CustomObjectsApi(api_config_handler())
+        try:
+            api_instance.list_namespaced_custom_object(
+                group="ray.io",
+                version="v1",
+                namespace=self.config.namespace,
+                plural="rayclusters",
+            )
+        except ApiException as e:
+            if e.status == 404:
+                raise RuntimeError(
+                    "RayCluster CustomResourceDefinition unavailable contact your administrator."
+                )
+            else:
+                raise RuntimeError(
+                    "Failed to get RayCluster CustomResourceDefinition: " + str(e)
+                )
+
     def down(self):
         """
         Deletes the AppWrapper yaml, scaling-down and deleting all resources
         associated with the cluster.
         """
         namespace = self.config.namespace
+        self._throw_for_no_raycluster()
         try:
             config_check()
             api_instance = client.CustomObjectsApi(api_config_handler())

Diff for: tests/unit_test.py

@@ -481,6 +481,7 @@ def arg_check_del_effect(group, version, namespace, plural, name, *args):
 def test_cluster_up_down(mocker):
     mocker.patch("kubernetes.client.ApisApi.get_api_versions")
     mocker.patch("kubernetes.config.load_kube_config", return_value="ignore")
+    mocker.patch("codeflare_sdk.cluster.cluster.Cluster._throw_for_no_raycluster")
     mocker.patch(
         "kubernetes.client.CustomObjectsApi.get_cluster_custom_object",
         return_value={"spec": {"domain": ""}},
@@ -503,6 +504,7 @@ def test_cluster_up_down(mocker):
 
 
 def test_cluster_up_down_no_mcad(mocker):
+    mocker.patch("codeflare_sdk.cluster.cluster.Cluster._throw_for_no_raycluster")
     mocker.patch("kubernetes.client.ApisApi.get_api_versions")
     mocker.patch("kubernetes.config.load_kube_config", return_value="ignore")
     mocker.patch(
@@ -3273,6 +3275,29 @@ def test_gen_app_wrapper_with_oauth(mocker: MockerFixture):
     )
 
 
+def test_cluster_throw_for_no_raycluster(mocker: MockerFixture):
+    mocker.patch("kubernetes.client.ApisApi.get_api_versions")
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.get_current_namespace",
+        return_value="opendatahub",
+    )
+    mocker.patch(
+        "kubernetes.client.CustomObjectsApi.list_namespaced_custom_object",
+        side_effect=client.ApiException(status=404),
+    )
+    cluster = Cluster(
+        ClusterConfiguration(
+            "test_cluster",
+            openshift_oauth=True,
+            image="quay.io/project-codeflare/ray:latest-py39-cu118",
+            ingress_domain="apps.cluster.awsroute.org",
+            write_to_file=False,
+        )
+    )
+    with pytest.raises(RuntimeError):
+        cluster.up()
+
+
 """
 Ray Jobs tests
 """