Fix class cast exception

raniejade · Dec 3, 2016 · 423dff3 · 423dff3
1 parent 2ccab27
commit 423dff3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/plugin/src/main/kotlin/org/jetbrains/spek/idea/SpekUtils.kt b/plugin/src/main/kotlin/org/jetbrains/spek/idea/SpekUtils.kt
@@ -50,7 +50,7 @@ object SpekUtils {
         val calleeExpression = callExpression.calleeExpression!! as KtNameReferenceExpression
         val resolved = calleeExpression.mainReference.resolve()
 
-        if (resolved != null && resolved is KtNamedFunction) {
+        if (resolved != null && resolved is KtNamedFunction && isContainedWithinLambda(callExpression)) {
             if (lambda != null && parameters.size == 2 && isDslExtension(resolved)) {
                 val desc = parameters.first().children.firstOrNull()
                 if (desc != null && desc is KtStringTemplateExpression) {
@@ -150,6 +150,8 @@ object SpekUtils {
         return callExpression.parent.parent.parent as KtLambdaExpression
     }
 
+    fun isContainedWithinLambda(callExpression: KtCallExpression) = callExpression.parent.parent.parent is KtLambdaExpression
+
     fun isDslExtension(function: KtNamedFunction): Boolean {
         val receiverTypeReference = function.receiverTypeReference
         if (receiverTypeReference != null) {