check type of containing file in getTestClass()

2015-07-02 18:01:41 +02:00
parent 49714d0817
commit d29e63e2b8
1 changed files with 2 additions and 1 deletions
@@ -130,9 +130,10 @@ public class KotlinJUnitRunConfigurationProducer : RunConfigurationProducer<JUni
    }

    private fun getTestClass(leaf: PsiElement): PsiClass? {
+        val containingFile = leaf.getContainingFile() as? JetFile ?: return null
        var jetClass = leaf.getParentOfType<JetClass>(false)
        if (!jetClass.isJUnitTestClass()) {
-            jetClass = getTestClassInFile(leaf.getContainingFile() as JetFile)
+            jetClass = getTestClassInFile(containingFile)
        }
        if (jetClass != null) {
            return LightClassUtil.getPsiClass(jetClass)