RHOAIENG-39075: Add new E2E test structure and initial existing cluster test

Author: kryanbeane

src/codeflare_sdk/ray/cluster/config.py

@@ -98,9 +98,9 @@ class ClusterConfiguration:
 
     name: str
     namespace: Optional[str] = None
-    head_cpu_requests: Union[int, str] = 2
+    head_cpu_requests: Union[int, str] = 1
     head_cpu_limits: Union[int, str] = 2
-    head_memory_requests: Union[int, str] = 8
+    head_memory_requests: Union[int, str] = 5
     head_memory_limits: Union[int, str] = 8
     head_extended_resource_requests: Dict[str, Union[str, int]] = field(
         default_factory=dict
@@ -109,8 +109,8 @@ class ClusterConfiguration:
     worker_cpu_requests: Union[int, str] = 1
     worker_cpu_limits: Union[int, str] = 1
     num_workers: int = 1
-    worker_memory_requests: Union[int, str] = 2
-    worker_memory_limits: Union[int, str] = 2
+    worker_memory_requests: Union[int, str] = 3
+    worker_memory_limits: Union[int, str] = 6
     worker_tolerations: Optional[List[V1Toleration]] = None
     appwrapper: bool = False
     envs: Dict[str, str] = field(default_factory=dict)

src/codeflare_sdk/ray/cluster/test_pretty_print.py

@@ -85,15 +85,15 @@ def test_ray_details(mocker, capsys):
         name="raytest1",
         status=RayClusterStatus.READY,
         num_workers=1,
-        worker_mem_requests="2G",
-        worker_mem_limits="2G",
+        worker_mem_requests="3G",
+        worker_mem_limits="6G",
         worker_cpu_requests=1,
         worker_cpu_limits=1,
         namespace="ns",
         dashboard="fake-uri",
-        head_cpu_requests=2,
+        head_cpu_requests=1,
         head_cpu_limits=2,
-        head_mem_requests=8,
+        head_mem_requests=5,
         head_mem_limits=8,
     )
     mocker.patch(
@@ -150,7 +150,7 @@ def test_ray_details(mocker, capsys):
         " │   ╭── Workers ──╮  ╭───────── Worker specs(each) ─────────╮   │ \n"
         " │   │  # Workers  │  │  Memory      CPU         GPU         │   │ \n"
         " │   │             │  │                                      │   │ \n"
-        " │   │  1          │  │  2G~2G       1~1         0           │   │ \n"
+        " │   │  1          │  │  3G~6G       1~1         0           │   │ \n"
         " │   │             │  │                                      │   │ \n"
         " │   ╰─────────────╯  ╰──────────────────────────────────────╯   │ \n"
         " ╰───────────────────────────────────────────────────────────────╯ \n"
@@ -168,7 +168,7 @@ def test_ray_details(mocker, capsys):
         " │   ╭── Workers ──╮  ╭───────── Worker specs(each) ─────────╮   │ \n"
         " │   │  # Workers  │  │  Memory      CPU         GPU         │   │ \n"
         " │   │             │  │                                      │   │ \n"
-        " │   │  1          │  │  2G~2G       1~1         0           │   │ \n"
+        " │   │  1          │  │  3G~6G       1~1         0           │   │ \n"
         " │   │             │  │                                      │   │ \n"
         " │   ╰─────────────╯  ╰──────────────────────────────────────╯   │ \n"
         " ╰───────────────────────────────────────────────────────────────╯ \n"
@@ -184,7 +184,7 @@ def test_ray_details(mocker, capsys):
         "│   ╭── Workers ──╮  ╭───────── Worker specs(each) ─────────╮   │\n"
         "│   │  # Workers  │  │  Memory      CPU         GPU         │   │\n"
         "│   │             │  │                                      │   │\n"
-        "│   │  1          │  │  2G~2G       1~1         0           │   │\n"
+        "│   │  1          │  │  3G~6G       1~1         0           │   │\n"
         "│   │             │  │                                      │   │\n"
         "│   ╰─────────────╯  ╰──────────────────────────────────────╯   │\n"
         "╰───────────────────────────────────────────────────────────────╯\n"

src/codeflare_sdk/ray/rayjobs/rayjob.py

@@ -478,6 +478,9 @@ def _validate_priority_class(self):
         Raises ValueError if the priority class is definitively known not to exist.
         If we cannot verify (e.g., permission denied), logs a warning and allows submission.
         """
+        if self._cluster_config is None:
+            return
+
         if self.priority_class:
             logger.debug(f"Validating priority class '{self.priority_class}'...")
             exists = priority_class_exists(self.priority_class)

tests/e2e_v2/__init__.py

@@ -0,0 +1,2 @@
+# E2E Test Suite v2
+# Restructured pytest-based E2E tests for CodeFlare SDK