junit: ensure isTest check is performed (IDEA-172627)

This commit is contained in:
Anna Kozlova
2017-05-10 13:09:56 +03:00
parent 267e87add3
commit be01c64a9e
@@ -434,7 +434,7 @@ public class JUnitUtil {
PsiElement element = location.getPsiElement();
if (element instanceof PsiClassOwner) {
PsiClass[] classes = ((PsiClassOwner)element).getClasses();
if (classes.length == 1) return classes[0];
if (classes.length == 1 && isTestClass(classes[0], false, true)) return classes[0];
}
return null;
}