From f711edc17b47fbb5e78966005a57168e952cef0d Mon Sep 17 00:00:00 2001
From: "Ilya.Kazakevich" <Ilya.Kazakevich@jetbrains.com>
Date: Tue, 27 Mar 2018 01:02:52 +0300
Subject: [PATCH] test fix: use correct modality state

---
 .../com/jetbrains/env/python/dotNet/SkeletonTestTask.java     | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java b/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java
index 1cc70aedf04e..ea00031c55d3 100644
--- a/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java
+++ b/python/testSrc/com/jetbrains/env/python/dotNet/SkeletonTestTask.java
@@ -116,7 +116,7 @@ class SkeletonTestTask extends PyExecutionFixtureTestTask {
     myFixture.enableInspections(PyUnresolvedReferencesInspection.class); // This inspection should suggest us to generate stubs
 
 
-    UIUtil.invokeAndWaitIfNeeded((Runnable)() -> {
+    ApplicationManager.getApplication().invokeAndWait(() -> {
       PsiDocumentManager.getInstance(myFixture.getProject()).commitAllDocuments();
       final String intentionName = PyBundle.message("sdk.gen.stubs.for.binary.modules", myUseQuickFixWithThisModuleOnly);
       IntentionAction intention = myFixture.findSingleIntention(intentionName);
@@ -132,7 +132,7 @@ class SkeletonTestTask extends PyExecutionFixtureTestTask {
                         Matchers.instanceOf(GenerateBinaryStubsFix.class));
       final Task fixTask = ((GenerateBinaryStubsFix)quickFix).getFixTask(myFixture.getFile());
       fixTask.run(new AbstractProgressIndicatorBase());
-    });
+    }, ModalityState.defaultModalityState());
 
     FileUtil.copy(skeletonFile, new File(myFixture.getTempDirPath(), skeletonFile.getName()));
     if (myExpectedSkeletonFile != null) {