@@ -86,15 +86,15 @@ def check_default_storageclass(k8s_client_storage):
8686 return default_storage_class
8787
8888
89- def create_pvc_pod (k8s_api_instance , storage_class ):
89+ def create_pvc_pod (k8s_api_instance , storage_class , pvc_name = PVC_NAME , pod_name = POD_NAME , pv_name = PV_NAME , namespace = NAMESPACE , num_retries = NUM_RETRIES ):
9090 """
9191 1. Create PersistantVolumeClaim
9292 2. Create pod which uses the PersitantVolumeClaim
9393 """
9494 # 1. Create PersistantVolumeClaim
95- logger .debug (f"create pvc: { PVC_NAME } " )
95+ logger .debug (f"create pvc: { pvc_name } " )
9696
97- pvc_meta = client .V1ObjectMeta (name = PVC_NAME )
97+ pvc_meta = client .V1ObjectMeta (name = pvc_name )
9898 pvc_resources = client .V1ResourceRequirements (
9999 requests = {"storage" : "1Gi" },
100100 )
@@ -112,11 +112,11 @@ def create_pvc_pod(k8s_api_instance, storage_class):
112112 )
113113
114114 # 2. Create a pod which makes use of the PersitantVolumeClaim
115- logger .debug (f"create pod: { POD_NAME } " )
115+ logger .debug (f"create pod: { pod_name } " )
116116
117117 pod_vol = client .V1Volume (
118118 name = PV_NAME ,
119- persistent_volume_claim = client .V1PersistentVolumeClaimVolumeSource (PVC_NAME ),
119+ persistent_volume_claim = client .V1PersistentVolumeClaimVolumeSource (pvc_name ),
120120 )
121121 pod_con = client .V1Container (
122122 name = "nginx" ,
@@ -130,7 +130,7 @@ def create_pvc_pod(k8s_api_instance, storage_class):
130130 pod_body = client .V1Pod (
131131 api_version = "v1" ,
132132 kind = "Pod" ,
133- metadata = client .V1ObjectMeta (name = POD_NAME ),
133+ metadata = client .V1ObjectMeta (name = pod_name ),
134134 spec = pod_spec ,
135135 )
136136
@@ -141,7 +141,7 @@ def create_pvc_pod(k8s_api_instance, storage_class):
141141 pod_status = pod_info ["status" ]["phase" ]
142142
143143 retries = 0
144- while pod_status != "Running" and retries <= NUM_RETRIES :
144+ while pod_status != "Running" and retries <= num_retries :
145145 api_response = k8s_api_instance .read_namespaced_pod (
146146 POD_NAME , NAMESPACE , _preload_content = False
147147 )
0 commit comments