@@ -888,6 +888,10 @@ def test_ray_details(mocker, capsys):
888
888
"codeflare_sdk.cluster.cluster.Cluster.cluster_dashboard_uri" ,
889
889
return_value = "" ,
890
890
)
891
+ mocker .patch (
892
+ "codeflare_sdk.utils.generate_yaml.local_queue_exists" ,
893
+ return_value = "true" ,
894
+ )
891
895
cf = Cluster (
892
896
ClusterConfiguration (
893
897
name = "raytest2" ,
@@ -1933,6 +1937,10 @@ def test_get_cluster_openshift(mocker):
1933
1937
MagicMock (versions = [MagicMock (group_version = "route.openshift.io/v1" )])
1934
1938
]
1935
1939
mocker .patch ("kubernetes.client.ApisApi" , return_value = mock_api )
1940
+ mocker .patch (
1941
+ "codeflare_sdk.utils.generate_yaml.local_queue_exists" ,
1942
+ return_value = "true" ,
1943
+ )
1936
1944
1937
1945
assert is_openshift_cluster ()
1938
1946
@@ -1997,6 +2005,10 @@ def test_get_cluster(mocker):
1997
2005
"kubernetes.client.NetworkingV1Api.list_namespaced_ingress" ,
1998
2006
return_value = ingress_retrieval (cluster_name = "quicktest" , client_ing = True ),
1999
2007
)
2008
+ mocker .patch (
2009
+ "codeflare_sdk.utils.generate_yaml.local_queue_exists" ,
2010
+ return_value = "true" ,
2011
+ )
2000
2012
cluster = get_cluster ("quicktest" )
2001
2013
cluster_config = cluster .config
2002
2014
assert cluster_config .name == "quicktest" and cluster_config .namespace == "ns"
@@ -2025,6 +2037,10 @@ def test_get_cluster_no_mcad(mocker):
2025
2037
"kubernetes.client.NetworkingV1Api.list_namespaced_ingress" ,
2026
2038
return_value = ingress_retrieval (cluster_name = "quicktest" , client_ing = True ),
2027
2039
)
2040
+ mocker .patch (
2041
+ "codeflare_sdk.utils.generate_yaml.local_queue_exists" ,
2042
+ return_value = "true" ,
2043
+ )
2028
2044
cluster = get_cluster ("quicktest" )
2029
2045
cluster_config = cluster .config
2030
2046
assert cluster_config .name == "quicktest" and cluster_config .namespace == "ns"
@@ -2250,6 +2266,10 @@ def test_list_queue_rayclusters(mocker, capsys):
2250
2266
def test_cluster_status (mocker ):
2251
2267
mocker .patch ("kubernetes.client.ApisApi.get_api_versions" )
2252
2268
mocker .patch ("kubernetes.config.load_kube_config" , return_value = "ignore" )
2269
+ mocker .patch (
2270
+ "codeflare_sdk.utils.generate_yaml.local_queue_exists" ,
2271
+ return_value = "true" ,
2272
+ )
2253
2273
fake_aw = AppWrapper ("test" , AppWrapperStatus .FAILED )
2254
2274
fake_ray = RayCluster (
2255
2275
name = "test" ,
@@ -2341,6 +2361,10 @@ def test_wait_ready(mocker, capsys):
2341
2361
mocker .patch ("kubernetes.config.load_kube_config" , return_value = "ignore" )
2342
2362
mocker .patch ("codeflare_sdk.cluster.cluster._app_wrapper_status" , return_value = None )
2343
2363
mocker .patch ("codeflare_sdk.cluster.cluster._ray_cluster_status" , return_value = None )
2364
+ mocker .patch (
2365
+ "codeflare_sdk.utils.generate_yaml.local_queue_exists" ,
2366
+ return_value = "true" ,
2367
+ )
2344
2368
mocker .patch .object (
2345
2369
client .CustomObjectsApi ,
2346
2370
"list_namespaced_custom_object" ,
0 commit comments