test: add unit tests for status, details, and list CLI commands

Author: carsonmh

tests/unit_test.py

@@ -236,6 +236,135 @@ def test_list_clusters_all_namespaces(mocker, capsys):
     )
 
 
+def test_raycluster_details_cli(mocker):
+    runner = CliRunner()
+    mocker.patch(
+        "kubernetes.client.CustomObjectsApi.list_namespaced_custom_object",
+        side_effect=get_ray_obj,
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.get_current_namespace",
+        return_value="ns",
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.Cluster.status",
+        return_value=(False, CodeFlareClusterStatus.UNKNOWN),
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.Cluster.cluster_dashboard_uri",
+        return_value="",
+    )
+    mocker.patch.object(client, "ApiClient")
+    raycluster_details_command = """
+                                details raycluster quicktest
+                                """
+    result = runner.invoke(cli, raycluster_details_command)
+    quicktest_details = (
+        " ╭──────────────────────────────────────────────────────────────╮ \n"
+        + " │   Name                                                       │ \n"
+        + " │   quicktest                                   Inactive ❌    │ \n"
+        + " │                                                              │ \n"
+        + " │   URI: ray://quicktest-head-svc.ns.svc:10001                 │ \n"
+        + " │                                                              │ \n"
+        + " │   Dashboard🔗                                                │ \n"
+        + " │                                                              │ \n"
+        + " │                      Cluster Resources                       │ \n"
+        + " │   ╭─ Workers ──╮  ╭───────── Worker specs(each) ─────────╮   │ \n"
+        + " │   │  Min  Max  │  │  Memory      CPU         GPU         │   │ \n"
+        + " │   │            │  │                                      │   │ \n"
+        + " │   │  1    1    │  │  2~2         1           0           │   │ \n"
+        + " │   │            │  │                                      │   │ \n"
+        + " │   ╰────────────╯  ╰──────────────────────────────────────╯   │ \n"
+        + " ╰──────────────────────────────────────────────────────────────╯ "
+    )
+    assert quicktest_details in result.output
+
+
+def test_raycluster_status_cli(mocker):
+    runner = CliRunner()
+    test_raycluster = RayCluster(
+        "quicktest",
+        RayClusterStatus.READY,
+        1,
+        1,
+        "1",
+        "1",
+        1,
+        1,
+        "default",
+        "dashboard-url",
+    )
+    mocker.patch(
+        "kubernetes.client.CustomObjectsApi.list_namespaced_custom_object",
+        side_effect=get_ray_obj,
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.get_current_namespace",
+        return_value="ns",
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.Cluster.cluster_dashboard_uri",
+        return_value="",
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster._app_wrapper_status",
+        return_value=test_raycluster,
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster._ray_cluster_status",
+        return_value=test_raycluster,
+    )
+    mocker.patch.object(client, "ApiClient")
+    raycluster_status_command = """
+                                status raycluster quicktest
+                                """
+    result = runner.invoke(cli, raycluster_status_command)
+    assert "Active" in result.output
+
+
+def test_raycluster_list_cli(mocker):
+    runner = CliRunner()
+    mocker.patch(
+        "kubernetes.client.CustomObjectsApi.list_namespaced_custom_object",
+        side_effect=get_ray_obj,
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.get_current_namespace",
+        return_value="ns",
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.Cluster.status",
+        return_value=(False, CodeFlareClusterStatus.UNKNOWN),
+    )
+    mocker.patch(
+        "codeflare_sdk.cluster.cluster.Cluster.cluster_dashboard_uri",
+        return_value="",
+    )
+    mocker.patch.object(client, "ApiClient")
+    list_rayclusters_command = """
+                                list rayclusters --namespace=ns
+                                """
+    result = runner.invoke(cli, list_rayclusters_command)
+    assert (
+        " ╭──────────────────────────────────────────────────────────────╮ \n"
+        + " │   Name                                                       │ \n"
+        + " │   quicktest                                   Active ✅      │ \n"
+        + " │                                                              │ \n"
+        + " │   URI: ray://quicktest-head-svc.ns.svc:10001                 │ \n"
+        + " │                                                              │ \n"
+        + " │   Dashboard🔗                                                │ \n"
+        + " │                                                              │ \n"
+        + " │                      Cluster Resources                       │ \n"
+        + " │   ╭─ Workers ──╮  ╭───────── Worker specs(each) ─────────╮   │ \n"
+        + " │   │  Min  Max  │  │  Memory      CPU         GPU         │   │ \n"
+        + " │   │            │  │                                      │   │ \n"
+        + " │   │  1    1    │  │  2G~2G       1           0           │   │ \n"
+        + " │   │            │  │                                      │   │ \n"
+        + " │   ╰────────────╯  ╰──────────────────────────────────────╯   │ \n"
+        + " ╰──────────────────────────────────────────────────────────────╯ "
+    ) in result.output
+
+
 # For mocking openshift client results
 fake_res = openshift.Result("fake")