3 # Copyright (c) 2018 All rights reserved
4 # This program and the accompanying materials
5 # are made available under the terms of the Apache License, Version 2.0
6 # which accompanies this distribution, and is available at
8 # http://www.apache.org/licenses/LICENSE-2.0
12 Define the parent for Kubernetes testing.
15 from __future__ import division
23 from xtesting.core import testcase
26 class K8sTesting(testcase.TestCase):
27 """Kubernetes test runner"""
29 __logger = logging.getLogger(__name__)
31 def __init__(self, **kwargs):
32 super(K8sTesting, self).__init__(**kwargs)
38 def run_kubetest(self): # pylint: disable=too-many-branches
39 """Run the test suites"""
41 self.__logger.info("Starting k8s test: '%s'.", cmd_line)
43 process = subprocess.Popen(cmd_line, stdout=subprocess.PIPE,
44 stderr=subprocess.STDOUT)
45 output = process.stdout.read()
46 # Remove color code escape sequences
47 output = re.sub(r'\x1B\[[0-?]*[ -/]*[@-~]', '', str(output))
50 lines = output.split('\n')
53 if '[k8s.io]' in lines[i]:
54 if i != 0 and 'seconds' in lines[i-1]:
55 self.__logger.debug(lines[i-1])
56 while lines[i] != '-'*len(lines[i]):
57 if lines[i].startswith('STEP:') or ('INFO:' in lines[i]):
59 self.__logger.debug(lines[i])
62 success = 'SUCCESS!' in lines[i]
63 failure = 'FAIL!' in lines[i]
64 if success or failure:
65 if i != 0 and 'seconds' in lines[i-1]:
66 remarks.append(lines[i-1])
67 remarks = remarks + lines[i].replace('--', '|').split('|')
71 self.__logger.debug('-'*10)
72 self.__logger.info("Remarks:")
73 for remark in remarks:
74 if 'seconds' in remark:
75 self.__logger.debug(remark)
76 elif 'Passed' in remark:
77 self.__logger.info("Passed: %s", remark.split()[0])
78 elif 'Skipped' in remark:
79 self.__logger.info("Skipped: %s", remark.split()[0])
80 elif 'Failed' in remark:
81 self.__logger.info("Failed: %s", remark.split()[0])
88 def run(self, **kwargs):
90 if not os.path.isfile(os.getenv('KUBECONFIG')):
92 "Cannot run k8s testcases. Config file not found ")
93 return self.EX_RUN_ERROR
95 self.start_time = time.time()
99 except Exception: # pylint: disable=broad-except
100 self.__logger.exception("Error with running kubetest:")
101 res = self.EX_RUN_ERROR
103 self.stop_time = time.time()
106 def check_envs(self): # pylint: disable=no-self-use
107 """Check if required environment variables are set"""
109 assert 'DEPLOY_SCENARIO' in os.environ
110 assert 'KUBECONFIG' in os.environ
111 assert 'KUBE_MASTER_IP' in os.environ
112 assert 'KUBERNETES_PROVIDER' in os.environ
113 assert 'KUBE_MASTER_URL' in os.environ
114 except Exception as ex:
115 raise Exception("Cannot run k8s testcases. "
116 "Please check env var: %s" % str(ex))
119 class K8sSmokeTest(K8sTesting):
120 """Kubernetes smoke test suite"""
121 def __init__(self, **kwargs):
122 if "case_name" not in kwargs:
123 kwargs.get("case_name", 'k8s_smoke')
124 super(K8sSmokeTest, self).__init__(**kwargs)
126 self.cmd = ['/src/k8s.io/kubernetes/cluster/test-smoke.sh', '--host',
127 os.getenv('KUBE_MASTER_URL')]
130 class K8sConformanceTest(K8sTesting):
131 """Kubernetes conformance test suite"""
132 def __init__(self, **kwargs):
133 if "case_name" not in kwargs:
134 kwargs.get("case_name", 'k8s_conformance')
135 super(K8sConformanceTest, self).__init__(**kwargs)
137 self.cmd = ['/src/k8s.io/kubernetes/_output/bin/e2e.test',
138 '--ginkgo.focus', 'Conformance']