1 files changed, 149 insertions, 0 deletions
diff --git a/functest/opnfv_tests/openstack/refstack_client/refstack_client.py b/functest/opnfv_tests/openstack/refstack_client/refstack_client.py
new file mode 100755
index 00000000..d388dcd7
--- /dev/null
+++ b/functest/opnfv_tests/openstack/refstack_client/refstack_client.py
@@ -0,0 +1,149 @@
+#!/usr/bin/env python
+
+# matthew.lijun@huawei.com wangwulin@huawei.com
+# 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
+import argparse
+import os
+import sys
+
+from functest.core import testcase_base
+from functest.opnfv_tests.openstack.tempest import conf_utils
+from functest.utils import openstack_utils
+from functest.utils.constants import CONST
+import functest.utils.functest_logger as ft_logger
+import functest.utils.functest_utils as ft_utils
+
+""" logging configuration """
+logger = ft_logger.Logger("refstack_defcore").getLogger()
+
+
+class RefstackClient(testcase_base.TestcaseBase):
+
+    def __init__(self):
+        super(RefstackClient, self).__init__()
+        self.FUNCTEST_TEST = CONST.dir_functest_test
+        self.CONF_PATH = CONST.refstack_tempest_conf_path
+        self.DEFCORE_LIST = CONST.refstack_defcore_list
+        self.confpath = os.path.join(self.FUNCTEST_TEST,
+                                     self.CONF_PATH)
+        self.defcorelist = os.path.join(self.FUNCTEST_TEST,
+                                        self.DEFCORE_LIST)
+        self.VERIFIER_ID = conf_utils.get_verifier_id()
+        self.VERIFIER_REPO_DIR = conf_utils.get_verifier_repo_dir(
+            self.VERIFIER_ID)
+        self.DEPLOYMENT_ID = conf_utils.get_verifier_deployment_id()
+        self.DEPLOYMENT_DIR = conf_utils.get_verifier_deployment_dir(
+            self.VERIFIER_ID, self.DEPLOYMENT_ID)
+
+    def source_venv(self):
+
+        cmd = ("cd {0};"
+               ". .venv/bin/activate;"
+               "cd -;".format(CONST.dir_refstack_client))
+        ft_utils.execute_command(cmd)
+
+    def run_defcore(self, conf, testlist):
+        logger.debug("Generating test case list...")
+
+        cmd = ("cd {0};"
+               "./refstack-client test -c {1} -v --test-list {2};"
+               "cd -;".format(CONST.dir_refstack_client,
+                              conf,
+                              testlist))
+        ft_utils.execute_command(cmd)
+
+    def run_defcore_default(self):
+        logger.debug("Generating test case list...")
+
+        cmd = ("cd {0};"
+               "./refstack-client test -c {1} -v --test-list {2};"
+               "cd -;".format(CONST.dir_refstack_client,
+                              self.confpath,
+                              self.defcorelist))
+        ft_utils.execute_command(cmd)
+
+    def defcore_env_prepare(self):
+        try:
+            img_flavor_dict = conf_utils.create_tempest_resources(
+                use_custom_images=True, use_custom_flavors=True)
+            conf_utils.configure_tempest_defcore(
+                self.DEPLOYMENT_DIR, img_flavor_dict)
+            self.source_venv()
+            res = testcase_base.TestcaseBase.EX_OK
+        except KeyError as e:
+            logger.error("defcore prepare env error with: %s", e)
+            res = testcase_base.TestcaseBase.EX_RUN_ERROR
+
+        return res
+
+    def run(self):
+        try:
+            self.defcore_env_prepare()
+            self.run_defcore_default()
+            res = testcase_base.TestcaseBase.EX_OK
+        except Exception as e:
+            logger.error('Error with run: %s', e)
+            res = testcase_base.TestcaseBase.EX_RUN_ERROR
+
+        return res
+
+    def main(self, **kwargs):
+        try:
+            tempestconf = kwargs['config']
+            testlist = kwargs['testlist']
+        except KeyError as e:
+            logger.error("Cannot run refstack client. Please check "
+                         "%s", e)
+            return self.EX_RUN_ERROR
+        try:
+            openstack_utils.source_credentials(CONST.openstack_creds)
+            self.defcore_env_prepare()
+            self.run_defcore(tempestconf, testlist)
+            res = testcase_base.TestcaseBase.EX_OK
+        except Exception as e:
+            logger.error('Error with run: %s', e)
+            res = testcase_base.TestcaseBase.EX_RUN_ERROR
+
+        return res
+
+
+class RefstackClientParser(testcase_base.TestcaseBase):
+
+    def __init__(self):
+        super(RefstackClientParser, self).__init__()
+        self.FUNCTEST_TEST = CONST.dir_functest_test
+        self.CONF_PATH = CONST.refstack_tempest_conf_path
+        self.DEFCORE_LIST = CONST.refstack_defcore_list
+        self.confpath = os.path.join(self.FUNCTEST_TEST,
+                                     self.CONF_PATH)
+        self.defcorelist = os.path.join(self.FUNCTEST_TEST,
+                                        self.DEFCORE_LIST)
+        self.parser = argparse.ArgumentParser()
+        self.parser.add_argument(
+            '-c', '--config',
+            help='the file path of tempest.conf',
+            default=self.confpath)
+        self.parser.add_argument(
+            '-t', '--testlist',
+            help='Specify the file path or URL of a test list text file. '
+                 'This test list will contain specific test cases that '
+                 'should be tested.',
+            default=self.defcorelist)
+
+    def parse_args(self, argv=[]):
+        return vars(self.parser.parse_args(argv))
+
+
+if __name__ == '__main__':
+    refstackclient = RefstackClient()
+    parser = RefstackClientParser()
+    args = parser.parse_args(sys.argv[1:])
+    try:
+        result = refstackclient.main(**args)
+        if result != testcase_base.TestcaseBase.EX_OK:
+            sys.exit(result)
+    except Exception:
+        sys.exit(testcase_base.TestcaseBase.EX_RUN_ERROR)