openide/plugins/testng/testSources/com/theoryinpractice/testng/intention/AddTestNGJarFixTest.java

// Copyright 2000-2023 JetBrains s.r.o. and contributors. Use of this source code is governed by the Apache 2.0 license.

package com.theoryinpractice.testng.intention;

import com.intellij.codeInsight.intention.IntentionAction;
import com.intellij.openapi.application.PluginPathManager;
import com.intellij.openapi.util.Comparing;
import com.intellij.pom.java.LanguageLevel;
import com.intellij.psi.JavaPsiFacade;
import com.intellij.testFramework.IdeaTestUtil;
import com.intellij.testFramework.builders.JavaModuleFixtureBuilder;
import com.intellij.testFramework.fixtures.*;
import com.intellij.util.ui.UIUtil;
import org.jetbrains.annotations.NonNls;
import org.testng.Assert;
import org.testng.annotations.AfterMethod;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.DataProvider;
import org.testng.annotations.Test;

import java.util.List;

@Test
public class AddTestNGJarFixTest {
  protected CodeInsightTestFixture myFixture;

  private LanguageLevel myLanguageLevel;

  @BeforeMethod
  public void setUp() {
    UIUtil.invokeAndWaitIfNeeded(new Runnable() {
      @Override
      public void run() {
        try {
          final IdeaTestFixtureFactory fixtureFactory = IdeaTestFixtureFactory.getFixtureFactory();
          final TestFixtureBuilder<IdeaProjectTestFixture> testFixtureBuilder = fixtureFactory.createFixtureBuilder(getClass().getSimpleName());
          myFixture = JavaTestFixtureFactory.getFixtureFactory().createCodeInsightFixture(testFixtureBuilder.getFixture());
          final String dataPath = PluginPathManager.getPluginHomePath("testng") + "/testData";
          myFixture.setTestDataPath(dataPath);
          final JavaModuleFixtureBuilder builder = testFixtureBuilder.addModule(JavaModuleFixtureBuilder.class);

          builder.addContentRoot(myFixture.getTempDirPath()).addSourceRoot("");
          //    builder.addContentRoot(dataPath);
          builder.setMockJdkLevel(JavaModuleFixtureBuilder.MockJdkLevel.jdk15);
          myFixture.setUp();
          final JavaPsiFacade facade = JavaPsiFacade.getInstance(myFixture.getProject());
          myLanguageLevel = IdeaTestUtil.setProjectLanguageLevel(facade.getProject(), LanguageLevel.JDK_1_5);
        }
        catch (Exception e) {
          throw new RuntimeException(e);
        }
      }
    });
  }


  @AfterMethod
  public void tearDown() {
    UIUtil.invokeAndWaitIfNeeded(() -> {
      try {
        IdeaTestUtil.setProjectLanguageLevel(myFixture.getProject(), myLanguageLevel);
        myFixture.tearDown();
        myFixture = null;
      }
      catch (Exception e) {
        throw new RuntimeException(e);
      }
    });
  }

  @NonNls
  @DataProvider
  public Object[][] data() {
    return new String[][]{new String[]{"InsideReference"}, new String[]{"AfterReference"}};
  }

  @Test(dataProvider = "data")
  public void doTest(final String testName) {
    UIUtil.invokeAndWaitIfNeeded(() -> {
      try {
        IntentionAction resultAction = null;
        final List<IntentionAction> actions = myFixture.getAvailableIntentions("intention/testNGJar" + "/" + testName + ".java");
        for (IntentionAction action : actions) {
          if (Comparing.strEqual(action.getText(), "Add testng.jar to classpath")) {
            resultAction = action;
            break;
          }
        }
        Assert.assertNotNull(resultAction);
      }
      catch (Exception e) {
        throw new RuntimeException(e);
      }
    });
  }


}