From d0b3d35209d38b4349eeb3699f35ed8d087e5548 Mon Sep 17 00:00:00 2001
From: Dmitry Batkovich <dmitry.batkovich@jetbrains.com>
Date: Tue, 6 Feb 2018 16:36:03 +0300
Subject: [PATCH] test discovery: take only distinct modules

---
 .../testDiscovery/TestDiscoveryConfigurationProducer.java   | 2 +-
 .../testDiscovery/actions/DiscoveredTestsTree.java          | 6 ++++--
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/java/execution/impl/src/com/intellij/execution/testDiscovery/TestDiscoveryConfigurationProducer.java b/java/execution/impl/src/com/intellij/execution/testDiscovery/TestDiscoveryConfigurationProducer.java
index 821b00a86b5b..0fb16e69a8ea 100644
--- a/java/execution/impl/src/com/intellij/execution/testDiscovery/TestDiscoveryConfigurationProducer.java
+++ b/java/execution/impl/src/com/intellij/execution/testDiscovery/TestDiscoveryConfigurationProducer.java
@@ -141,7 +141,7 @@ public abstract class TestDiscoveryConfigurationProducer extends JavaRunConfigur
     };
   }
 
-  public static Module detectTargetModule(List<Module> survivedModules, Project project) {
+  public static Module detectTargetModule(Collection<Module> survivedModules, Project project) {
     ModuleManager moduleManager = ModuleManager.getInstance(project);
     final Set<Module> allModules = new HashSet<>(Arrays.asList(moduleManager.getModules()));
     survivedModules
diff --git a/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java b/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java
index 4feb44d28360..33e3e5fc265d 100644
--- a/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java
+++ b/java/execution/impl/src/com/intellij/execution/testDiscovery/actions/DiscoveredTestsTree.java
@@ -35,6 +35,7 @@ import javax.swing.tree.TreeModel;
 import javax.swing.tree.TreePath;
 import javax.swing.tree.TreeSelectionModel;
 import java.util.List;
+import java.util.Set;
 import java.util.stream.Collectors;
 
 class DiscoveredTestsTree extends Tree implements DataProvider {
@@ -97,11 +98,12 @@ class DiscoveredTestsTree extends Tree implements DataProvider {
     myModel.addTest(testClass, testMethod);
   }
 
-  public List<Module> getContainingModules() {
+  @NotNull
+  public Set<Module> getContainingModules() {
     return myModel.getTestClasses().stream()
                   .map(element -> ModuleUtilCore.findModuleForPsiElement(element))
                   .filter(module -> module != null)
-                  .collect(Collectors.toList());
+                  .collect(Collectors.toSet());
   }
 
   public PsiMethod[] getTestMethods() {