Remove duplicated code related to snaps creds
[functest.git] / functest / tests / unit / openstack / rally / test_rally.py
index c782861..92177f3 100644 (file)
@@ -5,6 +5,8 @@
 # which accompanies this distribution, and is available at
 # http://www.apache.org/licenses/LICENSE-2.0
 
+# pylint: disable=missing-docstring
+
 import json
 import logging
 import os
@@ -16,87 +18,67 @@ from functest.core import testcase
 from functest.opnfv_tests.openstack.rally import rally
 from functest.utils.constants import CONST
 
+from snaps.openstack.os_credentials import OSCreds
 
-class OSRallyTesting(unittest.TestCase):
-
-    logging.disable(logging.CRITICAL)
 
+class OSRallyTesting(unittest.TestCase):
     def setUp(self):
-        self.nova_client = mock.Mock()
-        self.neutron_client = mock.Mock()
-        self.cinder_client = mock.Mock()
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.get_nova_client',
-                        return_value=self.nova_client), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.get_neutron_client',
-                       return_value=self.neutron_client), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.get_cinder_client',
-                       return_value=self.cinder_client):
+        os_creds = OSCreds(
+            username='user', password='pass',
+            auth_url='http://foo.com:5000/v3', project_name='bar')
+        with mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                        'get_credentials', return_value=os_creds) as m:
             self.rally_base = rally.RallyBase()
-            self.rally_base.network_dict['net_id'] = 'test_net_id'
-            self.polling_iter = 2
+        self.assertTrue(m.called)
 
     def test_build_task_args_missing_floating_network(self):
-        CONST.OS_AUTH_URL = None
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.get_external_net',
-                        return_value=None):
-            task_args = self.rally_base._build_task_args('test_file_name')
-            self.assertEqual(task_args['floating_network'], '')
+        CONST.__setattr__('OS_AUTH_URL', None)
+        self.rally_base.ext_net_name = ''
+        task_args = self.rally_base._build_task_args('test_file_name')
+        self.assertEqual(task_args['floating_network'], '')
 
     def test_build_task_args_missing_net_id(self):
-        CONST.OS_AUTH_URL = None
-        self.rally_base.network_dict['net_id'] = ''
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.get_external_net',
-                        return_value='test_floating_network'):
-            task_args = self.rally_base._build_task_args('test_file_name')
-            self.assertEqual(task_args['netid'], '')
-
-    def test_build_task_args_missing_auth_url(self):
-        CONST.OS_AUTH_URL = None
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.get_external_net',
-                        return_value='test_floating_network'):
-            task_args = self.rally_base._build_task_args('test_file_name')
-            self.assertEqual(task_args['request_url'], '')
-
-    def check_scenario_file(self, value):
+        CONST.__setattr__('OS_AUTH_URL', None)
+        self.rally_base.priv_net_id = ''
+        task_args = self.rally_base._build_task_args('test_file_name')
+        self.assertEqual(task_args['netid'], '')
+
+    @staticmethod
+    def check_scenario_file(value):
         yaml_file = 'opnfv-{}.yaml'.format('test_file_name')
         if yaml_file in value:
             return False
         return True
 
-    def test_prepare_test_list_missing_scenario_file(self):
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.path.exists',
-                        side_effect=self.check_scenario_file), \
-                self.assertRaises(Exception):
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.path.exists')
+    def test_prepare_test_list_missing_scenario_file(self, mock_func):
+        mock_func.side_effect = self.check_scenario_file
+        with self.assertRaises(Exception):
             self.rally_base._prepare_test_list('test_file_name')
+        mock_func.assert_called()
 
-    def check_temp_dir(self, value):
+    @staticmethod
+    def check_temp_dir(value):
         yaml_file = 'opnfv-{}.yaml'.format('test_file_name')
         if yaml_file in value:
             return True
         return False
 
