[RIC-571] Add Automation tests + bug fix in Setup of existing eNB flow

[ric-plt/e2mgr.git] / Automation / Tests / Scripts / k8s_helper.py

diff --git a/Automation/Tests/Scripts/k8s_helper.py b/Automation/Tests/Scripts/k8s_helper.py
index 2a84943..4fe6dd8 100644 (file)
--- a/Automation/Tests/Scripts/k8s_helper.py
+++ b/Automation/Tests/Scripts/k8s_helper.py
@@ -29,4 +29,12 @@ def extract_service_ip(service_name):
 
     service_ip = subprocess.check_output(["/bin/bash", "-c", k8s_command], universal_newlines=True)
 
-    return service_ip.strip()
\ No newline at end of file
+    return service_ip.strip()
+
+def extract_pod_name(pod_base_name):
+    k8s_command = "kubectl get pods -n ricplt | /bin/grep {} | awk \'{{print $1}}\'" \
+        .format(pod_base_name)
+
+    pod_name = subprocess.check_output(["/bin/bash", "-c", k8s_command], universal_newlines=True)
+
+    return pod_name.strip()
\ No newline at end of file