smRunner/testSrc/com/intellij/execution/testframework/sm/SMLightFixtureTestCase.java

   1 package com.intellij.execution.testframework.sm;
   2
   3 import com.intellij.openapi.module.Module;
   4 import com.intellij.openapi.module.ModuleType;
   5 import com.intellij.openapi.project.Project;
   6 import com.intellij.openapi.projectRoots.Sdk;
   7 import com.intellij.openapi.roots.ModifiableRootModel;
   8 import com.intellij.psi.PsiFile;
   9 import com.intellij.testFramework.LightProjectDescriptor;
  10 import com.intellij.testFramework.UsefulTestCase;
  11 import com.intellij.testFramework.fixtures.CodeInsightTestFixture;
  12 import com.intellij.testFramework.fixtures.IdeaProjectTestFixture;
  13 import com.intellij.testFramework.fixtures.IdeaTestFixtureFactory;
  14 import com.intellij.testFramework.fixtures.TestFixtureBuilder;
  15 import com.intellij.testFramework.fixtures.impl.LightTempDirTestFixtureImpl;
  16 import org.jetbrains.annotations.Nullable;
  17
  18 import java.io.IOException;
  19
  20 /**
  21  * @author Roman.Chernyatchik
  22  */
  23 public class SMLightFixtureTestCase extends UsefulTestCase {
  24   protected static final LightProjectDescriptor ourDescriptor = new LightProjectDescriptor() {
  25     public ModuleType getModuleType() {
  26       return ModuleType.EMPTY;
  27     }
  28
  29     public Sdk getSdk() {
  30       return null;
  31     }
  32
  33     public void configureModule(Module module, ModifiableRootModel model) {
  34       //Do nothing
  35     }
  36   };
  37   protected CodeInsightTestFixture myFixture;
  38
  39   @Override
  40   protected void setUp() throws Exception {
  41     super.setUp();
  42
  43     final IdeaTestFixtureFactory factory = IdeaTestFixtureFactory.getFixtureFactory();
  44     final TestFixtureBuilder<IdeaProjectTestFixture> fixtureBuilder =
  45       factory.createLightFixtureBuilder(getProjectDescriptor());
  46
  47     final IdeaProjectTestFixture fixture = fixtureBuilder.getFixture();
  48     final LightTempDirTestFixtureImpl tempDirTestFixture = new LightTempDirTestFixtureImpl(true);
  49     myFixture = IdeaTestFixtureFactory.getFixtureFactory().createCodeInsightFixture(fixture, tempDirTestFixture);
  50     myFixture.setUp();
  51
  52     setupFixtureWhenInitialized();
  53   }
  54
  55   protected void setupFixtureWhenInitialized() throws IOException {
  56   }
  57
  58   @Override
  59   protected void tearDown() throws Exception {
  60     myFixture.tearDown();
  61     myFixture = null;
  62
  63     super.tearDown();
  64   }
  65
  66   @Nullable
  67   protected LightProjectDescriptor getProjectDescriptor() {
  68     return null;
  69   }
  70
  71   protected Project getProject() {
  72     return myFixture.getProject();
  73   }
  74
  75   protected void createAndAddFile(final String relativePath, final String text) throws IOException {
  76     final PsiFile psiFile = myFixture.addFileToProject(relativePath, text);
  77     myFixture.configureFromExistingVirtualFile(psiFile.getVirtualFile());
  78   }
  79 }