-    def test_prepare_test_list_missing_temp_dir(self):
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.path.exists',
-                        side_effect=self.check_temp_dir), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os.makedirs') as mock_os_makedir, \
-            mock.patch.object(self.rally_base, 'apply_blacklist',
-                              return_value=mock.Mock()) as mock_method:
-            yaml_file = 'opnfv-{}.yaml'.format('test_file_name')
-            ret_val = os.path.join(self.rally_base.TEMP_DIR, yaml_file)
-            self.assertEqual(self.rally_base.
-                             _prepare_test_list('test_file_name'),
-                             ret_val)
-            self.assertTrue(mock_method.called)
-            self.assertTrue(mock_os_makedir.called)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.path.exists')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.makedirs')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_apply_blacklist')
+    def test_prepare_test_list_missing_temp_dir(
+            self, mock_method, mock_os_makedirs, mock_path_exists):
+        mock_path_exists.side_effect = self.check_temp_dir
+
+        yaml_file = 'opnfv-{}.yaml'.format('test_file_name')
+        ret_val = os.path.join(self.rally_base.TEMP_DIR, yaml_file)
+        self.assertEqual(self.rally_base._prepare_test_list('test_file_name'),
+                         ret_val)
+        mock_path_exists.assert_called()
+        mock_method.assert_called()
+        mock_os_makedirs.assert_called()
 
     def test_get_task_id_default(self):
         cmd_raw = 'Task 1: started'
@@ -121,153 +103,164 @@ class OSRallyTesting(unittest.TestCase):
         self.assertEqual(self.rally_base.task_succeed(json_raw),
                          True)
 
-    def polling(self):
-        if self.polling_iter == 0:
-            return "something"
-        self.polling_iter -= 1
-        return None
-
     def test_get_cmd_output(self):
         proc = mock.Mock()
-        attrs = {'poll.side_effect': self.polling,
-                 'stdout.readline.return_value': 'line'}
-        proc.configure_mock(**attrs)
+        proc.stdout.__iter__ = mock.Mock(return_value=iter(['line1', 'line2']))
         self.assertEqual(self.rally_base.get_cmd_output(proc),
-                         'lineline')
-
-    def test_excl_scenario_default(self):
-        CONST.INSTALLER_TYPE = 'test_installer'
-        CONST.DEPLOY_SCENARIO = 'test_scenario'
-        dic = {'scenario': [{'scenarios': ['test_scenario'],
-                             'installers': ['test_installer'],
-                             'tests': ['test']}]}
-        with mock.patch('__builtin__.open', mock.mock_open()), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'yaml.safe_load',
-                       return_value=dic):
-                self.assertEqual(self.rally_base.excl_scenario(),
-                                 ['test'])
-
-    def test_excl_scenario_exception(self):
-        with mock.patch('__builtin__.open', side_effect=Exception):
-                self.assertEqual(self.rally_base.excl_scenario(),
-                                 [])
-
-    def test_excl_func_default(self):
-        CONST.INSTALLER_TYPE = 'test_installer'
-        CONST.DEPLOY_SCENARIO = 'test_scenario'
-        dic = {'functionality': [{'functions': ['no_live_migration'],
-                                  'tests': ['test']}]}
-        with mock.patch('__builtin__.open', mock.mock_open()), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'yaml.safe_load',
-                       return_value=dic), \
-            mock.patch.object(self.rally_base, 'live_migration_supported',
-                              return_value=False):
-                self.assertEqual(self.rally_base.excl_func(),
-                                 ['test'])
-
-    def test_excl_func_exception(self):
-        with mock.patch('__builtin__.open', side_effect=Exception):
-                self.assertEqual(self.rally_base.excl_func(),
-                                 [])
-
-    def test_file_is_empty_default(self):
-        mock_obj = mock.Mock()
+                         'line1line2')
+
+    @mock.patch('__builtin__.open', mock.mock_open())
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.yaml.safe_load',
+                return_value={'scenario': [
+                    {'scenarios': ['test_scenario'],
+                     'installers': ['test_installer'],
+                     'tests': ['test']},
+                    {'scenarios': ['other_scenario'],
+                     'installers': ['test_installer'],
+                     'tests': ['other_test']}]})
+    def test_excl_scenario_default(self, mock_func):
+        CONST.__setattr__('INSTALLER_TYPE', 'test_installer')
+        CONST.__setattr__('DEPLOY_SCENARIO', 'test_scenario')
+        self.assertEqual(self.rally_base.excl_scenario(), ['test'])
+        mock_func.assert_called()
+
+    @mock.patch('__builtin__.open', mock.mock_open())
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.yaml.safe_load',
+                return_value={'scenario': [
+                    {'scenarios': ['^os-[^-]+-featT-modeT$'],
+                     'installers': ['test_installer'],
+                     'tests': ['test1']},
+                    {'scenarios': ['^os-ctrlT-[^-]+-modeT$'],
+                     'installers': ['test_installer'],
+                     'tests': ['test2']},
+                    {'scenarios': ['^os-ctrlT-featT-[^-]+$'],
+                     'installers': ['test_installer'],
+                     'tests': ['test3']},
+                    {'scenarios': ['^os-'],
+                     'installers': ['test_installer'],
+                     'tests': ['test4']},
+                    {'scenarios': ['other_scenario'],
+                     'installers': ['test_installer'],
+                     'tests': ['test0a']},
+                    {'scenarios': [''],  # empty scenario
+                     'installers': ['test_installer'],
+                     'tests': ['test0b']}]})
+    def test_excl_scenario_regex(self, mock_func):
+        CONST.__setattr__('INSTALLER_TYPE', 'test_installer')
+        CONST.__setattr__('DEPLOY_SCENARIO', 'os-ctrlT-featT-modeT')
+        self.assertEqual(self.rally_base.excl_scenario(),
+                         ['test1', 'test2', 'test3', 'test4'])
+        mock_func.assert_called()
+
+    @mock.patch('__builtin__.open', side_effect=Exception)
+    def test_excl_scenario_exception(self, mock_open):
+        self.assertEqual(self.rally_base.excl_scenario(), [])
+        mock_open.assert_called()
+
+    @mock.patch('__builtin__.open', mock.mock_open())
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.yaml.safe_load',
+                return_value={'functionality': [
+                    {'functions': ['no_migration'], 'tests': ['test']}]})
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_migration_supported', return_value=False)
+    def test_excl_func_default(self, mock_func, mock_yaml_load):
+        CONST.__setattr__('INSTALLER_TYPE', 'test_installer')
+        CONST.__setattr__('DEPLOY_SCENARIO', 'test_scenario')
+        self.assertEqual(self.rally_base.excl_func(), ['test'])
+        mock_func.assert_called()
+        mock_yaml_load.assert_called()
+
+    @mock.patch('__builtin__.open', side_effect=Exception)
+    def test_excl_func_exception(self, mock_open):
+        self.assertEqual(self.rally_base.excl_func(), [])
+        mock_open.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.stat')
+    def test_file_is_empty_default(self, mock_os_stat):
         attrs = {'st_size': 10}
