def test_main_start(self, mock_logger_info, mock_env_var, mock_create_dir, mock_source_rc, mock_patch_config, mock_verify_depl, mock_install_rally, mock_install_temp, mock_create_flavor, mock_check_env): 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_create_dir.called) self.assertTrue(mock_source_rc.called) self.assertTrue(mock_patch_config.called) self.assertTrue(mock_verify_depl.called) 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") self.assertTrue(mock_check_env.called)
def test_main_no_arg(self, mock_logger_error): args = {'action': 'not_valid'} self.assertEqual(prepare_env.main(**args), -1) mock_logger_error.assert_called_once_with('Argument not valid.')
def test_main_check(self, mock_check_env): args = {'action': 'check'} self.assertEqual(prepare_env.main(**args), 0) self.assertTrue(mock_check_env.called)