From 4a9e24c4aef664e28043b94b602f0a39153617b6 Mon Sep 17 00:00:00 2001
From: anna <anna.kozlova@jetbrains.com>
Date: Wed, 26 May 2010 13:01:07 +0400
Subject: [PATCH] for non-java files

---
 .../impl/src/com/intellij/execution/junit/JUnitUtil.java      | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java b/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java
index e6a6fd806ee2..34e5e88d8f5d 100644
--- a/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java
+++ b/java/execution/impl/src/com/intellij/execution/junit/JUnitUtil.java
@@ -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;