Automation/Tests/Scripts/k8s_helper.py

##############################################################################
#
#   Copyright (c) 2019 AT&T Intellectual Property.
#
#   Licensed under the Apache License, Version 2.0 (the "License");
#   you may not use this file except in compliance with the License.
#   You may obtain a copy of the License at
#
#       http://www.apache.org/licenses/LICENSE-2.0
#
#   Unless required by applicable law or agreed to in writing, software
#   distributed under the License is distributed on an "AS IS" BASIS,
#   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
#   See the License for the specific language governing permissions and
#   limitations under the License.
#
##############################################################################
#
#   This source code is part of the near-RT RIC (RAN Intelligent Controller)
#   platform project (RICP).
#

import subprocess


def extract_service_ip(service_name):
    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 {} | awk \'{{print $1}}\'" \
        .format(pod_base_name)

    pod_name = subprocess.check_output(["/bin/bash", "-c", k8s_command], universal_newlines=True)

    return pod_name.strip()