Add RayCLuster Oauth Authentication test

Author: Srihari1192

tests/e2e/mnist_raycluster_sdk_auth.py

@@ -0,0 +1,80 @@
+import sys
+import os
+
+from time import sleep
+
+from torchx.specs.api import AppState, is_terminal
+
+from codeflare_sdk.cluster.cluster import Cluster, ClusterConfiguration
+from codeflare_sdk.job.jobs import DDPJobDefinition
+from codeflare_sdk.cluster.auth import TokenAuthentication
+
+namespace = sys.argv[1]
+ray_image = os.getenv("RAY_IMAGE")
+isAuthLoginEnable = os.getenv("OAUTH_LOGIN")
+cluster_token = os.getenv("CLUSTER_TOKEN")
+server_url = os.getenv("SERVER_URL")
+
+cluster = Cluster(
+    ClusterConfiguration(
+        name="mnist",
+        namespace=namespace,
+        num_workers=1,
+        head_cpus="500m",
+        head_memory=2,
+        min_cpus="500m",
+        max_cpus=1,
+        min_memory=1,
+        max_memory=2,
+        num_gpus=0,
+        instascale=False,
+        image=ray_image,
+        openshift_oauth=True,
+    )
+)
+
+if isAuthLoginEnable == "true":
+    auth = TokenAuthentication(token=cluster_token, server=server_url, skip_tls=True)
+    auth.login()
+
+cluster.up()
+
+cluster.status()
+
+cluster.wait_ready()
+
+cluster.status()
+
+cluster.details()
+
+jobdef = DDPJobDefinition(
+    name="mnist",
+    script="mnist.py",
+    scheduler_args={"requirements": "requirements.txt"},
+)
+job = jobdef.submit(cluster)
+
+done = False
+time = 0
+timeout = 900
+while not done:
+    status = job.status()
+    if is_terminal(status.state):
+        break
+    if not done:
+        print(status)
+        if timeout and time >= timeout:
+            raise TimeoutError(f"job has timed out after waiting {timeout}s")
+        sleep(5)
+        time += 5
+
+print(f"Job has completed: {status.state}")
+
+print(job.logs())
+
+cluster.down()
+
+if not status.state == AppState.SUCCEEDED:
+    exit(1)
+else:
+    exit(0)