for non-java files

This commit is contained in:
anna
2010-05-26 13:28:40 +04:00
parent 42b987a6a2
commit 4a9e24c4ae
@@ -248,8 +248,8 @@ public class JUnitUtil {
if (isTestClass(classLocation)) return classLocation.getPsiElement();
}
PsiElement element = location.getPsiElement();
if (element instanceof PsiJavaFile) {
PsiClass[] classes = ((PsiJavaFile)element).getClasses();
if (element instanceof PsiClassOwner) {
PsiClass[] classes = ((PsiClassOwner)element).getClasses();
if (classes.length == 1) return classes[0];
}
return null;