Set default VM for JavaSE-9 execution environment during module tests

See: #4421

Author: trancexpress

org.eclipse.jdt.core.tests.model/src/org/eclipse/jdt/core/tests/model/ModuleBuilderTests.java

@@ -34,6 +34,7 @@
 import org.eclipse.core.runtime.IPath;
 import org.eclipse.core.runtime.IStatus;
 import org.eclipse.core.runtime.Path;
+import org.eclipse.core.runtime.Status;
 import org.eclipse.jdt.core.*;
 import org.eclipse.jdt.core.dom.AST;
 import org.eclipse.jdt.core.dom.CompilationUnit;
@@ -45,8 +46,15 @@
 import org.eclipse.jdt.internal.core.ClasspathEntry;
 import org.eclipse.jdt.internal.core.builder.ClasspathJrt;
 import org.eclipse.jdt.internal.core.util.Messages;
+import org.eclipse.jdt.launching.IVMInstall;
+import org.eclipse.jdt.launching.JavaRuntime;
+import org.eclipse.jdt.launching.environments.IExecutionEnvironment;
+import org.eclipse.jdt.launching.environments.IExecutionEnvironmentsManager;
 
 public class ModuleBuilderTests extends ModifyingResourceTests {
+
+	private static final String JAVASE_9 = "JavaSE-9";
+
 	public ModuleBuilderTests(String name) {
 		super(name);
 	}
@@ -691,6 +699,7 @@ public void _test019() throws CoreException, IOException {
 	}
 	public void testConvertToModule() throws CoreException, IOException {
 		Hashtable<String, String> javaCoreOptions = JavaCore.getOptions();
+		IVMInstall vm = prepareExecutionEnvironment(JAVASE_9);
 		try {
 			IJavaProject project = setUpJavaProject("ConvertToModule");
 			Map<String, String> options = new HashMap<>();
@@ -721,10 +730,12 @@ else if (isJRE10)
 				assertStringsEqual("incorrect result", new String[]{"java.desktop", "java.rmi", "java.sql"}, modules);
 		} finally {
 			this.deleteProject("ConvertToModule");
+			restoreExecutionEnvironment(JAVASE_9, vm);
 			 JavaCore.setOptions(javaCoreOptions);
 		}
 	}
 	public void testConvertToModuleWithRelease9() throws CoreException, IOException {
+		IVMInstall vm = prepareExecutionEnvironment(JAVASE_9);
 		Hashtable<String, String> javaCoreOptions = JavaCore.getOptions();
 		try {
 			IJavaProject project = setUpJavaProject("ConvertToModule");
@@ -756,6 +767,7 @@ else if (isJRE10)
 				assertStringsEqual("incorrect result", new String[]{"java.desktop", "java.rmi", "java.sql"}, modules);
 		} finally {
 			this.deleteProject("ConvertToModule");
+			restoreExecutionEnvironment(JAVASE_9, vm);
 			 JavaCore.setOptions(javaCoreOptions);
 		}
 	}
@@ -9136,4 +9148,25 @@ protected void sortMarkers(IMarker[] markers) {
 		Arrays.sort(markers, Comparator.comparingInt((IMarker a) -> a.getAttribute(IMarker.CHAR_START, 0))
 									   .thenComparing((IMarker a) -> a.getAttribute(IMarker.MESSAGE, "")));
 	}
+
+	private IVMInstall prepareExecutionEnvironment(String environmentId) {
+		IVMInstall vm = JavaRuntime.getDefaultVMInstall();
+		IExecutionEnvironment environment = getExecutionEnvironment(environmentId);
+		IVMInstall defaultVM9 = environment.getDefaultVM();
+		environment.setDefaultVM(vm);
+		JavaCore.getPlugin().getLog().log(Status.info("Set VM \"" + vm.getName() + "\" for execution environments: " + environment.getId()));
+		return defaultVM9;
+	}
+
+	private void restoreExecutionEnvironment(String environmentId, IVMInstall defaultVM) {
+		IExecutionEnvironment environment = getExecutionEnvironment(environmentId);
+		environment.setDefaultVM(defaultVM);
+		JavaCore.getPlugin().getLog().log(Status.info("Restored default VM for execution environment: " + environment.getId()));
+	}
+
+	private static IExecutionEnvironment getExecutionEnvironment(String id) {
+		IExecutionEnvironmentsManager manager = JavaRuntime.getExecutionEnvironmentsManager();
+		IExecutionEnvironment[] environments = manager.getExecutionEnvironments();
+		return Arrays.stream(environments).filter(e -> id.equals(e.getId())).findFirst().orElseThrow();
+	}
 }