suggest to run tests with different module: also for testng

2026-09-27 10:03:11 +07:00 · 2015-04-07 11:25:44 +02:00
parent 3d418054d5
commit 2868999097
3 changed files with 156 additions and 144 deletions
@@ -0,0 +1,128 @@
+/*
+ * Copyright 2000-2015 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.intellij.execution.testframework;
+
+import com.intellij.execution.CommonJavaRunConfigurationParameters;
+import com.intellij.execution.ExecutionException;
+import com.intellij.execution.Executor;
+import com.intellij.execution.configurations.JavaRunConfigurationModule;
+import com.intellij.execution.configurations.ModuleBasedConfiguration;
+import com.intellij.execution.executors.DefaultDebugExecutor;
+import com.intellij.execution.executors.DefaultRunExecutor;
+import com.intellij.execution.runners.ExecutionEnvironmentBuilder;
+import com.intellij.openapi.diagnostic.Logger;
+import com.intellij.openapi.module.Module;
+import com.intellij.openapi.module.ModuleManager;
+import com.intellij.openapi.module.ModuleUtilCore;
+import com.intellij.openapi.project.Project;
+import com.intellij.openapi.ui.MessageType;
+import com.intellij.openapi.ui.popup.Balloon;
+import com.intellij.openapi.util.text.StringUtil;
+import com.intellij.openapi.wm.ToolWindowId;
+import com.intellij.openapi.wm.ToolWindowManager;
+import com.intellij.psi.JavaPsiFacade;
+import com.intellij.psi.PsiDirectory;
+import com.intellij.psi.PsiPackage;
+import com.intellij.ui.HyperlinkAdapter;
+import com.intellij.util.Function;
+
+import javax.swing.event.HyperlinkEvent;
+import java.util.HashSet;
+import java.util.Set;
+
+public class ResetConfigurationModuleAdapter extends HyperlinkAdapter {
+  private static final Logger LOG = Logger.getInstance("#" + ResetConfigurationModuleAdapter.class);
+  private final Project myProject;
+  private final boolean myIsDebug;
+  private final ToolWindowManager myToolWindowManager;
+  private final String myTestRunDebugId;
+  private final ModuleBasedConfiguration myConfiguration;
+
+  public ResetConfigurationModuleAdapter(ModuleBasedConfiguration configuration, final Project project,
+                                         final boolean isDebug,
+                                         final ToolWindowManager toolWindowManager,
+                                         final String testRunDebugId) {
+    myProject = project;
+    myIsDebug = isDebug;
+    myToolWindowManager = toolWindowManager;
+    myTestRunDebugId = testRunDebugId;
+    myConfiguration = configuration;
+  }
+
+  public static
+  <T extends ModuleBasedConfiguration<JavaRunConfigurationModule> & CommonJavaRunConfigurationParameters>
+  boolean tryWithAnotherModule(T configuration, boolean isDebug) {
+    final String packageName = configuration.getPackage();
+    if (packageName == null) return false;
+    final Project project = configuration.getProject();
+    final PsiPackage aPackage = JavaPsiFacade.getInstance(project).findPackage(packageName);
+    if (aPackage == null) return false;
+    final Module module = configuration.getConfigurationModule().getModule();
+    if (module == null) return false;
+    final Set<Module> modulesWithPackage = new HashSet<Module>();
+    final PsiDirectory[] directories = aPackage.getDirectories();
+    for (PsiDirectory directory : directories) {
+      final Module currentModule = ModuleUtilCore.findModuleForFile(directory.getVirtualFile(), project);
+      if (module != currentModule && currentModule != null) {
+        modulesWithPackage.add(currentModule);
+      }
+    }
+    if (!modulesWithPackage.isEmpty()) {
+      final String testRunDebugId = isDebug ? ToolWindowId.DEBUG : ToolWindowId.RUN;
+      final ToolWindowManager toolWindowManager = ToolWindowManager.getInstance(project);
+      final Function<Module, String> moduleNameRef = new Function<Module, String>() {
+        @Override
+        public String fun(Module module) {
+          final String moduleName = module.getName();
+          return "<a href=\"" + moduleName + "\">" + moduleName + "</a>";
+        }
+      };
+      String message = "Tests were not found in module \"" + module.getName() + "\".\n" +
+                       "Use ";
+      if (modulesWithPackage.size() == 1) {
+        message += "module \"" + moduleNameRef.fun(modulesWithPackage.iterator().next()) + "\" ";
+      }
+      else {
+        message += "one of\n" + StringUtil.join(modulesWithPackage, moduleNameRef, "\n") + "\n";
+      }
+      message += "instead";
+      toolWindowManager.notifyByBalloon(testRunDebugId, MessageType.WARNING, message, null,
+                                        new ResetConfigurationModuleAdapter(configuration, project, isDebug, toolWindowManager, testRunDebugId));
+      return true;
+    }
+    return false;
+  }
+
+  @Override
+  protected void hyperlinkActivated(HyperlinkEvent e) {
+    final Module moduleByName = ModuleManager.getInstance(myProject).findModuleByName(e.getDescription());
+    if (moduleByName != null) {
+      myConfiguration.getConfigurationModule().setModule(moduleByName);
+      try {
+        Executor executor = myIsDebug ? DefaultDebugExecutor.getDebugExecutorInstance()
+                                      : DefaultRunExecutor.getRunExecutorInstance();
+        ExecutionEnvironmentBuilder.create(myProject, executor, myConfiguration).contentToReuse(null).buildAndExecute();
+        Balloon balloon = myToolWindowManager.getToolWindowBalloon(myTestRunDebugId);
+        if (balloon != null) {
+          balloon.hide();
+        }
+      }
+      catch (ExecutionException e1) {
+        LOG.error(e1);
+      }
+    }
+  }
+}