From 8a4282313fd94744d07f75e58faabcd74836f399 Mon Sep 17 00:00:00 2001
From: Eugene Petrenko <eugene.petrenko@jetbrains.com>
Date: Thu, 16 Jan 2020 13:09:36 +0100
Subject: [PATCH] fix tests in SdkSetupNotificationTest

GitOrigin-RevId: 41829da4813944554078e5e2970cf3241e2985ea
---
 .../daemon/impl/SdkSetupNotificationTestBase.java            | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/impl/SdkSetupNotificationTestBase.java b/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/impl/SdkSetupNotificationTestBase.java
index 8c9c5b510629..e5e43b20b181 100644
--- a/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/impl/SdkSetupNotificationTestBase.java
+++ b/java/java-tests/testSrc/com/intellij/java/codeInsight/daemon/impl/SdkSetupNotificationTestBase.java
@@ -7,6 +7,7 @@ import com.intellij.idea.TestFor;
 import com.intellij.openapi.Disposable;
 import com.intellij.openapi.application.ReadAction;
 import com.intellij.openapi.application.WriteAction;
+import com.intellij.openapi.application.impl.NonBlockingReadActionImpl;
 import com.intellij.openapi.fileEditor.FileEditor;
 import com.intellij.openapi.fileEditor.ex.FileEditorManagerEx;
 import com.intellij.openapi.projectRoots.ProjectJdkTable;
@@ -21,6 +22,7 @@ import com.intellij.testFramework.fixtures.JavaCodeInsightFixtureTestCase;
 import com.intellij.testFramework.fixtures.JavaCodeInsightTestFixture;
 import com.intellij.ui.EditorNotificationPanel;
 import com.intellij.ui.EditorNotificationsImpl;
+import com.intellij.util.ui.UIUtil;
 import org.jetbrains.annotations.NotNull;
 import org.jetbrains.annotations.Nullable;
 
@@ -58,6 +60,9 @@ public abstract class SdkSetupNotificationTestBase extends JavaCodeInsightFixtur
   public static EditorNotificationPanel runOnText(@NotNull JavaCodeInsightTestFixture fixture,
                                                   @NotNull String fileName,
                                                   @NotNull String fileText) {
+    NonBlockingReadActionImpl.waitForAsyncTaskCompletion();
+    UIUtil.dispatchAllInvocationEvents();
+
     final PsiFile psiFile = fixture.configureByText(fileName, fileText);
     FileEditorManagerEx fileEditorManager = FileEditorManagerEx.getInstanceEx(fixture.getProject());
     VirtualFile virtualFile = psiFile.getVirtualFile();