test/vcpe/healthcheck-k8s.py

   1 #!/usr/bin/env python
   2
   3 import argparse
   4 import json
   5 import logging
   6 from subprocess import Popen,PIPE,STDOUT,check_output,CalledProcessError
   7 import sys
   8
   9 def parse_args():
  10     """
  11     Parse command line arguments
  12     :return: arguments
  13     """
  14     parser = argparse.ArgumentParser(description='Processing arguments for vCPE healthcheck',formatter_class=argparse.ArgumentDefaultsHelpFormatter)
  15     parser.add_argument('--namespace', default="onap",
  16                              metavar=('<namespace_name>'),
  17                              help='Namespace name for existing ONAP deployment')
  18     parser.add_argument('--environment', default="dev",
  19                              metavar=('<environment_name>'),
  20                              help='Environment name for existing ONAP deployment visible in <environment_name>-sdnc-sdnc-0 pod name')
  21     args = parser.parse_args()
  22     return args
  23
  24
  25 args = parse_args()
  26
  27 namespace = args.namespace
  28 environment = args.environment
  29
  30 # config section
  31 kube_cmd = 'kubectl -n {0} exec {1}-sdnc-sdnc-0 -c sdnc -- bash -c '.format(namespace, environment)
  32 curl_check_cmd = 'apk info -e curl'
  33 curl_install_cmd = 'sudo apk add curl'
  34 curl_cmd = 'curl -s -u admin:admin -X GET http://{0}:8183/restconf/config/ietf-interfaces:interfaces'
  35 endpoints = {
  36     "vGMUX": '10.0.101.21',
  37     "vBRG": '10.3.0.2'
  38 }
  39 # end of config section
  40
  41 # Install curl command in SDNC container
  42 try:
  43     check_output(kube_cmd.split() + [curl_check_cmd], stderr=STDOUT)
  44 except CalledProcessError:
  45     try:
  46         check_output(kube_cmd.split() + [curl_install_cmd], stderr=STDOUT)
  47     except CalledProcessError:
  48         print('Curl package installation failed, exiting.')
  49         sys.exit(1)
  50     else:
  51         print("Curl package was installed in SDNC container")
  52
  53 for ename,eip in endpoints.items():
  54     print('Checking {0} REST API from SDNC'.format(ename))
  55     p = Popen(kube_cmd.split() + [curl_cmd.format(eip)], stdout=PIPE, stderr=PIPE)
  56     (output, error) = p.communicate()
  57     if p.returncode:
  58         print(error)
  59         sys.exit(p.returncode)
  60     else:
  61         print(json.dumps(json.loads(output), indent=4))
  62     print('\n')
  63
  64 print('Checking SDNC DB for vBRG MAC address')
  65 kube_db_cmd = 'kubectl -n {0} exec {1}-mariadb-galera-mariadb-galera-0 -- bash -c'
  66 db_cmd = "mysql -uroot -psecretpassword sdnctl -e 'select * from DHCP_MAP'"
  67 p = Popen(kube_db_cmd.format(namespace, environment).split() + [db_cmd], stdout=PIPE)
  68 (output, error) = p.communicate()
  69 print(output)