2696f2e32a
- added tests for nginx chart version and chart used by nginx pod Change-Id: I8b57b8bc2aa11b6bb136a893fec8a77f7ea7b096 Signed-off-by: Thomas Sunil <sunil.thomas@windriver.com>
74 lines
2.1 KiB
Python
74 lines
2.1 KiB
Python
from typing import Union
|
|
|
|
from framework.exceptions.keyword_exception import KeywordException
|
|
from keywords.k8s.helm.object.kubectl_helm_object import KubectlHelmObject
|
|
|
|
|
|
class KubectlGetHelmOutput:
|
|
"""Parser for 'kubectl get helmcharts' command output."""
|
|
|
|
def __init__(self, kubectl_output: Union[str, list[str]]):
|
|
"""Initialize KubectlGetHelmOutput.
|
|
|
|
Args:
|
|
kubectl_output (Union[str, list[str]]): Raw kubectl get helmcharts output.
|
|
"""
|
|
self.helms = self._parse_output(kubectl_output)
|
|
|
|
def _parse_output(self, output: Union[str, list[str]]) -> list[KubectlHelmObject]:
|
|
"""Parse kubectl get helmcharts output into helm objects.
|
|
|
|
Args:
|
|
output (Union[str, list[str]]): Raw command output.
|
|
|
|
Returns:
|
|
list[KubectlHelmObject]: List of parsed helm objects.
|
|
"""
|
|
helms = []
|
|
content = "\n".join(output) if isinstance(output, list) else output
|
|
lines = content.strip().split("\n")
|
|
|
|
if len(lines) < 2:
|
|
return helms
|
|
|
|
# Skip header line
|
|
for line in lines[1:]:
|
|
if line.strip():
|
|
parts = line.split()
|
|
if len(parts) >= 3:
|
|
name = parts[0]
|
|
chart = parts[1]
|
|
version = parts[2]
|
|
|
|
helm = KubectlHelmObject(name)
|
|
helm.set_chart(chart)
|
|
helm.set_version(version)
|
|
helms.append(helm)
|
|
|
|
return helms
|
|
|
|
def get_helmcharts(self) -> list[KubectlHelmObject]:
|
|
"""Get all helms.
|
|
|
|
Returns:
|
|
list[KubectlHelmObject]: List of all helms.
|
|
"""
|
|
return self.helms
|
|
|
|
def get_helmchart(self, name: str) -> KubectlHelmObject:
|
|
"""Get specific helm by name.
|
|
|
|
Args:
|
|
name (str): Name of the helm.
|
|
|
|
Returns:
|
|
KubectlHelmObject: The helm object.
|
|
|
|
Raises:
|
|
KeywordException: If helm not found.
|
|
"""
|
|
for helm in self.helms:
|
|
if helm.get_name() == name:
|
|
return helm
|
|
raise KeywordException(f"Helm {name} not found")
|