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)