5cea358ef6cf471912b4d658294596042c001d38
[idea/community.git] / platform / testFramework / src / com / intellij / testFramework / EdtTestUtil.kt
1 /*
2  * Copyright 2000-2015 JetBrains s.r.o.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 package com.intellij.testFramework
17
18 import com.intellij.openapi.application.ApplicationManager
19 import com.intellij.openapi.application.ModalityState
20 import com.intellij.util.ThrowableRunnable
21 import org.jetbrains.annotations.TestOnly
22 import java.lang.reflect.InvocationTargetException
23 import javax.swing.SwingUtilities
24
25 class EdtTestUtil {
26   companion object {
27     @TestOnly @JvmStatic fun runInEdtAndWait(runnable: ThrowableRunnable<Throwable>) {
28       runInEdtAndWait { runnable.run() }
29     }
30
31     @TestOnly @JvmStatic fun runInEdtAndWait(runnable: Runnable) {
32       runInEdtAndWait { runnable.run() }
33     }
34   }
35 }
36
37 @TestOnly
38 fun runInEdtAndWait(runnable: () -> Unit) {
39   val application = ApplicationManager.getApplication()
40   if (application != null) {
41     application.invokeAndWait(runnable, ModalityState.defaultModalityState())
42     return
43   }
44
45   if (SwingUtilities.isEventDispatchThread()) {
46     runnable()
47   }
48   else {
49     try {
50       SwingUtilities.invokeAndWait(runnable)
51     }
52     catch (e: InvocationTargetException) {
53       throw e.cause ?: e
54     }
55   }
56 }