From 8e210803cd0e1b854a27283b2de834c28e946112 Mon Sep 17 00:00:00 2001
From: Will Hains <github@willhains.com>
Date: Sun, 19 Jan 2014 22:20:40 +0900
Subject: [PATCH] Patch submission for #14.

Define a custom problem marker type for test failures.

When a JUnitLoop test session starts, remove all test failure markers
from the workspace.

On each test failure, create a new test failure marker, which appears in
the Problems view and links to the line in the test case where the
failure occurred.
---
 de.devboost.eclipse.junitloop/plugin.xml      |  6 ++
 .../junitloop/JUnitLoopTestRunListener.java   | 56 ++++++++++++++++++-
 2 files changed, 61 insertions(+), 1 deletion(-)

diff --git a/de.devboost.eclipse.junitloop/plugin.xml b/de.devboost.eclipse.junitloop/plugin.xml
index dedb345..2d544e4 100644
--- a/de.devboost.eclipse.junitloop/plugin.xml
+++ b/de.devboost.eclipse.junitloop/plugin.xml
@@ -14,6 +14,12 @@
             class="de.devboost.eclipse.junitloop.JUnitLoopTestRunListener">
       </testRunListener>
    </extension>
+   <extension
+      id="de.devboost.eclipse.junitloop.testFailure"
+      point="org.eclipse.core.resources.markers"
+      name="Test Failure">
+      <super type="org.eclipse.core.resources.problemmarker"/>
+   </extension>
    <extension
          point="org.eclipse.ui.viewActions">
       <viewContribution
diff --git a/de.devboost.eclipse.junitloop/src/de/devboost/eclipse/junitloop/JUnitLoopTestRunListener.java b/de.devboost.eclipse.junitloop/src/de/devboost/eclipse/junitloop/JUnitLoopTestRunListener.java
index 3a2dc6d..25dae41 100644
--- a/de.devboost.eclipse.junitloop/src/de/devboost/eclipse/junitloop/JUnitLoopTestRunListener.java
+++ b/de.devboost.eclipse.junitloop/src/de/devboost/eclipse/junitloop/JUnitLoopTestRunListener.java
@@ -18,13 +18,17 @@
 import java.util.ArrayList;
 import java.util.List;
 
+import org.eclipse.core.resources.IMarker;
 import org.eclipse.core.resources.IResource;
+import org.eclipse.core.resources.ResourcesPlugin;
+import org.eclipse.core.runtime.CoreException;
 import org.eclipse.jdt.core.IJavaProject;
 import org.eclipse.jdt.core.IType;
 import org.eclipse.jdt.core.JavaModelException;
 import org.eclipse.jdt.junit.TestRunListener;
 import org.eclipse.jdt.junit.model.ITestCaseElement;
 import org.eclipse.jdt.junit.model.ITestElement;
+import org.eclipse.jdt.junit.model.ITestElement.FailureTrace;
 import org.eclipse.jdt.junit.model.ITestElement.Result;
 import org.eclipse.jdt.junit.model.ITestRunSession;
 
@@ -39,6 +43,8 @@
  */
 public class JUnitLoopTestRunListener extends TestRunListener {
 	
+	public static final String MARKER_TEST_FAILURE = "de.devboost.eclipse.junitloop.testFailure";
+	
 	private IJavaProject launchedProject;
 
 	@Override
@@ -46,6 +52,17 @@ public void sessionLaunched(ITestRunSession session) {
 		super.sessionLaunched(session);
 		launchedProject = session.getLaunchedProject();
 		JUnitLoopPlugin.getDefault().notifySessionLaunched();
+		
+		// Delete all test failure markers
+		try {
+			ResourcesPlugin
+					.getWorkspace()
+					.getRoot()
+					.deleteMarkers(MARKER_TEST_FAILURE, true,
+							IResource.DEPTH_INFINITE);
+		} catch (final CoreException e) {
+			JLoopPlugin.logError("Couldn't delete old warnings", e);
+		}
 	}
 
 	@Override
@@ -80,9 +97,11 @@ public void testCaseFinished(ITestCaseElement testCaseElement) {
 			return;
 		}
 		
+		// Add warning to Problems view
 		if (testResult == ITestElement.Result.FAILURE ||
 			testResult == ITestElement.Result.ERROR) {
 			failedTests.add(testClass);
+			writeMarkers(testCaseElement);
 		} else if (testResult == ITestElement.Result.OK) {
 			succeededTests.add(testClass);
 		}
@@ -91,7 +110,42 @@ public void testCaseFinished(ITestCaseElement testCaseElement) {
 		scheduler.addFailedTests(failedTests);
 		scheduler.addSucceededTests(succeededTests);
 	}
-
+	
+	private void writeMarkers(ITestCaseElement testCaseElement) {
+		final String testClassName = testCaseElement.getTestClassName();
+		final FailureTrace failureTrace = testCaseElement.getFailureTrace();
+		final String trace = failureTrace.getTrace();
+		String traceLine1 = null;
+		for (final String traceLine : trace.split("[\\n\\r]+")) {
+			if (traceLine1 == null) {
+				traceLine1 = traceLine;
+			}
+			if (traceLine.indexOf(testClassName) > 0) {
+				final String line = traceLine.substring(
+						traceLine.indexOf(':') + 1, traceLine.length() - 1);
+				try {
+					final IType testClassType = launchedProject
+							.findType(testClassName);
+					final IResource resource = testClassType.getResource();
+					
+					final IMarker marker = resource
+							.createMarker(MARKER_TEST_FAILURE);
+					marker.setAttribute(IMarker.MESSAGE, traceLine1);
+					marker.setAttribute(IMarker.SEVERITY,
+							IMarker.SEVERITY_WARNING);
+					marker.setAttribute(IMarker.LINE_NUMBER,
+							Integer.parseInt(line));
+					marker.setAttribute(IMarker.LOCATION, "line " + line);
+					marker.setAttribute(IMarker.TRANSIENT, true);
+					
+					break;
+				} catch (final Exception e) {
+					JLoopPlugin.logError("Couldn't write problem marker", e);
+				}
+			}
+		}
+	}
+	
 	private TestClass createTestClass(String testClassName) {
 		// create local copy to avoid potential multi-threading problems
 		IJavaProject localProject = launchedProject;