6e2e722491
- Add get_volumesnapshots keywords - Fix k8s_table_parser_base.py substring postfix issue - Add unit test kubectl_get_volumesnapshots_table_parser_test.py Change-Id: I60784f722da5d87cee7efc86be6b2939a24e0bc2 Signed-off-by: ppeng <peng.peng@windriver.com>
71 lines
2.9 KiB
Python
71 lines
2.9 KiB
Python
import time
|
|
|
|
from framework.ssh.ssh_connection import SSHConnection
|
|
from keywords.base_keyword import BaseKeyword
|
|
from keywords.k8s.k8s_command_wrapper import export_k8s_config
|
|
from keywords.k8s.volumesnapshots.object.kubectl_get_volumesnapshots_output import KubectlGetVolumesnapshotsOutput
|
|
|
|
|
|
class KubectlGetVolumesnapshotsKeywords(BaseKeyword):
|
|
"""
|
|
Class for 'kubectl get volumesnapshots.snapshot.storage.k8s.io' keywords
|
|
"""
|
|
|
|
def __init__(self, ssh_connection: SSHConnection):
|
|
"""
|
|
Initialize the KubectlGetVolumesnapshotsKeywords class.
|
|
|
|
Args:
|
|
ssh_connection (SSHConnection): An SSH connection object to the target system.
|
|
"""
|
|
self.ssh_connection = ssh_connection
|
|
|
|
def get_volumesnapshots(self, namespace: str = None, label: str = None) -> KubectlGetVolumesnapshotsOutput:
|
|
"""
|
|
Gets the k8s volumesnapshots that are available using '-o wide'.
|
|
|
|
Args:
|
|
namespace(str, optional): The namespace to search for volumesnapshots. If None, it will search in all namespaces.
|
|
label (str, optional): The label to search for volumesnapshots.
|
|
|
|
Returns:
|
|
KubectlGetVolumesnapshotsOutput: An object containing the parsed output of the command.
|
|
|
|
"""
|
|
arg_namespace = ""
|
|
|
|
if namespace:
|
|
arg_namespace = f"-n {namespace}"
|
|
|
|
kubectl_get_volumesnapshots_output = self.ssh_connection.send(export_k8s_config(f"kubectl {arg_namespace} -o wide get volumesnapshots.snapshot.storage.k8s.io"))
|
|
self.validate_success_return_code(self.ssh_connection)
|
|
volumesnapshots_list_output = KubectlGetVolumesnapshotsOutput(kubectl_get_volumesnapshots_output)
|
|
|
|
return volumesnapshots_list_output
|
|
|
|
def wait_for_volumesnapshot_status(self, volumesnapshot_name: str, expected_status: str, namespace: str = None, timeout: int = 600) -> bool:
|
|
"""
|
|
Waits timeout amount of time for the given volumesnapshot to be in the given status
|
|
|
|
Args:
|
|
volumesnapshot_name (str): the volumesnapshot name
|
|
expected_status (str): the expected status
|
|
namespace (str): the namespace
|
|
timeout (int): the timeout in secs
|
|
|
|
Returns:
|
|
bool: True if the volumesnapshot is in the expected status
|
|
|
|
"""
|
|
volumesnapshot_status_timeout = time.time() + timeout
|
|
|
|
while time.time() < volumesnapshot_status_timeout:
|
|
volumesnapshots_output = self.get_volumesnapshots()
|
|
if volumesnapshots_output:
|
|
volumesnapshot_status = self.get_volumesnapshots(namespace).get_volumesnapshot(volumesnapshot_name).get_ready_to_use()
|
|
if volumesnapshot_status == expected_status:
|
|
return True
|
|
time.sleep(5)
|
|
|
|
raise ValueError(f"volumesnapshot is not at expected status {expected_status}, after {timeout}s.")
|