Remove call to fetch_os_creds.sh
[functest.git] / functest / tests / unit / ci / test_prepare_env.py
index 41b3f6a..d38fa97 100644 (file)
@@ -18,7 +18,8 @@ from opnfv.utils import constants as opnfv_constants
 
 class PrepareEnvTesting(unittest.TestCase):
 
-    logging.disable(logging.CRITICAL)
+    def setUp(self):
+        self.prepare_envparser = prepare_env.PrepareEnvParser()
 
     @mock.patch('functest.ci.prepare_env.logger.info')
     def test_print_separator(self, mock_logger_info):
@@ -30,7 +31,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_missing_inst_type(self, mock_logger_warn,
                                                    mock_logger_info):
-        CONST.INSTALLER_TYPE = None
+        CONST.__setattr__('INSTALLER_TYPE', None)
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -41,24 +42,19 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_missing_inst_ip(self, mock_logger_warn,
                                                  mock_logger_info):
-        CONST.INSTALLER_IP = None
+        CONST.__setattr__('INSTALLER_IP', None)
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
-        mock_logger_warn.assert_any_call("The env variable 'INSTALLER_IP'"
-                                         " is not defined. It is needed to"
-                                         " fetch the OpenStack credentials."
-                                         " If the credentials are not"
-                                         " provided to the container as a"
-                                         " volume, please add this env"
-                                         " variable to the 'docker run'"
-                                         " command.")
+        mock_logger_warn.assert_any_call(
+            "The env variable 'INSTALLER_IP' is not defined. It is recommended"
+            " to extract some information from the deployment")
 
     @mock.patch('functest.ci.prepare_env.logger.info')
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_with_inst_ip(self, mock_logger_warn,
                                               mock_logger_info):
-        CONST.INSTALLER_IP = mock.Mock()
+        CONST.__setattr__('INSTALLER_IP', mock.Mock())
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -69,7 +65,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_missing_scenario(self, mock_logger_warn,
                                                   mock_logger_info):
-        CONST.DEPLOY_SCENARIO = None
+        CONST.__setattr__('DEPLOY_SCENARIO', None)
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -81,7 +77,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_with_scenario(self, mock_logger_warn,
                                                mock_logger_info):
-        CONST.DEPLOY_SCENARIO = mock.Mock()
+        CONST.__setattr__('DEPLOY_SCENARIO', 'test_scenario')
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -92,7 +88,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_with_ci_debug(self, mock_logger_warn,
                                                mock_logger_info):
-        CONST.CI_DEBUG = mock.Mock()
+        CONST.__setattr__('CI_DEBUG', mock.Mock())
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -103,7 +99,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_with_node(self, mock_logger_warn,
                                            mock_logger_info):
-        CONST.NODE_NAME = mock.Mock()
+        CONST.__setattr__('NODE_NAME', mock.Mock())
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -114,7 +110,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_with_build_tag(self, mock_logger_warn,
                                                 mock_logger_info):
-        CONST.BUILD_TAG = mock.Mock()
+        CONST.__setattr__('BUILD_TAG', mock.Mock())
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -126,7 +122,7 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.logger.warning')
     def test_check_env_variables_with_is_ci_run(self, mock_logger_warn,
                                                 mock_logger_info):
-        CONST.IS_CI_RUN = mock.Mock()
+        CONST.__setattr__('IS_CI_RUN', mock.Mock())
         prepare_env.check_env_variables()
         mock_logger_info.assert_any_call("Checking environment variables"
                                          "...")
@@ -134,6 +130,47 @@ class PrepareEnvTesting(unittest.TestCase):
         mock_logger_info.assert_any_call(test_utils.
                                          SubstrMatch("    IS_CI_RUN="))
 