-        mock_obj.configure_mock(**attrs)
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.stat',
-                        return_value=mock_obj):
-            self.assertEqual(self.rally_base.file_is_empty('test_file_name'),
-                             False)
-
-    def test_file_is_empty_exception(self):
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.stat',
-                        side_effect=Exception):
-            self.assertEqual(self.rally_base.file_is_empty('test_file_name'),
-                             True)
-
-    def test_run_task_missing_task_file(self):
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.path.exists',
-                        return_value=False), \
-                self.assertRaises(Exception):
-            self.rally_base._run_task('test_name')
+        mock_os_stat.return_value.configure_mock(**attrs)
+        self.assertEqual(self.rally_base.file_is_empty('test_file_name'),
+                         False)
+        mock_os_stat.assert_called()
 
-    @mock.patch('functest.opnfv_tests.openstack.rally.rally.logger.info')
-    def test_run_task_no_tests_for_scenario(self, mock_logger_info):
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.path.exists',
-                        return_value=True), \
-            mock.patch.object(self.rally_base, '_prepare_test_list',
-                              return_value='test_file_name'), \
-            mock.patch.object(self.rally_base, 'file_is_empty',
-                              return_value=True):
-            self.rally_base._run_task('test_name')
-            str = 'No tests for scenario "test_name"'
-            mock_logger_info.assert_any_call(str)
-
-    @mock.patch('functest.opnfv_tests.openstack.rally.rally.logger.error')
-    def test_run_task_taskid_missing(self, mock_logger_error):
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.path.exists',
-                        return_value=True), \
-            mock.patch.object(self.rally_base, '_prepare_test_list',
-                              return_value='test_file_name'), \
-            mock.patch.object(self.rally_base, 'file_is_empty',
-                              return_value=False), \
-            mock.patch.object(self.rally_base, '_build_task_args',
-                              return_value={}), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'subprocess.Popen'), \
-            mock.patch.object(self.rally_base, '_get_output',
-                              return_value=mock.Mock()), \
-            mock.patch.object(self.rally_base, 'get_task_id',
-                              return_value=None), \
-            mock.patch.object(self.rally_base, 'get_cmd_output',
-                              return_value=''):
-            self.rally_base._run_task('test_name')
-            str = 'Failed to retrieve task_id, validating task...'
-            mock_logger_error.assert_any_call(str)
-
-    @mock.patch('functest.opnfv_tests.openstack.rally.rally.logger.info')
-    @mock.patch('functest.opnfv_tests.openstack.rally.rally.logger.error')
-    def test_run_task_default(self, mock_logger_error,
-                              mock_logger_info):
-        popen = mock.Mock()
-        attrs = {'read.return_value': 'json_result'}
-        popen.configure_mock(**attrs)
-
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os.path.exists',
-                        return_value=True), \
-            mock.patch.object(self.rally_base, '_prepare_test_list',
-                              return_value='test_file_name'), \
-            mock.patch.object(self.rally_base, 'file_is_empty',
-                              return_value=False), \
-            mock.patch.object(self.rally_base, '_build_task_args',
-                              return_value={}), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'subprocess.Popen'), \
-            mock.patch.object(self.rally_base, '_get_output',
-                              return_value=mock.Mock()), \
-            mock.patch.object(self.rally_base, 'get_task_id',
-                              return_value='1'), \
-            mock.patch.object(self.rally_base, 'get_cmd_output',
-                              return_value=''), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os.makedirs'), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os.popen',
-                       return_value=popen), \
-            mock.patch('__builtin__.open', mock.mock_open()), \
-            mock.patch.object(self.rally_base, 'task_succeed',
-                              return_value=True):
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.stat',
+                side_effect=Exception)
+    def test_file_is_empty_exception(self, mock_os_stat):
+        self.assertEqual(self.rally_base.file_is_empty('test_file_name'), True)
+        mock_os_stat.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.path.exists',
+                return_value=False)
+    def test_run_task_missing_task_file(self, mock_path_exists):
+        with self.assertRaises(Exception):
             self.rally_base._run_task('test_name')
