Skip to content

Commit e27e16f

Browse files
committed
fix: address comments
1 parent 707d994 commit e27e16f

File tree

2 files changed

+8
-6
lines changed

2 files changed

+8
-6
lines changed

tests/conftest.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -285,7 +285,8 @@ def unprivileged_client(
285285
@pytest.fixture(scope="session")
286286
def dsci_resource(admin_client: DynamicClient) -> DSCInitialization:
287287
name = py_config["dsci_name"]
288-
for dsci in DSCInitialization.get(dyn_client=admin_client, name=name):
288+
dsci = DSCInitialization(client=admin_client, name=name)
289+
if dsci.exists:
289290
return dsci
290291

291292
raise ResourceNotFoundError(f"DSCI resource {name} not found")
@@ -294,7 +295,8 @@ def dsci_resource(admin_client: DynamicClient) -> DSCInitialization:
294295
@pytest.fixture(scope="session")
295296
def dsc_resource(admin_client: DynamicClient) -> DataScienceCluster:
296297
name = py_config["dsc_name"]
297-
for dsc in DataScienceCluster.get(dyn_client=admin_client, name=name):
298+
dsc = DataScienceCluster(client=admin_client, name=name)
299+
if dsc.exists:
298300
return dsc
299301

300302
raise ResourceNotFoundError(f"DSC resource {name} not found")

utilities/infra.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -826,13 +826,13 @@ def wait_for_isvc_pods(client: DynamicClient, isvc: InferenceService, runtime_na
826826
return get_pods_by_isvc_label(client=client, isvc=isvc, runtime_name=runtime_name)
827827

828828

829-
def assert_dsci_ready(dsci_resource: DSCInitialization) -> None:
829+
def verify_dsci_status_ready(dsci_resource: DSCInitialization) -> None:
830830
LOGGER.info(f"Verify DSCI {dsci_resource.name} are {dsci_resource.Status.READY}.")
831831
if dsci_resource.status != dsci_resource.Status.READY:
832832
raise ResourceNotReadyError(f"DSCI {dsci_resource.name} is not ready.\nStatus: {dsci_resource.instance.status}")
833833

834834

835-
def assert_dsc_ready(dsc_resource: DataScienceCluster) -> None:
835+
def verify_dsc_status_ready(dsc_resource: DataScienceCluster) -> None:
836836
LOGGER.info(f"Verify DSC {dsc_resource.name} are {dsc_resource.Status.READY}.")
837837
if dsc_resource.status != dsc_resource.Status.READY:
838838
raise ResourceNotReadyError(f"DSC {dsc_resource.name} is not ready.\nStatus: {dsc_resource.instance.status}")
@@ -873,8 +873,8 @@ def cluster_sanity(
873873
LOGGER.warning(f"Skipping RHOAI resource checks, got {skip_rhoai_check}")
874874

875875
else:
876-
assert_dsci_ready(dsci_resource=dsci_resource)
877-
assert_dsc_ready(dsc_resource=dsc_resource)
876+
verify_dsci_status_ready(dsci_resource=dsci_resource)
877+
verify_dsc_status_ready(dsc_resource=dsc_resource)
878878

879879
except (ResourceNotReadyError, NodeUnschedulableError, NodeNotReadyError) as ex:
880880
error_msg = f"Cluster sanity check failed: {str(ex)}"

0 commit comments

Comments
 (0)