+    def test_get_deployment_handler_missing_const_vars(self):
+        with mock.patch('functest.ci.prepare_env.'
+                        'factory.Factory.get_handler') as m:
+            CONST.__setattr__('INSTALLER_IP', None)
+            prepare_env.get_deployment_handler()
+            self.assertFalse(m.called)
+
+            CONST.__setattr__('INSTALLER_TYPE', None)
+            prepare_env.get_deployment_handler()
+            self.assertFalse(m.called)
+
+    @mock.patch('functest.ci.prepare_env.logger.debug')
+    def test_get_deployment_handler_missing_print_deploy_info(self,
+                                                              mock_debug):
+        with mock.patch('functest.ci.prepare_env.'
+                        'factory.Factory.get_handler') as m, \
+            mock.patch('functest.ci.prepare_env.'
+                       'ft_utils.get_parameter_from_yaml',
+                       side_effect=ValueError):
+            CONST.__setattr__('INSTALLER_IP', 'test_ip')
+            CONST.__setattr__('INSTALLER_TYPE', 'test_inst_type')
+            opnfv_constants.INSTALLERS = ['test_inst_type']
+            prepare_env.get_deployment_handler()
+            msg = ('Printing deployment info is not supported for '
+                   'test_inst_type')
+            mock_debug.assert_any_call(msg)
+            self.assertFalse(m.called)
+
+    @mock.patch('functest.ci.prepare_env.logger.debug')
+    def test_get_deployment_handler_exception(self, mock_debug):
+        with mock.patch('functest.ci.prepare_env.'
+                        'factory.Factory.get_handler',
+                        side_effect=Exception), \
+            mock.patch('functest.ci.prepare_env.'
+                       'ft_utils.get_parameter_from_yaml'):
+            CONST.__setattr__('INSTALLER_IP', 'test_ip')
+            CONST.__setattr__('INSTALLER_TYPE', 'test_inst_type')
+            opnfv_constants.INSTALLERS = ['test_inst_type']
+            prepare_env.get_deployment_handler()
+            self.assertTrue(mock_debug.called)
+
     @mock.patch('functest.ci.prepare_env.logger.info')
     @mock.patch('functest.ci.prepare_env.logger.debug')
     def test_create_directories_missing_dir(self, mock_logger_debug,
@@ -144,12 +181,21 @@ class PrepareEnvTesting(unittest.TestCase):
                 as mock_method:
             prepare_env.create_directories()
             mock_logger_info.assert_any_call("Creating needed directories...")
-            mock_method.assert_any_call(CONST.dir_functest_conf)
-            mock_method.assert_any_call(CONST.dir_functest_data)
+            mock_method.assert_any_call(
+                CONST.__getattribute__('dir_functest_conf'))
+            mock_method.assert_any_call(
+                CONST.__getattribute__('dir_functest_data'))
+            mock_method.assert_any_call(
+                CONST.__getattribute__('dir_functest_images'))
+            mock_logger_info.assert_any_call("    %s created." %
+                                             CONST.__getattribute__(
+                                                 'dir_functest_conf'))
             mock_logger_info.assert_any_call("    %s created." %
-                                             CONST.dir_functest_conf)
+                                             CONST.__getattribute__(
+                                                 'dir_functest_data'))
             mock_logger_info.assert_any_call("    %s created." %
-                                             CONST.dir_functest_data)
+                                             CONST.__getattribute__(
+                                                 'dir_functest_images'))
 
     @mock.patch('functest.ci.prepare_env.logger.info')
     @mock.patch('functest.ci.prepare_env.logger.debug')