-            str = 'Test scenario: "test_name" OK.\n'
-            mock_logger_info.assert_any_call(str)
+        mock_path_exists.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.path.exists',
+                return_value=True)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_prepare_test_list', return_value='test_file_name')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'file_is_empty', return_value=True)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.LOGGER.info')
+    def test_run_task_no_tests_for_scenario(self, mock_logger_info,
+                                            mock_file_empty, mock_prep_list,
+                                            mock_path_exists):
+        self.rally_base._run_task('test_name')
+        mock_logger_info.assert_any_call('No tests for scenario \"%s\"',
+                                         'test_name')
+        mock_file_empty.assert_called()
+        mock_prep_list.assert_called()
+        mock_path_exists.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_prepare_test_list', return_value='test_file_name')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'file_is_empty', return_value=False)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_build_task_args', return_value={})
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_append_summary')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'get_task_id', return_value=None)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'get_cmd_output', return_value='')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.path.exists',
+                return_value=True)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.subprocess.Popen')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.LOGGER.error')
+    def test_run_task_taskid_missing(self, mock_logger_error, *args):
+        self.rally_base._run_task('test_name')
+        text = 'Failed to retrieve task_id, validating task...'
+        mock_logger_error.assert_any_call(text)
+
+    @mock.patch('__builtin__.open', mock.mock_open())
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_prepare_test_list', return_value='test_file_name')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'file_is_empty', return_value=False)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_build_task_args', return_value={})
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_append_summary')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'get_task_id', return_value='1')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'get_cmd_output', return_value='')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                'task_succeed', return_value=True)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.path.exists',
+                return_value=True)
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.subprocess.Popen')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.os.makedirs')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.LOGGER.info')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.LOGGER.error')
+    def test_run_task_default(self, mock_logger_error, mock_logger_info,
+                              *args):
+        self.rally_base._run_task('test_name')
+        text = 'Test scenario: "test_name" OK.\n'
+        mock_logger_info.assert_any_call(text)
+        mock_logger_error.assert_not_called()
 
     def test_prepare_env_testname_invalid(self):
         self.rally_base.TESTS = ['test1', 'test2']
@@ -275,104 +268,173 @@ class OSRallyTesting(unittest.TestCase):
         with self.assertRaises(Exception):
             self.rally_base._prepare_env()
 
-    def test_prepare_env_volume_creation_failed(self):
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_active_compute_cnt')
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_ext_net_name', return_value='test_net_name')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_image',
+                return_value=None)
+    def test_prepare_env_image_missing(
+            self, mock_get_img, mock_get_net, mock_get_comp_cnt):
         self.rally_base.TESTS = ['test1', 'test2']
         self.rally_base.test_name = 'test1'
-        volume_type = None
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.list_volume_types',
-                        return_value=None), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.create_volume_type',
-                       return_value=volume_type), \
-                self.assertRaises(Exception):
+        with self.assertRaises(Exception):
             self.rally_base._prepare_env()
