[RIC-571] Fix automation tests

[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 80b1829..a1cce5f 100644 (file)
--- a/Automation/Tests/Scripts/k8s_helper.py
+++ b/Automation/Tests/Scripts/k8s_helper.py
@@ -24,9 +24,17 @@ import subprocess
 
 
 def extract_service_ip(service_name):
-    k8s_command = "kubectl -n ricplt get services | /bin/grep {} | /bin/grep ClusterIP |  awk \'{{print $3}}\'"\
+    k8s_command = "kubectl -n ricplt get services | /bin/grep {} | /bin/grep ClusterIP |  awk \'{{print $3}}\'" \
         .format(service_name)
 
     service_ip = subprocess.check_output(["/bin/bash", "-c", k8s_command], universal_newlines=True)
 
     return service_ip.strip()
+
+def extract_pod_name(pod_base_name):
+    k8s_command = "kubectl get pods -n ricplt | /bin/grep {} | /bin/grep Running | 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