X-Git-Url: http://git.jetbrains.org/?p=idea%2Fcommunity.git;a=blobdiff_plain;f=platform%2Flang-impl%2Fsrc%2Fcom%2Fintellij%2FtestIntegration%2FTestFinderHelper.java;h=8c3e1d7900449cdf4356050db31ff912e1567acb;hp=ed7cafc6c28802cb9ff0027963449dc8941cb304;hb=03ea364c4c66158e69e3cefff24eb1bbec8eceba;hpb=3057f56cae25b437f742a34b2f576b0ddc3685c5

diff --git a/platform/lang-impl/src/com/intellij/testIntegration/TestFinderHelper.java b/platform/lang-impl/src/com/intellij/testIntegration/TestFinderHelper.java
index ed7cafc..8c3e1d7 100644
--- a/platform/lang-impl/src/com/intellij/testIntegration/TestFinderHelper.java
+++ b/platform/lang-impl/src/com/intellij/testIntegration/TestFinderHelper.java
@@ -40,8 +40,7 @@ public class TestFinderHelper {
   public static Collection<PsiElement> findTestsForClass(@NotNull final PsiElement element) {
     Collection<PsiElement> result = new LinkedHashSet<>();
     for (TestFinder each : getFinders()) {
-      final PsiElement selectedElement = each.findSelectedElement(element);
-      if (selectedElement != null) result.addAll(each.findTestsForClass(selectedElement));
+      result.addAll(each.findTestsForClass(element));
     }
     return result;
   }
@@ -49,8 +48,7 @@ public class TestFinderHelper {
   public static Collection<PsiElement> findClassesForTest(@NotNull final PsiElement element) {
     Collection<PsiElement> result = new LinkedHashSet<>();
     for (TestFinder each : getFinders()) {
-      final PsiElement selectedElement = each.findSelectedElement(element);
-      if (selectedElement != null) result.addAll(each.findClassesForTest(selectedElement));
+      result.addAll(each.findClassesForTest(element));
     }
     return result;
   }