-
-    def test_prepare_env_image_missing(self):
+        mock_get_img.assert_called()
+        mock_get_net.assert_called()
+        mock_get_comp_cnt.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_active_compute_cnt')
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_ext_net_name', return_value='test_net_name')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_image')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_network',
+                return_value=None)
+    def test_prepare_env_network_creation_failed(
+            self, mock_create_net, mock_get_img, mock_get_net,
+            mock_get_comp_cnt):
         self.rally_base.TESTS = ['test1', 'test2']
         self.rally_base.test_name = 'test1'
-        volume_type = mock.Mock()
-        image_id = None
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.list_volume_types',
-                        return_value=None), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.create_volume_type',
-                       return_value=volume_type), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.get_or_create_image',
-                       return_value=(True, image_id)), \
-                self.assertRaises(Exception):
+        with self.assertRaises(Exception):
             self.rally_base._prepare_env()
-
-    def test_prepare_env_image_shared_network_creation_failed(self):
+        mock_create_net.assert_called()
+        mock_get_img.assert_called()
+        mock_get_net.assert_called()
+        mock_get_comp_cnt.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_active_compute_cnt')
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_ext_net_name', return_value='test_net_name')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_image')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_network')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_router',
+                return_value=None)
+    def test_prepare_env_router_creation_failed(
+            self, mock_create_router, mock_create_net, mock_get_img,
+            mock_get_net, mock_get_comp_cnt):
         self.rally_base.TESTS = ['test1', 'test2']
         self.rally_base.test_name = 'test1'
-        volume_type = mock.Mock()
-        image_id = 'image_id'
-        network_dict = None
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.list_volume_types',
-                        return_value=None), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.create_volume_type',
-                       return_value=volume_type), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.get_or_create_image',
-                       return_value=(True, image_id)), \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.create_shared_network_full',
-                       return_value=network_dict), \
-                self.assertRaises(Exception):
+        with self.assertRaises(Exception):
             self.rally_base._prepare_env()
-
-    def test_run_tests_all(self):
+        mock_create_net.assert_called()
+        mock_get_img.assert_called()
+        mock_get_net.assert_called()
+        mock_create_router.assert_called()
+        mock_get_comp_cnt.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_active_compute_cnt')
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_ext_net_name', return_value='test_net_name')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_image')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_network')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_router')
+    @mock.patch('snaps.openstack.create_flavor.OpenStackFlavor.create',
+                return_value=None)
+    def test_prepare_env_flavor_creation_failed(
+            self, mock_create_flavor, mock_create_router, mock_create_net,
+            mock_get_img, mock_get_net, mock_get_comp_cnt):
+        self.rally_base.TESTS = ['test1', 'test2']
+        self.rally_base.test_name = 'test1'
+        with self.assertRaises(Exception):
+            self.rally_base._prepare_env()
+        mock_create_net.assert_called()
+        mock_get_img.assert_called()
+        mock_get_net.assert_called()
+        mock_create_router.assert_called()
+        mock_get_comp_cnt.assert_called()
+        mock_create_flavor.assert_called_once()
+
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_active_compute_cnt')
+    @mock.patch('functest.opnfv_tests.openstack.snaps.snaps_utils.'
+                'get_ext_net_name', return_value='test_net_name')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_image')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_network')
+    @mock.patch('snaps.openstack.utils.deploy_utils.create_router')
+    @mock.patch('snaps.openstack.create_flavor.OpenStackFlavor.create',
+                side_effect=[mock.Mock, None])
+    def test_prepare_env_flavor_alt_creation_failed(
+            self, mock_create_flavor, mock_create_router, mock_create_net,
+            mock_get_img, mock_get_net, mock_get_comp_cnt):
+        self.rally_base.TESTS = ['test1', 'test2']
+        self.rally_base.test_name = 'test1'
+        with self.assertRaises(Exception):
+            self.rally_base._prepare_env()
+        mock_create_net.assert_called()
+        mock_get_img.assert_called()
+        mock_get_net.assert_called()
+        mock_create_router.assert_called()
+        mock_get_comp_cnt.assert_called()
+        self.assertEqual(mock_create_flavor.call_count, 2)
+
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_run_task')
+    def test_run_tests_all(self, mock_run_task):
         self.rally_base.TESTS = ['test1', 'test2']
         self.rally_base.test_name = 'all'
