From 7a7be0b70d9edb0169cb86594b95f1ed5a39b078 Mon Sep 17 00:00:00 2001 From: "Ilya.Kazakevich" Date: Mon, 10 Apr 2017 22:26:32 +0300 Subject: [PATCH] IronPython test fix --- .../com/jetbrains/env/python/dotNet/SkeletonTestTask.java | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java b/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java index a8860ab7756c..be4fe1d9e2de 100644 --- a/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java +++ b/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java @@ -2,6 +2,7 @@ package com.jetbrains.env.python.dotNet; import com.google.common.collect.Sets; import com.intellij.codeInsight.intention.IntentionAction; +import com.intellij.codeInsight.intention.IntentionActionDelegate; import com.intellij.codeInspection.LocalQuickFix; import com.intellij.codeInspection.ex.QuickFixWrapper; import com.intellij.openapi.application.ApplicationManager; @@ -119,7 +120,12 @@ class SkeletonTestTask extends PyExecutionFixtureTestTask { public void run() { PsiDocumentManager.getInstance(myFixture.getProject()).commitAllDocuments(); final String intentionName = PyBundle.message("sdk.gen.stubs.for.binary.modules", myUseQuickFixWithThisModuleOnly); - final IntentionAction intention = myFixture.findSingleIntention(intentionName); + IntentionAction intention = myFixture.findSingleIntention(intentionName); + + if (intention instanceof IntentionActionDelegate) { + intention = ((IntentionActionDelegate)intention).getDelegate(); + } + Assert.assertNotNull("No intention found to generate skeletons!", intention); Assert.assertThat("Intention should be quick fix to run", intention, Matchers.instanceOf(QuickFixWrapper.class)); final LocalQuickFix quickFix = ((QuickFixWrapper)intention).getFix();