Automatically save editor of compilation unit if dirty before running a test

Author: Hélios GILLES

org.moreunit.plugin/src/org/moreunit/handler/RunTestsActionExecutor.java

@@ -17,13 +17,15 @@
 import java.util.LinkedHashSet;
 
 import org.eclipse.core.resources.IFile;
+import org.eclipse.core.runtime.NullProgressMonitor;
 import org.eclipse.jdt.core.ICompilationUnit;
 import org.eclipse.jdt.core.IJavaElement;
 import org.eclipse.jdt.core.IJavaProject;
 import org.eclipse.jdt.core.IMember;
 import org.eclipse.jdt.core.IMethod;
 import org.eclipse.jdt.core.IType;
 import org.eclipse.jdt.core.JavaCore;
+import org.eclipse.jdt.internal.ui.javaeditor.EditorUtility;
 import org.eclipse.ui.IEditorPart;
 import org.moreunit.actions.RunTestAction;
 import org.moreunit.actions.RunTestFromCompilationUnitAction;
@@ -99,6 +101,7 @@ public void executeRunTestAction(ICompilationUnit compilationUnit, String launch
 
     private void executeRunAllTestsAction(ICompilationUnit compilationUnit, String launchMode)
     {
+        saveIfNeeded(compilationUnit);
         Jobs.waitForIndexExecuteAndRunInUI("Running tests ... ", () -> {
 
             Collection<IType> testCases = new LinkedHashSet<>();
@@ -135,6 +138,15 @@ private void executeRunAllTestsAction(ICompilationUnit compilationUnit, String l
         }, testCases -> runTests(testCases, launchMode));
     }
 
+    private void saveIfNeeded(ICompilationUnit compilationUnit)
+    {
+        IEditorPart editorPart = EditorUtility.isOpenInEditor(compilationUnit);
+        if(editorPart != null && editorPart.isDirty())
+        {
+            editorPart.doSave(new NullProgressMonitor());
+        }
+    }
+
     public void executeRunTestsOfSelectedMemberAction(IEditorPart editorPart, String launchMode)
     {
         ICompilationUnit compilationUnit = createCompilationUnitFrom(editorPart);
@@ -143,6 +155,7 @@ public void executeRunTestsOfSelectedMemberAction(IEditorPart editorPart, String
 
     private void executeRunTestsOfSelectedMemberAction(IEditorPart editorPart, ICompilationUnit compilationUnit, String launchMode)
     {
+        saveIfNeeded(compilationUnit);
         Jobs.waitForIndexExecuteAndRunInUI("Running tests ... ", () -> {
             Collection<IMember> testElements = new LinkedHashSet<>();
             IType selectedJavaType = compilationUnit.findPrimaryType();