From 1b447a02316a7dfae80fd409d7584052d13cd6f1 Mon Sep 17 00:00:00 2001
From: Nikolay Krasko <nikolay.krasko@jetbrains.com>
Date: Tue, 19 Sep 2017 11:23:11 +0300
Subject: [PATCH] Execute action without writeActionPriority in tests

---
 idea/src/org/jetbrains/kotlin/idea/ProgressUtil.kt | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/idea/src/org/jetbrains/kotlin/idea/ProgressUtil.kt b/idea/src/org/jetbrains/kotlin/idea/ProgressUtil.kt
index f956f593b05..e3d51869b52 100644
--- a/idea/src/org/jetbrains/kotlin/idea/ProgressUtil.kt
+++ b/idea/src/org/jetbrains/kotlin/idea/ProgressUtil.kt
@@ -41,16 +41,20 @@ fun <T : Any> runInReadActionWithWriteActionPriorityWithPCE(f: () -> T): T {
 
 fun <T : Any> runInReadActionWithWriteActionPriority(f: () -> T): T? {
     var r: T? = null
-    val complete = ProgressIndicatorUtils.runInReadActionWithWriteActionPriority {
+    if (!with(ApplicationManager.getApplication()) { isDispatchThread && isUnitTestMode }) {
+        val complete = ProgressIndicatorUtils.runInReadActionWithWriteActionPriority {
+            r = f()
+        }
+
+        if (!complete) return null
+    }
+    else {
         r = f()
     }
 
-    if (!complete) return null
     return r
 }
 
-
-
 fun <T : Any> Project.runSynchronouslyWithProgress(progressTitle: String, canBeCanceled: Boolean, action: () -> T): T? {
     var result: T? = null
     ProgressManager.getInstance().runProcessWithProgressSynchronously({ result = action() }, progressTitle, canBeCanceled, this)