fix appcode tests
[idea/community.git] / platform / testFramework / src / com / intellij / idea / IdeaTestApplication.java
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.idea;
17
18 import com.intellij.ide.plugins.PluginManagerCore;
19 import com.intellij.openapi.Disposable;
20 import com.intellij.openapi.actionSystem.DataProvider;
21 import com.intellij.openapi.application.Application;
22 import com.intellij.openapi.application.ApplicationManager;
23 import com.intellij.openapi.application.ex.ApplicationEx;
24 import com.intellij.openapi.application.ex.ApplicationManagerEx;
25 import com.intellij.openapi.util.Disposer;
26 import com.intellij.testFramework.PlatformTestCase;
27 import org.jetbrains.annotations.Nullable;
28
29 import javax.swing.*;
30 import java.lang.reflect.InvocationTargetException;
31
32 public class IdeaTestApplication extends CommandLineApplication implements Disposable {
33   private DataProvider myDataContext;
34
35   private IdeaTestApplication() {
36     super(true, true, true);
37   }
38
39   public void setDataProvider(@Nullable DataProvider dataContext) {
40     myDataContext = dataContext;
41   }
42
43   @Override
44   @Nullable
45   public Object getData(String dataId) {
46     return myDataContext == null ? null : myDataContext.getData(dataId);
47   }
48
49   public static IdeaTestApplication getInstance() {
50     return getInstance(null);
51   }
52
53   public static synchronized IdeaTestApplication getInstance(@Nullable final String configPath) {
54     if (ourInstance == null) {
55       PlatformTestCase.doAutodetectPlatformPrefix();
56       new IdeaTestApplication();
57       PluginManagerCore.getPlugins();
58       final ApplicationEx app = ApplicationManagerEx.getApplicationEx();
59       if (app.isDispatchThread()) {
60         app.load(configPath);
61       }
62       else {
63         try {
64           SwingUtilities.invokeAndWait(new Runnable() {
65             @Override
66             public void run() {
67               app.load(configPath);
68             }
69           });
70         }
71         catch (InterruptedException e) {
72           throw new RuntimeException(e);
73         }
74         catch (InvocationTargetException e) {
75           throw new RuntimeException(e);
76         }
77       }
78     }
79     return (IdeaTestApplication)ourInstance;
80   }
81
82   @Override
83   public void dispose() {
84     disposeInstance();
85   }
86
87   private static void disposeInstance() {
88     if (ourInstance == null) return;
89     Application applicationEx = ApplicationManager.getApplication();
90     if (applicationEx != null) {
91       Disposer.dispose(applicationEx);
92       //ApplicationManagerEx.setApplication(null); it will set automatically back to null
93     }
94     ourInstance = null;
95   }
96 }