test for framework support provider added

2026-09-18 09:34:34 +07:00 · 2010-07-15 11:06:04 +04:00
parent 504ec74782
commit 6f8694707c
1 changed files with 92 additions and 0 deletions
@@ -0,0 +1,92 @@
+package com.intellij.ide.util.frameworkSupport;
+
+import com.intellij.facet.Facet;
+import com.intellij.facet.FacetManager;
+import com.intellij.facet.FacetTypeId;
+import com.intellij.facet.ui.FacetBasedFrameworkSupportProvider;
+import com.intellij.ide.util.frameworkSupport.FrameworkSupportConfigurable;
+import com.intellij.ide.util.frameworkSupport.FrameworkSupportModel;
+import com.intellij.ide.util.frameworkSupport.FrameworkSupportProvider;
+import com.intellij.ide.util.newProjectWizard.impl.FrameworkSupportCommunicator;
+import com.intellij.ide.util.newProjectWizard.impl.FrameworkSupportModelImpl;
+import com.intellij.openapi.roots.ModifiableRootModel;
+import com.intellij.openapi.roots.ModuleRootManager;
+import com.intellij.openapi.vfs.VirtualFile;
+import com.intellij.testFramework.IdeaTestCase;
+import com.intellij.testFramework.PsiTestUtil;
+import org.jetbrains.annotations.NotNull;
+
+import java.io.IOException;
+import java.util.*;
+
+/**
+ * @author nik
+ */
+public abstract class FrameworkSupportProviderTestCase extends IdeaTestCase {
+  private FrameworkSupportModel myFrameworkSupportModel;
+  private Map<FrameworkSupportProvider, FrameworkSupportConfigurable> myConfigurables;
+  private Set<FrameworkSupportProvider> mySelected;
+
+  @Override
+  protected void setUp() throws Exception {
+    super.setUp();
+    myFrameworkSupportModel = new FrameworkSupportModelImpl(getProject(), null);
+    myConfigurables = new HashMap<FrameworkSupportProvider, FrameworkSupportConfigurable>();
+    mySelected = new HashSet<FrameworkSupportProvider>();
+  }
+
+  protected void addSupport() throws IOException {
+    final VirtualFile root = getVirtualFile(createTempDir("contentRoot"));
+    PsiTestUtil.addContentRoot(myModule, root);
+    final ModifiableRootModel model = ModuleRootManager.getInstance(myModule).getModifiableModel();
+    List<FrameworkSupportConfigurable> selectedConfigurables = new ArrayList<FrameworkSupportConfigurable>();
+    for (FrameworkSupportProvider provider : mySelected) {
+      final FrameworkSupportConfigurable configurable = myConfigurables.get(provider);
+      configurable.addSupport(myModule, model, null);
+      selectedConfigurables.add(configurable);
+    }
+    for (FrameworkSupportCommunicator communicator : FrameworkSupportCommunicator.EP_NAME.getExtensions()) {
+      communicator.onFrameworkSupportAdded(myModule, model, selectedConfigurables, myFrameworkSupportModel);
+    }
+    model.commit();
+  }
+
+  protected void selectFramework(@NotNull FacetTypeId<?> id) {
+    selectFramework(FacetBasedFrameworkSupportProvider.getProviderId(id));
+  }
+
+  protected void selectFramework(@NotNull String id) {
+    for (FrameworkSupportProvider provider : FrameworkSupportProvider.EXTENSION_POINT.getExtensions()) {
+      if (provider.getId().equals(id)) {
+        selectFramework(provider);
+        return;
+      }
+    }
+    fail("Framework provider with id='" + id + "' not found");
+  }
+
+  protected void selectFramework(@NotNull FrameworkSupportProvider provider) {
+    final FrameworkSupportConfigurable configurable = getOrCreateConfigurable(provider);
+    configurable.onFrameworkSelectionChanged(true);
+    mySelected.add(provider);
+  }
+
+  private FrameworkSupportConfigurable getOrCreateConfigurable(FrameworkSupportProvider provider) {
+    FrameworkSupportConfigurable configurable = myConfigurables.get(provider);
+    if (configurable == null) {
+      configurable = provider.createConfigurable(myFrameworkSupportModel);
+      myConfigurables.put(provider, configurable);
+    }
+    return configurable;
+  }
+
+  protected <F extends Facet> F getFacet(FacetTypeId<F> id) {
+    final F facet = FacetManager.getInstance(myModule).getFacetByType(id);
+    assertNotNull(id + " facet not found", facet);
+    return facet;
+  }
+
+  protected VirtualFile getContentRoot() {
+    return ModuleRootManager.getInstance(myModule).getContentRoots()[0];
+  }
+}