From d29e63e2b8da1646d11d7ab64807938757a5c1fa Mon Sep 17 00:00:00 2001
From: Dmitry Jemerov <yole@jetbrains.com>
Date: Thu, 2 Jul 2015 18:01:41 +0200
Subject: [PATCH] check type of containing file in getTestClass()

---
 .../kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt b/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
index 015201848b5..e6e387ca2ea 100644
--- a/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/run/KotlinJUnitRunConfigurationProducer.kt
@@ -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)