@@ -160,9 +206,14 @@ class PrepareEnvTesting(unittest.TestCase):
             prepare_env.create_directories()
             mock_logger_info.assert_any_call("Creating needed directories...")
             mock_logger_debug.assert_any_call("   %s already exists." %
-                                              CONST.dir_functest_conf)
+                                              CONST.__getattribute__(
+                                                  'dir_functest_conf'))
+            mock_logger_debug.assert_any_call("   %s already exists." %
+                                              CONST.__getattribute__(
+                                                  'dir_functest_data'))
             mock_logger_debug.assert_any_call("   %s already exists." %
-                                              CONST.dir_functest_data)
+                                              CONST.__getattribute__(
+                                                  'dir_functest_images'))
 
     def _get_env_cred_dict(self, os_prefix=''):
         return {'OS_USERNAME': os_prefix + 'username',
@@ -186,24 +237,24 @@ class PrepareEnvTesting(unittest.TestCase):
                 mock.patch('functest.ci.prepare_env.os.path.getsize',
                            return_value=0), \
                 self.assertRaises(Exception):
-            CONST.openstack_creds = 'test_creds'
+            CONST.__setattr__('openstack_creds', 'test_creds')
             prepare_env.source_rc_file()
 
     def test_source_rc_missing_installer_ip(self):
         with mock.patch('functest.ci.prepare_env.os.path.isfile',
                         return_value=False), \
                 self.assertRaises(Exception):
-            CONST.INSTALLER_IP = None
-            CONST.openstack_creds = 'test_creds'
+            CONST.__setattr__('INSTALLER_IP', None)
+            CONST.__setattr__('openstack_creds', 'test_creds')
             prepare_env.source_rc_file()
 
     def test_source_rc_missing_installer_type(self):
         with mock.patch('functest.ci.prepare_env.os.path.isfile',
                         return_value=False), \
                 self.assertRaises(Exception):
-            CONST.INSTALLER_IP = 'test_ip'
-            CONST.openstack_creds = 'test_creds'
-            CONST.INSTALLER_TYPE = 'test_type'
+            CONST.__setattr__('INSTALLER_IP', 'test_ip')
+            CONST.__setattr__('openstack_creds', 'test_creds')
+            CONST.__setattr__('INSTALLER_TYPE', 'test_type')
             opnfv_constants.INSTALLERS = []
             prepare_env.source_rc_file()
 
@@ -215,9 +266,9 @@ class PrepareEnvTesting(unittest.TestCase):
                 mock.patch('functest.ci.prepare_env.subprocess.Popen') \
                 as mock_subproc_popen, \
                 self.assertRaises(Exception):
-            CONST.openstack_creds = 'test_creds'
-            CONST.INSTALLER_IP = None
-            CONST.INSTALLER_TYPE = 'test_type'
+            CONST.__setattr__('openstack_creds', None)
+            CONST.__setattr__('INSTALLER_IP', 'test_ip')
+            CONST.__setattr__('INSTALLER_TYPE', 'test_type')
             opnfv_constants.INSTALLERS = ['test_type']
 
             process_mock = mock.Mock()
@@ -228,6 +279,36 @@ class PrepareEnvTesting(unittest.TestCase):
 
             prepare_env.source_rc_file()
 
+    @mock.patch('functest.ci.prepare_env.logger.debug')
+    def test_patch_file(self, mock_logger_debug):
+        with mock.patch("__builtin__.open", mock.mock_open()), \
+            mock.patch('functest.ci.prepare_env.yaml.safe_load',
+                       return_value={'test_scenario': {'tkey': 'tvalue'}}), \
+            mock.patch('functest.ci.prepare_env.ft_utils.get_functest_yaml',
+                       return_value={'tkey1': 'tvalue1'}), \
+            mock.patch('functest.ci.prepare_env.os.remove') as m, \
+                mock.patch('functest.ci.prepare_env.yaml.dump'):
+            CONST.__setattr__('DEPLOY_SCENARIO', 'test_scenario')
+            prepare_env.patch_file('test_file')
+            self.assertTrue(m.called)
+
+    @mock.patch('functest.ci.prepare_env.logger.info')
+    def test_verify_deployment_error(self, mock_logger_error):
+        mock_popen = mock.Mock()
+        attrs = {'poll.return_value': None,
+                 'stdout.readline.return_value': 'ERROR'}
+        mock_popen.configure_mock(**attrs)
+
+        with mock.patch('functest.ci.prepare_env.print_separator') as m, \
+            mock.patch('functest.ci.prepare_env.subprocess.Popen',
+                       return_value=mock_popen), \
+                self.assertRaises(Exception) as context:
+            prepare_env.verify_deployment()
+            self.assertTrue(m.called)
+            msg = "Problem while running 'check_os.sh'."
+            mock_logger_error.assert_called_once_with('ERROR')
+            self.assertTrue(msg in context)
+
     def _get_rally_creds(self):
         return {"type": "ExistingCloud",
                 "admin": {"username": 'test_user_name',
@@ -236,8 +317,10 @@ class PrepareEnvTesting(unittest.TestCase):
 
     @mock.patch('functest.ci.prepare_env.os_utils.get_credentials_for_rally')
     @mock.patch('functest.ci.prepare_env.logger.info')
+    @mock.patch('functest.ci.prepare_env.ft_utils.execute_command_raise')
     @mock.patch('functest.ci.prepare_env.ft_utils.execute_command')
-    def test_install_rally(self, mock_exec, mock_logger_info, mock_os_utils):
+    def test_install_rally(self, mock_exec, mock_exec_raise, mock_logger_info,
+                           mock_os_utils):
 
         mock_os_utils.return_value = self._get_rally_creds()
 
@@ -245,19 +328,19 @@ class PrepareEnvTesting(unittest.TestCase):
 
         cmd = "rally deployment destroy opnfv-rally"
         error_msg = "Deployment %s does not exist." % \
-                    CONST.rally_deployment_name
+                    CONST.__getattribute__('rally_deployment_name')
         mock_logger_info.assert_any_call("Creating Rally environment...")
         mock_exec.assert_any_call(cmd, error_msg=error_msg, verbose=False)
 
         cmd = "rally deployment create --file=rally_conf.json --name="
-        cmd += CONST.rally_deployment_name
+        cmd += CONST.__getattribute__('rally_deployment_name')
         error_msg = "Problem while creating Rally deployment"
-        mock_exec.assert_any_call(cmd, error_msg=error_msg)
+        mock_exec_raise.assert_any_call(cmd, error_msg=error_msg)
 
         cmd = "rally deployment check"
         error_msg = ("OpenStack not responding or "
                      "faulty Rally deployment.")
-        mock_exec.assert_any_call(cmd, error_msg=error_msg)
+        mock_exec_raise.assert_any_call(cmd, error_msg=error_msg)
 
         cmd = "rally deployment list"
         error_msg = ("Problem while listing "
@@ -269,6 +352,33 @@ class PrepareEnvTesting(unittest.TestCase):
                      "Rally plugins.")
         mock_exec.assert_any_call(cmd, error_msg=error_msg)
 
+    @mock.patch('functest.ci.prepare_env.logger.debug')
+    def test_install_tempest(self, mock_logger_debug):
+        mock_popen = mock.Mock()
+        attrs = {'poll.return_value': None,
+                 'stdout.readline.return_value': '0'}
+        mock_popen.configure_mock(**attrs)
+
+        CONST.__setattr__('tempest_deployment_name', 'test_dep_name')
+        with mock.patch('functest.ci.prepare_env.'
+                        'ft_utils.execute_command_raise',
+                        side_effect=Exception), \
+            mock.patch('functest.ci.prepare_env.subprocess.Popen',
+                       return_value=mock_popen), \
+                self.assertRaises(Exception):
+            prepare_env.install_tempest()
+            mock_logger_debug.assert_any_call("Tempest test_dep_name"
+                                              " does not exist")
+
+    def test_create_flavor(self):
+        with mock.patch('functest.ci.prepare_env.'
+                        'os_utils.get_or_create_flavor',
+                        return_value=('test_', None)), \
+                self.assertRaises(Exception) as context:
+            prepare_env.create_flavor()
+            msg = 'Failed to create flavor'
+            self.assertTrue(msg in context)
+
     @mock.patch('functest.ci.prepare_env.sys.exit')
     @mock.patch('functest.ci.prepare_env.logger.error')
     def test_check_environment_missing_file(self, mock_logger_error,
@@ -276,7 +386,7 @@ class PrepareEnvTesting(unittest.TestCase):
         with mock.patch('functest.ci.prepare_env.os.path.isfile',
                         return_value=False), \
                 self.assertRaises(Exception):
-                prepare_env.check_environment()
+            prepare_env.check_environment()
 
     @mock.patch('functest.ci.prepare_env.sys.exit')
     @mock.patch('functest.ci.prepare_env.logger.error')
@@ -297,6 +407,7 @@ class PrepareEnvTesting(unittest.TestCase):
                 mock_logger_info.assert_any_call("Functest environment"
                                                  " is installed.")
 
+    @mock.patch('functest.ci.prepare_env.print_deployment_info')
     @mock.patch('functest.ci.prepare_env.check_environment')
     @mock.patch('functest.ci.prepare_env.create_flavor')
     @mock.patch('functest.ci.prepare_env.install_tempest')
@@ -305,19 +416,21 @@ class PrepareEnvTesting(unittest.TestCase):
     @mock.patch('functest.ci.prepare_env.patch_config_file')
     @mock.patch('functest.ci.prepare_env.source_rc_file')
     @mock.patch('functest.ci.prepare_env.create_directories')
+    @mock.patch('functest.ci.prepare_env.get_deployment_handler')
     @mock.patch('functest.ci.prepare_env.check_env_variables')
     @mock.patch('functest.ci.prepare_env.logger.info')
-    def test_main_start(self, mock_logger_info, mock_env_var,
+    def test_main_start(self, mock_logger_info, mock_env_var, mock_dep_handler,
                         mock_create_dir, mock_source_rc, mock_patch_config,
                         mock_verify_depl, mock_install_rally,
                         mock_install_temp, mock_create_flavor,
-                        mock_check_env):
+                        mock_check_env, mock_print_info):
         with mock.patch("__builtin__.open", mock.mock_open()) as m:
             args = {'action': 'start'}
             self.assertEqual(prepare_env.main(**args), 0)
             mock_logger_info.assert_any_call("######### Preparing Functest "
                                              "environment #########\n")
             self.assertTrue(mock_env_var.called)
+            self.assertTrue(mock_dep_handler.called)
             self.assertTrue(mock_create_dir.called)
             self.assertTrue(mock_source_rc.called)
             self.assertTrue(mock_patch_config.called)
@@ -325,8 +438,10 @@ class PrepareEnvTesting(unittest.TestCase):
             self.assertTrue(mock_install_rally.called)
             self.assertTrue(mock_install_temp.called)
             self.assertTrue(mock_create_flavor.called)
-            m.assert_called_once_with(CONST.env_active, "w")
+            m.assert_called_once_with(
+                CONST.__getattribute__('env_active'), "w")
             self.assertTrue(mock_check_env.called)
+            self.assertTrue(mock_print_info.called)
 
     @mock.patch('functest.ci.prepare_env.check_environment')
     def test_main_check(self, mock_check_env):
@@ -342,4 +457,5 @@ class PrepareEnvTesting(unittest.TestCase):
 
 
 if __name__ == "__main__":
+    logging.disable(logging.CRITICAL)
     unittest.main(verbosity=2)