Add test coverage to validate the functionality of the get_cluster method

Author: Srihari1192

tests/e2e/mnist_raycluster_sdk_kind_test.py

@@ -2,7 +2,7 @@
 
 from time import sleep
 
-from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication
+from codeflare_sdk import Cluster, ClusterConfiguration, get_cluster
 from codeflare_sdk.ray.client import RayJobClient
 
 import pytest
@@ -44,8 +44,6 @@ def run_mnist_raycluster_sdk_kind(
                 num_workers=1,
                 head_cpu_requests="500m",
                 head_cpu_limits="500m",
-                head_memory_requests=2,
-                head_memory_limits=2,
                 worker_cpu_requests="500m",
                 worker_cpu_limits=1,
                 worker_memory_requests=1,
@@ -68,6 +66,10 @@ def run_mnist_raycluster_sdk_kind(
 
         self.assert_jobsubmit_withoutlogin_kind(cluster, accelerator, number_of_gpus)
 
+        self.assert_get_cluster_and_jobsubmit(
+            "mnist", self.namespace, accelerator, number_of_gpus
+        )
+
     # Assertions
 
     def assert_jobsubmit_withoutlogin_kind(self, cluster, accelerator, number_of_gpus):
@@ -105,12 +107,47 @@ def assert_jobsubmit_withoutlogin_kind(self, cluster, accelerator, number_of_gpu
 
         client.delete_job(submission_id)
 
-        cluster.down()
-
     def assert_job_completion(self, status):
         if status == "SUCCEEDED":
             print(f"Job has completed: '{status}'")
             assert True
         else:
             print(f"Job has completed: '{status}'")
             assert False
+
+    def assert_get_cluster_and_jobsubmit(
+        self, cluster_name, namespace, accelerator, number_of_gpus
+    ):
+        # Retrieve the cluster
+        cluster = get_cluster(cluster_name, namespace)
+
+        cluster.details()
+
+        cluster.config.verify_tls = False
+
+        # Initialize the job client
+        client = cluster.job_client
+
+        # Submit a job and get the submission ID
+        submission_id = client.submit_job(
+            entrypoint="python mnist.py",
+            runtime_env={
+                "working_dir": "./tests/e2e/",
+                "pip": "./tests/e2e/mnist_pip_requirements.txt",
+                "env_vars": get_setup_env_variables(ACCELERATOR=accelerator),
+            },
+            entrypoint_num_gpus=number_of_gpus,
+        )
+        print(f"Submitted job with ID: {submission_id}")
+
+        # Fetch the list of jobs and validate
+        job_list = client.list_jobs()
+        print(f"List of Jobs: {job_list}")
+
+        # Validate the number of jobs in the list
+        assert len(job_list) == 1
+
+        # Validate the submission ID matches
+        assert job_list[0].submission_id == submission_id
+
+        cluster.down()

tests/e2e/mnist_raycluster_sdk_oauth_test.py

@@ -2,7 +2,12 @@
 
 from time import sleep
 
-from codeflare_sdk import Cluster, ClusterConfiguration, TokenAuthentication
+from codeflare_sdk import (
+    Cluster,
+    ClusterConfiguration,
+    TokenAuthentication,
+    get_cluster,
+)
 from codeflare_sdk.ray.client import RayJobClient
 
 import pytest
@@ -44,8 +49,6 @@ def run_mnist_raycluster_sdk_oauth(self):
                 num_workers=1,
                 head_cpu_requests="500m",
                 head_cpu_limits="500m",
-                head_memory_requests=4,
-                head_memory_limits=4,
                 worker_cpu_requests=1,
                 worker_cpu_limits=1,
                 worker_memory_requests=1,
@@ -68,6 +71,7 @@ def run_mnist_raycluster_sdk_oauth(self):
 
         self.assert_jobsubmit_withoutLogin(cluster)
         self.assert_jobsubmit_withlogin(cluster)
+        self.assert_get_cluster_and_jobsubmit("mnist", self.namespace)
 
     # Assertions
 
@@ -132,12 +136,45 @@ def assert_jobsubmit_withlogin(self, cluster):
 
         client.delete_job(submission_id)
 
-        cluster.down()
-
     def assert_job_completion(self, status):
         if status == "SUCCEEDED":
             print(f"Job has completed: '{status}'")
             assert True
         else:
             print(f"Job has completed: '{status}'")
             assert False
+
+    def assert_get_cluster_and_jobsubmit(self, cluster_name, namespace):
+        # Retrieve the cluster
+        cluster = get_cluster(cluster_name, namespace)
+
+        cluster.details()
+
+        cluster.config.verify_tls = False
+
+        # Initialize the job client
+        client = cluster.job_client
+
+        # Submit a job and get the submission ID
+        submission_id = client.submit_job(
+            entrypoint="python mnist.py",
+            runtime_env={
+                "working_dir": "./tests/e2e/",
+                "pip": "./tests/e2e/mnist_pip_requirements.txt",
+                "env_vars": get_setup_env_variables(),
+            },
+            entrypoint_num_cpus=1,
+        )
+        print(f"Submitted job with ID: {submission_id}")
+
+        # Fetch the list of jobs and validate
+        job_list = client.list_jobs()
+        print(f"List of Jobs: {job_list}")
+
+        # Validate the number of jobs in the list
+        assert len(job_list) == 1
+
+        # Validate the submission ID matches
+        assert job_list[0].submission_id == submission_id
+
+        cluster.down()