openide/java/java-tests/testSrc/com/intellij/java/ide/hierarchy/JavaCallHierarchyTest.java

// Copyright 2000-2021 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.ide.hierarchy;

import com.intellij.JavaTestUtil;
import com.intellij.ide.hierarchy.HierarchyBrowserBaseEx;
import com.intellij.ide.hierarchy.JavaHierarchyUtil;
import com.intellij.ide.hierarchy.actions.BrowseTypeHierarchyAction;
import com.intellij.ide.hierarchy.call.CalleeMethodsTreeStructure;
import com.intellij.ide.hierarchy.call.CallerMethodsTreeStructure;
import com.intellij.ide.highlighter.XmlFileType;
import com.intellij.openapi.actionSystem.ex.ActionUtil;
import com.intellij.openapi.projectRoots.Sdk;
import com.intellij.pom.java.LanguageLevel;
import com.intellij.psi.JavaPsiFacade;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiField;
import com.intellij.psi.PsiMember;
import com.intellij.psi.search.ProjectScope;
import com.intellij.testFramework.IdeaTestUtil;
import com.intellij.testFramework.TestActionEvent;
import com.intellij.testFramework.codeInsight.hierarchy.HierarchyViewTestBase;
import org.jetbrains.annotations.NotNull;

import java.util.Arrays;

public class JavaCallHierarchyTest extends HierarchyViewTestBase {
  @Override
  protected @NotNull LanguageLevel getProjectLanguageLevel() {
    return LanguageLevel.JDK_1_8; // method refs are needed
  }

  @NotNull
  @Override
  protected String getTestDataPath() {
    return JavaTestUtil.getJavaTestDataPath();
  }

  @Override
  protected String getBasePath() {
    return "ide/hierarchy/call/" + getTestName(false);
  }

  private void doJavaCallerTypeHierarchyTest(@NotNull String classFqn, @NotNull String methodName, String @NotNull ... fileNames) throws Exception {
    doHierarchyTest(() -> {
      PsiClass psiClass = JavaPsiFacade.getInstance(getProject()).findClass(classFqn, ProjectScope.getProjectScope(getProject()));
      assertNotNull("Class '" + classFqn + "' not found", psiClass);
      PsiMember method = psiClass.findMethodsByName(methodName, false) [0];
      assertNotNull("Method '" + methodName + "' not found in " + classFqn + ". Available methods are " +
                    Arrays.toString(psiClass.getMethods()), method);
      return new CallerMethodsTreeStructure(getProject(), method, HierarchyBrowserBaseEx.SCOPE_PROJECT);
    }, JavaHierarchyUtil.getComparator(myProject), fileNames);
  }
  private void doJavaCalleeTypeHierarchyTest(@NotNull String classFqn, @NotNull String methodName, String @NotNull ... fileNames) throws Exception {
    doHierarchyTest(() -> {
      PsiClass psiClass = JavaPsiFacade.getInstance(getProject()).findClass(classFqn, ProjectScope.getProjectScope(getProject()));
      PsiMember method = psiClass.findMethodsByName(methodName, false) [0];
      return new CalleeMethodsTreeStructure(getProject(), method, HierarchyBrowserBaseEx.SCOPE_PROJECT);
    }, JavaHierarchyUtil.getComparator(myProject),fileNames);
  }

  public void testDirectRecursion() throws Exception {
    doJavaCallerTypeHierarchyTest("A", "recursive", "A.java");
  }

  public void testCalleeDirectRecursion() throws Exception {
    doJavaCalleeTypeHierarchyTest("A", "recursive", "A.java");
  }

  public void testIndirectRecursion() throws Exception {
    doJavaCallerTypeHierarchyTest("A", "recursive2", "A.java");
  }

  public void testIdeaDev41005() throws Exception {
    doJavaCallerTypeHierarchyTest("B", "xyzzy", "A.java");
  }

  public void testIdeaDev41005_Inheritance() throws Exception {
    doJavaCallerTypeHierarchyTest("D", "xyzzy", "A.java");
  }

  public void testIdeaDev41005_Sibling() throws Exception {
    doJavaCallerTypeHierarchyTest("D", "xyzzy", "A.java");
  }

  public void testIdeaDev41005_SiblingUnderInheritance() throws Exception {
    doJavaCallerTypeHierarchyTest("D", "xyzzy", "A.java");
  }

  public void testIdeaDev41232() throws Exception {
    doJavaCallerTypeHierarchyTest("A", "main", "A.java");
  }

  public void testDefaultConstructor() throws Exception {
    doJavaCallerTypeHierarchyTest("A", "A", "A.java");
  }

  public void testMethodRef() throws Exception {
    doHierarchyTest(() -> {
      PsiClass psiClass = JavaPsiFacade.getInstance(getProject()).findClass("A", ProjectScope.getProjectScope(getProject()));
      PsiMember method = psiClass.findMethodsByName("testMethod", false) [0];
      return new CalleeMethodsTreeStructure(getProject(), method, HierarchyBrowserBaseEx.SCOPE_PROJECT);
    }, JavaHierarchyUtil.getComparator(myProject),"A.java");
  }

  public void testField() throws Exception {
    doHierarchyTest(() -> {
      PsiClass psiClass = JavaPsiFacade.getInstance(getProject()).findClass("A", ProjectScope.getProjectScope(getProject()));
      PsiField field = psiClass.findFieldByName("testField", false);
      return new CallerMethodsTreeStructure(getProject(), field, HierarchyBrowserBaseEx.SCOPE_PROJECT);
    }, JavaHierarchyUtil.getComparator(myProject),"A.java");
  }

  public void testAnonymous2() throws Exception {
    doJavaCallerTypeHierarchyTest("A", "doIt", "A.java");
  }

  public void testActionAvailableInXml() {
    configureByText(XmlFileType.INSTANCE, "<foo>java.lang.Str<caret>ing</foo>");
    BrowseTypeHierarchyAction action = new BrowseTypeHierarchyAction();
    TestActionEvent e = new TestActionEvent(action);
    assertTrue(ActionUtil.lastUpdateAndCheckDumb(action, e, true));
    assertTrue(e.getPresentation().isEnabled() && e.getPresentation().isVisible());
  }

  @Override
  protected Sdk getTestProjectJdk() {
    return IdeaTestUtil.getMockJdk18();
  }

  public void testMustIgnoreJavadocReferences() throws Exception {
    doJavaCallerTypeHierarchyTest("p.X", "persist", "X.java");
  }
  public void testCallersOfBaseMethod() throws Exception {
    doJavaCallerTypeHierarchyTest("p.BaseClass", "method", "X.java");
  }
  public void testCallersOfSubMethod() throws Exception {
    doJavaCallerTypeHierarchyTest("p.BaseClass", "method", "X.java");
  }
  public void testEnclosingDeps() throws Exception {
    doJavaCallerTypeHierarchyTest("DummyImpl", "doSth", "A.java");
  }
  public void testThroughAnonymous() throws Exception {
    doJavaCallerTypeHierarchyTest("com.hierarchytest.AcmClientImpl", "getUser", "X.java");
  }
  public void testThroughAnonymousCalledByOther() throws Exception {
    doJavaCallerTypeHierarchyTest("x.AcmClientImpl", "returnSomething", "X.java");
  }
  public void testWildcards() throws Exception {
    doJavaCallerTypeHierarchyTest("p.BoardImpl", "getCount", "A.java");
  }
}