openide/java/java-tests/testSrc/com/intellij/java/codeInsight/navigation/FailedLineTest.java

// Copyright 2000-2020 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.intellij.java.codeInsight.navigation;

import com.intellij.codeInsight.daemon.impl.HighlightInfo;
import com.intellij.execution.TestStateStorage;
import com.intellij.execution.executors.DefaultDebugExecutor;
import com.intellij.execution.testframework.JavaTestLocator;
import com.intellij.execution.testframework.TestConsoleProperties;
import com.intellij.execution.testframework.TestFailedLineInspection;
import com.intellij.execution.testframework.sm.runner.MockRuntimeConfiguration;
import com.intellij.execution.testframework.sm.runner.SMTRunnerConsoleProperties;
import com.intellij.execution.testframework.sm.runner.SMTestProxy;
import com.intellij.execution.testframework.sm.runner.states.TestStateInfo;
import com.intellij.execution.testframework.sm.runner.ui.SMTRunnerConsoleView;
import com.intellij.execution.testframework.sm.runner.ui.SMTestRunnerResultsForm;
import com.intellij.execution.testframework.sm.runner.ui.TestStackTraceParser;
import com.intellij.lang.annotation.HighlightSeverity;
import com.intellij.openapi.application.ApplicationManager;
import com.intellij.openapi.editor.markup.EffectType;
import com.intellij.openapi.editor.markup.TextAttributes;
import com.intellij.openapi.project.DumbServiceImpl;
import com.intellij.openapi.util.Disposer;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.PsiMethodCallExpression;
import com.intellij.psi.util.PsiTreeUtil;
import com.intellij.psi.util.PsiUtilBase;
import com.intellij.testFramework.fixtures.LightJavaCodeInsightFixtureTestCase;
import com.intellij.testIntegration.TestFailedLineManager;
import com.intellij.util.concurrency.AppExecutorUtil;

import java.util.Date;
import java.util.List;
import java.util.concurrent.TimeUnit;

public class FailedLineTest extends LightJavaCodeInsightFixtureTestCase {

  public void testFailedLineManager() {

    configure();
    myFixture.enableInspections(new TestFailedLineInspection());
    myFixture.testHighlighting();
    PsiElement element = PsiUtilBase.getElementAtCaret(getEditor());
    PsiMethodCallExpression callExpression = PsiTreeUtil.getParentOfType(element, PsiMethodCallExpression.class);
    PsiMethod psiMethod = PsiTreeUtil.getParentOfType(callExpression, PsiMethod.class);
    TestFailedLineManager manager = TestFailedLineManager.getInstance(getProject());
    assertNotNull(manager.getTestInfo(psiMethod));
    TestStateStorage.Record record = manager.getFailedLineState(callExpression);
    assertNotNull(record);
  }

  public void testFailedLineInspection() {
    configure();
    myFixture.enableInspections(new TestFailedLineInspection());
    myFixture.testHighlighting();
    List<HighlightInfo> infos = myFixture.doHighlighting(HighlightSeverity.WARNING);
    assertEquals(1, infos.size());
    TextAttributes attributes = infos.get(0).forcedTextAttributes;
    assertNotNull(attributes);
    assertEquals(EffectType.BOLD_DOTTED_LINE, attributes.getEffectType());
  }

  public void testTopStacktraceLine() {
    TestStateStorage.Record record = configure();
    assertEquals("\tat MainTest.assertEquals(Assert.java:207)", record.topStacktraceLine);
  }

  public void testDumbMode() throws InterruptedException {
    TestConsoleProperties consoleProperties = new SMTRunnerConsoleProperties(new MockRuntimeConfiguration(getProject()), "SMRunnerTests",
                                                                             DefaultDebugExecutor.getDebugExecutorInstance());
    SMTRunnerConsoleView view = new SMTRunnerConsoleView(consoleProperties);
    try {
      DumbServiceImpl.getInstance(getProject()).setDumb(true);
      String url = "schema://url";
      SMTestProxy test = new SMTestProxy("foo", false, url);
      test.setLocator(JavaTestLocator.INSTANCE);

      view.initUI();
      SMTestRunnerResultsForm form = view.getResultsViewer();
      form.getTestsRootNode().addChild(test);
      test.setTestFailed("oops", "stacktrace", true);

      ApplicationManager.getApplication().invokeLater(() -> DumbServiceImpl.getInstance(getProject()).setDumb(false));
      form.onTestingFinished(form.getTestsRootNode());
      AppExecutorUtil.getAppExecutorService().awaitTermination(1, TimeUnit.SECONDS);
    }
    finally {
      Disposer.dispose(view);
      DumbServiceImpl.getInstance(getProject()).setDumb(false);
    }
  }

  private TestStateStorage.Record configure() {
    myFixture.addClass("package junit.framework; public class TestCase {}");
    myFixture.configureByText("MainTest.java", "  public class MainTest extends junit.framework.TestCase {\n" +
                                               "    public void testFoo() {\n" +
                                               "       <warning descr=\"oops\">assertE<caret>quals</warning>();\n" +
                                               "       assertEquals();\n" +
                                               "    }\n" +
                                               "    public void assertEquals() {}\n" +
                                               "  }");

    String url = "java:test://MainTest/testFoo";
    TestStackTraceParser pair = new TestStackTraceParser(url, "\tat junit.framework.Assert.fail(Assert.java:47)\n" +
                                                              "\tat MainTest.assertEquals(Assert.java:207)\n" +
                                                              "\tat MainTest.testFoo(MainTest.java:3)", "oops", JavaTestLocator.INSTANCE, getProject());
    assertEquals(3, pair.getFailedLine());
    assertEquals("assertEquals", pair.getFailedMethodName());
    TestStateStorage.Record record = new TestStateStorage.Record(TestStateInfo.Magnitude.FAILED_INDEX.getValue(), new Date(),
                                                                 0, pair.getFailedLine(), pair.getFailedMethodName(),
                                                                 pair.getErrorMessage(), pair.getTopLocationLine());
    TestStateStorage.getInstance(getProject()).writeState(url, record);
    return record;
  }
}