Refactor RallyBase

This provides a more generic way to integrate test case lists which are
not hosted in Functest.

Also removes vm scenarios since they have never been a part of actual
rally runs.

Change-Id: Ib0a020fe72800915bbf2d10ecc690a248d33c246
Signed-off-by: Juha Kosonen <juha.kosonen@nokia.com>

3 files changed, 37 insertions, 90 deletions

diff --git a/functest/opnfv_tests/openstack/rally/rally.py b/functest/opnfv_tests/openstack/rally/rally.py
index afaed4c65..a91059a2b 100644
--- a/functest/opnfv_tests/openstack/rally/rally.py
+++ b/functest/opnfv_tests/openstack/rally/rally.py
@@ -38,7 +38,7 @@ class RallyBase(singlevm.VmReady2):
 
     # pylint: disable=too-many-instance-attributes
     TESTS = ['authenticate', 'glance', 'cinder', 'gnocchi', 'heat',
-             'keystone', 'neutron', 'nova', 'quotas', 'vm', 'all']
+             'keystone', 'neutron', 'nova', 'quotas']
 
     RALLY_DIR = pkg_resources.resource_filename(
         'functest', 'opnfv_tests/openstack/rally')
@@ -75,7 +75,6 @@ class RallyBase(singlevm.VmReady2):
             project=self.project.project.id,
             domain=self.project.domain.id)
         self.creators = []
-        self.mode = ''
         self.summary = []
         self.scenario_dir = ''
         self.smoke = None
@@ -85,6 +84,9 @@ class RallyBase(singlevm.VmReady2):
         self.details = None
         self.compute_cnt = 0
         self.flavor_alt = None
+        self.tests = []
+        self.task_file = ''
+        self.run_cmd = ''
 
     def _build_task_args(self, test_file_name):
         """Build arguments for the Rally task."""
@@ -328,25 +330,11 @@ class RallyBase(singlevm.VmReady2):
         else:
             LOGGER.info('Test scenario: "%s" Failed.', test_name)
 
-    def _run_task(self, test_name):
+    def run_task(self, test_name):
         """Run a task."""
         LOGGER.info('Starting test scenario "%s" ...', test_name)
-
-        task_file = os.path.join(self.RALLY_DIR, 'task.yaml')
-        if not os.path.exists(task_file):
-            LOGGER.error("Task file '%s' does not exist.", task_file)
-            raise Exception("Task file '{}' does not exist.".format(task_file))
-
-        file_name = self._prepare_test_list(test_name)
-        if self.file_is_empty(file_name):
-            LOGGER.info('No tests for scenario "%s"', test_name)
-            return
-
-        cmd = (["rally", "task", "start", "--abort-on-sla-failure", "--task",
-                task_file, "--task-args",
-                str(self._build_task_args(test_name))])
-        LOGGER.debug('running command: %s', cmd)
-        proc = subprocess.Popen(cmd, stdout=subprocess.PIPE,
+        LOGGER.debug('running command: %s', self.run_cmd)
+        proc = subprocess.Popen(self.run_cmd, stdout=subprocess.PIPE,
                                 stderr=subprocess.STDOUT)
         output = proc.communicate()[0]
 
@@ -356,7 +344,6 @@ class RallyBase(singlevm.VmReady2):
             LOGGER.error("Failed to retrieve task_id")
             LOGGER.error("Result:\n%s", output)
             raise Exception("Failed to retrieve task id")
-
         self._save_results(test_name, task_id)
 
     def _append_summary(self, json_raw, test_name):
@@ -386,26 +373,43 @@ class RallyBase(singlevm.VmReady2):
                             'task_status': self.task_succeed(json_raw)}
         self.summary.append(scenario_summary)
 
-    def _prepare_env(self):
-        """Create resources needed by test scenarios."""
+    def prepare_run(self):
+        """Prepare resources needed by test scenarios."""
         assert self.cloud
         LOGGER.debug('Validating the test name...')
-        if self.test_name not in self.TESTS:
+        if self.test_name == 'all':
+            self.tests = self.TESTS
+        elif self.test_name in self.TESTS:
+            self.tests = [self.test_name]
+        else:
             raise Exception("Test name '%s' is invalid" % self.test_name)
 
+        self.task_file = os.path.join(self.RALLY_DIR, 'task.yaml')
+        if not os.path.exists(self.task_file):
+            LOGGER.error("Task file '%s' does not exist.", self.task_file)
+            raise Exception("Task file '{}' does not exist.".
+                            format(self.task_file))
+
         self.compute_cnt = len(self.cloud.list_hypervisors())
         self.flavor_alt = self.create_flavor_alt()
         LOGGER.debug("flavor: %s", self.flavor_alt)
 
