1 files changed, 96 insertions, 0 deletions
diff --git a/functest_kubernetes/k8stest.py b/functest_kubernetes/k8stest.py
new file mode 100644
index 00000000..3be56d6c
--- /dev/null
+++ b/functest_kubernetes/k8stest.py
@@ -0,0 +1,96 @@
+#!/usr/bin/env python
+#
+# Copyright (c) 2018 All rights reserved
+# This program and the accompanying materials
+# are made available under the terms of the Apache License, Version 2.0
+# which accompanies this distribution, and is available at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+
+"""
+Define the parent for Kubernetes testing.
+"""
+
+from __future__ import division
+
+import logging
+import os
+import subprocess
+import time
+
+from functest.core import testcase
+
+
+LOGGER = logging.getLogger(__name__)
+
+
+class K8sTesting(testcase.TestCase):
+    """Kubernetes test runner"""
+
+    def __init__(self, **kwargs):
+        super(K8sTesting, self).__init__(**kwargs)
+        self.cmd = []
+        self.result = 0
+        self.start_time = 0
+        self.stop_time = 0
+
+    def run_kubetest(self):
+        """Run the test suites"""
+        cmd_line = self.cmd
+        LOGGER.info("Starting k8s test: '%s'.", cmd_line)
+
+        process = subprocess.Popen(cmd_line, stdout=subprocess.PIPE,
+                                   stderr=subprocess.STDOUT)
+        remark = []
+        lines = process.stdout.readlines()
+        for i in range(len(lines) - 1, -1, -1):
+            new_line = str(lines[i])
+
+            if 'SUCCESS!' in new_line or 'FAIL!' in new_line:
+                remark = new_line.replace('--', '|').split('|')
+                break
+
+        if remark and 'SUCCESS!' in remark[0]:
+            self.result = 100
+
+    def run(self):
+
+        if not os.path.isfile(os.getenv('KUBECONFIG')):
+            LOGGER.error("Cannot run k8s testcases. Config file not found ")
+            return self.EX_RUN_ERROR
+
+        self.start_time = time.time()
+        try:
+            self.run_kubetest()
+            res = self.EX_OK
+        except Exception as ex:  # pylint: disable=broad-except
+            LOGGER.error("Error with running %s", str(ex))
+            res = self.EX_RUN_ERROR
+
+        self.stop_time = time.time()
+        return res
+
+    def check_envs(self):  # pylint: disable=no-self-use
+        """Check if required environment variables are set"""
+        try:
+            assert 'DEPLOY_SCENARIO' in os.environ
+            assert 'KUBECONFIG' in os.environ
+            assert 'KUBE_MASTER' in os.environ
+            assert 'KUBE_MASTER_IP' in os.environ
+            assert 'KUBERNETES_PROVIDER' in os.environ
+            assert 'KUBE_MASTER_URL' in os.environ
+        except Exception as ex:
+            raise Exception("Cannot run k8s testcases. "
+                            "Please check env var: %s" % str(ex))
+
+
+class K8sSmokeTest(K8sTesting):
+    """Kubernetes smoke test suite"""
+    def __init__(self, **kwargs):
+        if "case_name" not in kwargs:
+            kwargs.get("case_name", 'k8s_smoke')
+        super(K8sSmokeTest, self).__init__(**kwargs)
+        self.check_envs()
+        self.cmd = ['/src/k8s.io/kubernetes/cluster/test-smoke.sh', '--host',
+                    os.getenv('KUBE_MASTER_URL')]