-        with mock.patch.object(self.rally_base, '_run_task',
-                               return_value=mock.Mock()):
-            self.rally_base._run_tests()
-            self.rally_base._run_task.assert_any_call('test1')
-            self.rally_base._run_task.assert_any_call('test2')
+        self.rally_base._run_tests()
+        mock_run_task.assert_any_call('test1')
+        mock_run_task.assert_any_call('test2')
 
-    def test_run_tests_default(self):
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_run_task')
+    def test_run_tests_default(self, mock_run_task):
         self.rally_base.TESTS = ['test1', 'test2']
         self.rally_base.test_name = 'test1'
-        with mock.patch.object(self.rally_base, '_run_task',
-                               return_value=mock.Mock()):
-            self.rally_base._run_tests()
-            self.rally_base._run_task.assert_any_call('test1')
+        self.rally_base._run_tests()
+        mock_run_task.assert_any_call('test1')
 
     def test_clean_up_default(self):
-        self.rally_base.volume_type = mock.Mock()
-        self.rally_base.cinder_client = mock.Mock()
-        self.rally_base.image_exists = False
-        self.rally_base.image_id = 1
-        self.rally_base.nova_client = mock.Mock()
-        with mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                        'os_utils.delete_volume_type') as mock_vol_method, \
-            mock.patch('functest.opnfv_tests.openstack.rally.rally.'
-                       'os_utils.delete_glance_image') as mock_glance_method:
-            self.rally_base._clean_up()
-            mock_vol_method.assert_any_call(self.rally_base.cinder_client,
-                                            self.rally_base.volume_type)
-            mock_glance_method.assert_any_call(self.rally_base.nova_client,
-                                               1)
-
-    def test_run_default(self):
-        with mock.patch.object(self.rally_base, '_prepare_env'), \
-            mock.patch.object(self.rally_base, '_run_tests'), \
-            mock.patch.object(self.rally_base, '_generate_report'), \
-                mock.patch.object(self.rally_base, '_clean_up'):
-            self.assertEqual(self.rally_base.run(),
-                             testcase.TestCase.EX_OK)
-
-    def test_run_exception(self):
-        with mock.patch.object(self.rally_base, '_prepare_env',
-                               side_effect=Exception):
-            self.assertEqual(self.rally_base.run(),
-                             testcase.TestCase.EX_RUN_ERROR)
+        creator1 = mock.Mock()
+        creator2 = mock.Mock()
+        self.rally_base.creators = [creator1, creator2]
+        self.rally_base._clean_up()
+        self.assertTrue(creator1.clean.called)
+        self.assertTrue(creator2.clean.called)
+
+    @mock.patch('functest.opnfv_tests.openstack.tempest.conf_utils.'
+                'create_rally_deployment')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_prepare_env')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_run_tests')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_generate_report')
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_clean_up')
+    def test_run_default(self, *args):
+        self.assertEqual(self.rally_base.run(), testcase.TestCase.EX_OK)
+        map(lambda m: m.assert_called(), args)
+
+    @mock.patch('functest.opnfv_tests.openstack.tempest.conf_utils.'
+                'create_rally_deployment', side_effect=Exception)
+    def test_run_exception_create_rally_dep(self, mock_create_rally_dep):
+        self.assertEqual(self.rally_base.run(), testcase.TestCase.EX_RUN_ERROR)
+        mock_create_rally_dep.assert_called()
+
+    @mock.patch('functest.opnfv_tests.openstack.rally.rally.RallyBase.'
+                '_prepare_env', side_effect=Exception)
+    @mock.patch('functest.opnfv_tests.openstack.tempest.conf_utils.'
+                'create_rally_deployment', return_value=mock.Mock())
+    def test_run_exception_prepare_env(self, mock_create_rally_dep,
+                                       mock_prep_env):
+        self.assertEqual(self.rally_base.run(), testcase.TestCase.EX_RUN_ERROR)
+        mock_prep_env.assert_called()
+
+    def test_append_summary(self):
+        text = '[{"result":[{"error":[]},{"error":["err"]}],' \
+               '"full_duration": 17.312026}]'
+        self.rally_base._append_summary(text, "foo_test")
+        self.assertEqual(self.rally_base.summary[0]['test_name'], "foo_test")
+        self.assertEqual(self.rally_base.summary[0]['overall_duration'],
+                         17.312026)
+        self.assertEqual(self.rally_base.summary[0]['nb_tests'], 2)
+        self.assertEqual(self.rally_base.summary[0]['nb_success'], 1)
 
 
 if __name__ == "__main__":
+    logging.disable(logging.CRITICAL)
     unittest.main(verbosity=2)