/*******************************************************************************
* Copyright Technophobia Ltd 2012
*
* This file is part of the Substeps Eclipse Plugin.
*
* The Substeps Eclipse Plugin is free software: you can redistribute it and/or modify
* it under the terms of the Eclipse Public License v1.0.
*
* The Substeps Eclipse Plugin is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* Eclipse Public License for more details.
*
* You should have received a copy of the Eclipse Public License
* along with the Substeps Eclipse Plugin. If not, see <http://www.eclipse.org/legal/epl-v10.html>.
******************************************************************************/
package com.technophobia.substeps.junit.ui;
import java.io.BufferedWriter;
import java.io.File;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.text.MessageFormat;
import org.eclipse.core.runtime.CoreException;
import org.eclipse.core.runtime.IStatus;
import org.eclipse.core.runtime.Status;
import org.eclipse.debug.core.ILaunchConfiguration;
import org.eclipse.debug.core.ILaunchConfigurationWorkingCopy;
import org.eclipse.jface.dialogs.ErrorDialog;
import org.eclipse.swt.widgets.Shell;
import com.technophobia.eclipse.launcher.config.SubstepsLaunchConfigurationConstants;
import com.technophobia.eclipse.ui.Notifier;
import com.technophobia.substeps.FeatureRunnerPlugin;
import com.technophobia.substeps.model.structure.SubstepsTestElement;
import com.technophobia.substeps.supplier.Supplier;
public class FailedTestFirstTestRelauncher extends TestRelauncher {
private final Shell shell;
final Supplier<SubstepsRunSession> testRunSession;
public FailedTestFirstTestRelauncher(final Supplier<SubstepsRunSession> testRunSession, final Shell shell,
final Notifier<String> infoMessageNotifier) {
super(testRunSession, shell, infoMessageNotifier);
this.testRunSession = testRunSession;
this.shell = shell;
}
@Override
protected ILaunchConfiguration prepareLaunchConfigForRelaunch(final ILaunchConfiguration launchConfiguration) {
try {
final String oldName = launchConfiguration.getName();
final String oldFailuresFilename = launchConfiguration.getAttribute(
SubstepsLaunchConfigurationConstants.ATTR_FAILURES_NAMES, (String) null);
String configName;
if (oldFailuresFilename != null) {
configName = oldName;
} else {
configName = MessageFormat.format(
SubstepsFeatureMessages.SubstepsFeatureTestRunnerViewPart_rerunFailedFirstLaunchConfigName,
oldName);
}
final ILaunchConfigurationWorkingCopy tmp = launchConfiguration.copy(configName);
tmp.setAttribute(SubstepsLaunchConfigurationConstants.ATTR_FAILURES_NAMES, createFailureNamesFile());
return tmp;
} catch (final CoreException ex) {
ErrorDialog.openError(shell, SubstepsFeatureMessages.SubstepsFeatureTestRunnerViewPart_error_cannotrerun,
ex.getMessage(), ex.getStatus());
return null;
}
}
private String createFailureNamesFile() throws CoreException {
try {
final File file = File.createTempFile("testFailures", ".txt"); //$NON-NLS-1$ //$NON-NLS-2$
file.deleteOnExit();
final SubstepsTestElement[] failures = testRunSession.get().getAllFailedTestElements();
BufferedWriter bw = null;
try {
bw = new BufferedWriter(new OutputStreamWriter(new FileOutputStream(file), "UTF-8")); //$NON-NLS-1$
for (final SubstepsTestElement testElement : failures) {
bw.write(testElement.getTestName());
bw.newLine();
}
} finally {
if (bw != null) {
bw.close();
}
}
return file.getAbsolutePath();
} catch (final IOException e) {
throw new CoreException(new Status(IStatus.ERROR, FeatureRunnerPlugin.PLUGIN_ID, IStatus.ERROR, "", e)); //$NON-NLS-1$
}
}
}