def test_collect_logs(self, mkdirs_mock, copy_mock, tarfile_open_mock, get_files_mock, server_log_mock, launcher_log_mock): downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs") collect.logs() mkdirs_mock.assert_called_with(downloaded_logs_loc) copy_mock.assert_called_with( path.join(get_log_directory(), PRESTOADMIN_LOG_NAME), downloaded_logs_loc) tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:gz') tar = tarfile_open_mock.return_value tar.add.assert_called_with(downloaded_logs_loc, arcname=path.basename(downloaded_logs_loc))
def test_collect_logs(self, mkdirs_mock, copy_mock, tarfile_open_mock, get_files_mock, server_log_mock, launcher_log_mock): downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs") collect.logs() mkdirs_mock.assert_called_with(downloaded_logs_loc) copy_mock.assert_called_with(path.join(get_log_directory(), PRESTOADMIN_LOG_NAME), downloaded_logs_loc) tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2') tar = tarfile_open_mock.return_value tar.add.assert_called_with(downloaded_logs_loc, arcname=path.basename(downloaded_logs_loc))
def test_collect_logs(self, path_exists_mock, mkdirs_mock, copy_mock, tarfile_open_mock, file_get_mock, server_log_mock, launcher_log_mock): downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs") path_exists_mock.return_value = False collect.logs() mkdirs_mock.assert_called_with(downloaded_logs_loc) copy_mock.assert_called_with( path.join(PRESTOADMIN_LOG_DIR, PRESTOADMIN_LOG_NAME), downloaded_logs_loc) tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2') tar = tarfile_open_mock.return_value tar.add.assert_called_with(downloaded_logs_loc, arcname=path.basename(downloaded_logs_loc))
def test_collect_logs(self, path_exists_mock, mkdirs_mock, copy_mock, tarfile_open_mock, mock_execute): downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs") path_exists_mock.return_value = False collect.logs() mkdirs_mock.assert_any_call(TMP_PRESTO_DEBUG) mkdirs_mock.assert_called_with(downloaded_logs_loc) copy_mock.assert_called_with( path.join(PRESTOADMIN_LOG_DIR, PRESTOADMIN_LOG_NAME), downloaded_logs_loc) mock_execute.assert_called_with(collect.file_get, REMOTE_PRESTO_LOG_DIR, downloaded_logs_loc, roles=[]) tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2') tar = tarfile_open_mock.return_value tar.add.assert_called_with(downloaded_logs_loc, arcname=path.basename(downloaded_logs_loc))
def test_collect_logs(self, path_exists_mock, mkdirs_mock, copy_mock, tarfile_open_mock, mock_execute): downloaded_logs_loc = path.join(TMP_PRESTO_DEBUG, "logs") path_exists_mock.return_value = False collect.logs() mkdirs_mock.assert_any_call(TMP_PRESTO_DEBUG) mkdirs_mock.assert_called_with(downloaded_logs_loc) copy_mock.assert_called_with(path.join(PRESTOADMIN_LOG_DIR, PRESTOADMIN_LOG_NAME), downloaded_logs_loc) mock_execute.assert_called_with(collect.file_get, REMOTE_PRESTO_LOG_DIR, downloaded_logs_loc, roles=[]) tarfile_open_mock.assert_called_with(OUTPUT_FILENAME_FOR_LOGS, 'w:bz2') tar = tarfile_open_mock.return_value tar.add.assert_called_with(downloaded_logs_loc, arcname=path.basename(downloaded_logs_loc))