Skip to content

Commit a8f4dc2

Browse files
committed
Added is_openshift_cluster() unit test
1 parent 8e802b9 commit a8f4dc2

File tree

2 files changed

+16
-1
lines changed

2 files changed

+16
-1
lines changed

src/codeflare_sdk/utils/generate_yaml.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,7 @@ def gen_names(name):
4545

4646

4747
# Check if the ingress api cluster resource exists
48-
def is_openshift_cluster(): # pragma: no cover
48+
def is_openshift_cluster():
4949
try:
5050
config_check()
5151
api_instance = client.CustomObjectsApi(api_config_handler())

tests/unit_test.py

+15
Original file line numberDiff line numberDiff line change
@@ -2337,6 +2337,21 @@ def secret_ca_retreival(secret_name, namespace):
23372337
return client.models.V1Secret(data=data)
23382338

23392339

2340+
def test_is_openshift_cluster(mocker):
2341+
mocker.patch("kubernetes.config.load_kube_config", return_value="ignore")
2342+
mocker.patch.object(
2343+
client.CustomObjectsApi,
2344+
"get_cluster_custom_object",
2345+
side_effect=client.ApiException(status=404),
2346+
)
2347+
assert is_openshift_cluster() == False
2348+
mocker.patch(
2349+
"kubernetes.client.CustomObjectsApi.get_cluster_custom_object",
2350+
return_value={"spec": {"domain": ""}},
2351+
)
2352+
assert is_openshift_cluster() == True
2353+
2354+
23402355
def test_generate_tls_cert(mocker):
23412356
"""
23422357
test the function codeflare_sdk.utils.generate_ca_cert generates the correct outputs

0 commit comments

Comments
 (0)