-    def _run_tests(self):
+    def prepare_task(self, test_name):
+        """Prepare resources for test run."""
+        file_name = self._prepare_test_list(test_name)
+        if self.file_is_empty(file_name):
+            LOGGER.info('No tests for scenario "%s"', test_name)
+            return False
+        self.run_cmd = (["rally", "task", "start", "--abort-on-sla-failure",
+                         "--task", self.task_file, "--task-args",
+                         str(self._build_task_args(test_name))])
+        return True
+
+    def run_tests(self):
         """Execute tests."""
-        if self.test_name == 'all':
-            for test in self.TESTS:
-                if test == 'all' or test == 'vm':
-                    continue
-                self._run_task(test)
-        else:
-            self._run_task(self.test_name)
+        for test in self.tests:
+            if self.prepare_task(test):
+                self.run_task(test)
 
     def _generate_report(self):
         """Generate test execution summary report."""
@@ -492,8 +496,8 @@ class RallyBase(singlevm.VmReady2):
                 OS_PROJECT_ID=self.project.project.id,
                 OS_PASSWORD=self.project.password)
             conf_utils.create_rally_deployment(environ=environ)
-            self._prepare_env()
-            self._run_tests()
+            self.prepare_run()
+            self.run_tests()
             self._generate_report()
             res = testcase.TestCase.EX_OK
         except Exception as exc:   # pylint: disable=broad-except
@@ -512,7 +516,6 @@ class RallySanity(RallyBase):
         if "case_name" not in kwargs:
             kwargs["case_name"] = "rally_sanity"
         super(RallySanity, self).__init__(**kwargs)
-        self.mode = 'sanity'
         self.test_name = 'all'
         self.smoke = True
         self.scenario_dir = os.path.join(self.RALLY_SCENARIO_DIR, 'sanity')
@@ -526,7 +529,6 @@ class RallyFull(RallyBase):
         if "case_name" not in kwargs:
             kwargs["case_name"] = "rally_full"
         super(RallyFull, self).__init__(**kwargs)
-        self.mode = 'full'
         self.test_name = 'all'
         self.smoke = False
         self.scenario_dir = os.path.join(self.RALLY_SCENARIO_DIR, 'full')
diff --git a/functest/opnfv_tests/openstack/rally/scenario/opnfv-vm.yaml b/functest/opnfv_tests/openstack/rally/scenario/opnfv-vm.yaml
deleted file mode 100644
index 74f509925..000000000
--- a/functest/opnfv_tests/openstack/rally/scenario/opnfv-vm.yaml
+++ /dev/null
@@ -1,42 +0,0 @@
-  VMTasks.boot_runcommand_delete:
-    -
-      args:
-        {{ vm_params(image_name, flavor_name) }}
-        floating_network: {{ floating_network }}
-        force_delete: false
-        command:
-          interpreter: /bin/sh
-          script_file: {{ sup_dir }}/instance_dd_test.sh
-        username: cirros
-        nics:
-          - net-id: {{ netid }}
-      context:
-        {% call user_context(tenants_amount, users_amount, use_existing_users) %}
-        network: {}
-        {% endcall %}
-      runner:
-        {{ constant_runner(concurrency=concurrency, times=iterations, is_smoke=smoke) }}
-      sla:
-        {{ no_failures_sla() }}
-
-    -
-      args:
-        {{ vm_params(image_name, flavor_name) }}
-        fixed_network: private
-        floating_network: {{ floating_network }}
-        force_delete: false
-        command:
-          interpreter: /bin/sh
-          script_file: {{ sup_dir }}/instance_dd_test.sh
-        use_floatingip: true
-        username: cirros
-        nics:
-          - net-id: {{ netid }}
-        volume_args:
-          size: 2
-      context:
-        {{ user_context(tenants_amount, users_amount, use_existing_users) }}
-      runner:
-        {{ constant_runner(concurrency=concurrency, times=iterations, is_smoke=smoke) }}
-      sla:
-        {{ no_failures_sla() }}
diff --git a/functest/opnfv_tests/openstack/rally/scenario/support/instance_dd_test.sh b/functest/opnfv_tests/openstack/rally/scenario/support/instance_dd_test.sh
deleted file mode 100644
index e3bf23405..000000000
--- a/functest/opnfv_tests/openstack/rally/scenario/support/instance_dd_test.sh
+++ /dev/null
@@ -1,13 +0,0 @@
-#!/bin/sh
-time_seconds(){ (time -p $1 ) 2>&1 |awk '/real/{print $2}'; }
-file=/tmp/test.img
-c=${1:-$SIZE}
-c=${c:-1000} #default is 1GB
-write_seq=$(time_seconds "dd if=/dev/zero of=$file bs=1M count=$c")
-read_seq=$(time_seconds "dd if=$file of=/dev/null bs=1M count=$c")
-[ -f $file ] && rm $file
-
-echo "{
-    \"write_seq_${c}m\": $write_seq,
-    \"read_seq_${c}m\": $read_seq
-    }"