def create_storage(api, sc_manifest, pvc_manifest):
# type: (dict, dict)
"""Create a StorageClass and PersistentVolumeClaim for testing."""
s_api = get_storage_api_client()
s_api.create_storage_class(body=sc_manifest)
api.create_namespaced_persistent_volume_claim(body=pvc_manifest,
namespace='default')
def create_storage(api, sc_manifest, pvc_manifest):
# type: (dict, dict)
"""Create a StorageClass and PersistentVolumeClaim for testing."""
s_api = get_storage_api_client()
s_api.create_storage_class(
body=sc_manifest
)
api.create_namespaced_persistent_volume_claim(
body=pvc_manifest,
namespace='default')
def test_reset_env():
k8s_api_client = get_core_api_client()
k8s_storage_client = get_storage_api_client()
longhorn_api_client = get_longhorn_api_client()
pod_list = k8s_api_client.list_namespaced_pod("default")
for pod in pod_list.items:
if STRESS_POD_NAME_PREFIX in pod.metadata.name:
delete_and_wait_pod(k8s_api_client, pod.metadata.name)
pvc_list = \
k8s_api_client.list_namespaced_persistent_volume_claim("default")
for pvc in pvc_list.items:
if STRESS_PVC_NAME_PREFIX in pvc.metadata.name:
delete_and_wait_pvc(k8s_api_client, pvc.metadata.name)
pv_list = k8s_api_client.list_persistent_volume()
for pv in pv_list.items:
pv_name = pv.metadata.name
if STRESS_PV_NAME_PREFIX in pv_name:
try:
delete_and_wait_pv(k8s_api_client, pv_name)
except AssertionError:
volumeattachment_list = \
k8s_storage_client.list_volume_attachment()
for volumeattachment in volumeattachment_list.items:
volume_attachment_name = \
volumeattachment.spec.source.persistent_volume_name
if volume_attachment_name == pv_name:
delete_and_wait_volume_attachment(
k8s_storage_client,
volume_attachment_name
)
delete_and_wait_pv(k8s_api_client, pv.metadata.name)
volume_list = \
longhorn_api_client.list_volume()
for volume in volume_list.data:
if STRESS_VOLUME_NAME_PREFIX in volume.name:
delete_and_wait_longhorn(longhorn_api_client, volume.name)
def create_storage_class(sc_manifest):
api = get_storage_api_client()
api.create_storage_class(body=sc_manifest)