package org.testng.eclipse.ui;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringReader;
import java.io.StringWriter;
import org.eclipse.jdt.internal.junit.model.TestElement;
import org.eclipse.core.runtime.Assert;
import org.eclipse.swt.SWTError;
import org.eclipse.swt.dnd.Clipboard;
import org.eclipse.swt.dnd.DND;
import org.eclipse.swt.dnd.TextTransfer;
import org.eclipse.swt.dnd.Transfer;
import org.eclipse.ui.actions.SelectionListenerAction;
/**
* Used by TraceCopyAction and MessageCopyAction to copy the failure trace, or only its
* first line, to the clipboard.
*/
public abstract class AbstractTraceAction extends SelectionListenerAction {
private FailureTrace fView;
private final Clipboard fClipboard;
private TestElement fTestElement;
/**
* Constructor for CopyTraceAction.
* @param view
* @param clipboard
*/
public AbstractTraceAction(FailureTrace view, Clipboard clipboard, String title) {
super(title);
Assert.isNotNull(clipboard);
fView= view;
fClipboard= clipboard;
}
public void transfer(String copyMe) {
String trace = copyMe;
String source;
if (trace == null && fTestElement != null) {
source = fTestElement.getTestName();
} else {
source = convertLineTerminators(trace);
}
if (source == null || source.length() == 0)
return;
TextTransfer plainTextTransfer = TextTransfer.getInstance();
try{
fClipboard.setContents(
new String[]{ convertLineTerminators(source) },
new Transfer[]{ plainTextTransfer });
} catch (SWTError e){
if (e.code != DND.ERROR_CANNOT_SET_CLIPBOARD)
throw e;
}
}
public void handleTestSelected(TestElement test) {
fTestElement= test;
}
public FailureTrace getFView() {
return fView;
}
private String convertLineTerminators(String in) {
StringWriter stringWriter= new StringWriter();
PrintWriter printWriter= new PrintWriter(stringWriter);
StringReader stringReader= new StringReader(in);
BufferedReader bufferedReader= new BufferedReader(stringReader);
String line;
try {
while ((line= bufferedReader.readLine()) != null) {
printWriter.println(line);
}
} catch (IOException e) {
return in; // return the trace unfiltered
}
return stringWriter.toString();
}
}