Revert: RUBY-5449 find class related tests and vise versa for appropriate element...
[idea/community.git] / platform / lang-impl / src / com / intellij / testIntegration / TestFinderHelper.java
index ed7cafc6c28802cb9ff0027963449dc8941cb304..8c3e1d7900449cdf4356050db31ff912e1567acb 100644 (file)
@@ -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;
   }