java/java-tests/testSrc/com/intellij/projectView/JavaTreeStructureTest.java

/*
 * Copyright (c) 2004 JetBrains s.r.o. All  Rights Reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met: 
 *
 * -Redistributions of source code must retain the above copyright
 *  notice, this list of conditions and the following disclaimer.
 *
 * -Redistribution in binary form must reproduct the above copyright
 *  notice, this list of conditions and the following disclaimer in
 *  the documentation and/or other materials provided with the distribution.
 *
 * Neither the name of JetBrains or IntelliJ IDEA
 * may be used to endorse or promote products derived from this software
 * without specific prior written permission.
 *
 * This software is provided "AS IS," without a warranty of any kind. ALL
 * EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, INCLUDING
 * ANY IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE
 * OR NON-INFRINGEMENT, ARE HEREBY EXCLUDED. JETBRAINS AND ITS LICENSORS SHALL NOT
 * BE LIABLE FOR ANY DAMAGES OR LIABILITIES SUFFERED BY LICENSEE AS A RESULT
 * OF OR RELATING TO USE, MODIFICATION OR DISTRIBUTION OF THE SOFTWARE OR ITS
 * DERIVATIVES. IN NO EVENT WILL JETBRAINS OR ITS LICENSORS BE LIABLE FOR ANY LOST
 * REVENUE, PROFIT OR DATA, OR FOR DIRECT, INDIRECT, SPECIAL, CONSEQUENTIAL,
 * INCIDENTAL OR PUNITIVE DAMAGES, HOWEVER CAUSED AND REGARDLESS OF THE THEORY
 * OF LIABILITY, ARISING OUT OF THE USE OF OR INABILITY TO USE SOFTWARE, EVEN
 * IF JETBRAINS HAS BEEN ADVISED OF THE POSSIBILITY OF SUCH DAMAGES.
 *
 */
package com.intellij.projectView;

import com.intellij.JavaTestUtil;
import com.intellij.ide.structureView.impl.java.JavaInheritedMembersNodeProvider;
import com.intellij.ide.structureView.impl.java.PublicElementsFilter;
import com.intellij.ide.structureView.impl.java.VisibilitySorter;
import com.intellij.ide.structureView.newStructureView.StructureViewComponent;
import com.intellij.openapi.fileEditor.FileEditor;
import com.intellij.openapi.fileEditor.FileEditorManager;
import com.intellij.openapi.util.Disposer;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.psi.JavaDirectoryService;
import com.intellij.psi.PsiClass;
import com.intellij.testFramework.IdeaTestUtil;
import com.intellij.testFramework.TestSourceBasedTestCase;

import javax.swing.*;

public class JavaTreeStructureTest extends TestSourceBasedTestCase {
  @Override
  protected String getTestPath() {
    return "structureView";
  }

  public void testJavaClassStructure() throws Exception {
    dotest(new CheckAction() {
      @Override
      public void testClassStructure(StructureViewComponent structureViewComponent) {
        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, true);
        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(),
                                                                 "-Class1.java\n" +
                                                                 " -Class1\n" +
                                                                 "  getValue(): int\n" +
                                                                 "  getClass(): Class<? extends Object>\n" +
                                                                 "  hashCode(): int\n" +
                                                                 "  equals(Object): boolean\n" +
                                                                 "  clone(): Object\n" +
                                                                 "  toString(): String\n" +
                                                                 "  notify(): void\n" +
                                                                 "  notifyAll(): void\n" +
                                                                 "  wait(long): void\n" +
                                                                 "  wait(long, int): void\n" +
                                                                 "  wait(): void\n" +
                                                                 "  finalize(): void\n" +
                                                                 "  myField1: boolean\n" +
                                                                 "  myField2: boolean\n");

        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, false);
        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, true);
        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, false);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class1.java\n" +
                                                                 " -Class1\n" +
                                                                 "  getValue(): int\n" +
                                                                 "  myField1: boolean\n" +
                                                                 "  myField2: boolean\n");

        structureViewComponent.setActionActive(PublicElementsFilter.ID, true);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class1.java\n" +
                                                                 " -Class1\n" +
                                                                 "  getValue(): int\n" +
                                                                 "  myField1: boolean\n" +
                                                                 "  myField2: boolean\n");

        structureViewComponent.setActionActive(PublicElementsFilter.ID, false);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class1.java\n" +
                                                                 " -Class1\n" +
                                                                 "  getValue(): int\n" +
                                                                 "  myField1: boolean\n" +
                                                                 "  myField2: boolean\n");
      }
    });
  }

  public void testShowClassMembers() throws Exception {
    dotest(new CheckAction() {
      @Override
      public void testClassStructure(StructureViewComponent structureViewComponent) {
        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, true);
        final JTree tree = structureViewComponent.getTree();
        tree.collapseRow(2);
        IdeaTestUtil.assertTreeEqual(tree, "-Class2.java\n" +
                                           " -Class2\n" +
                                           "  +InnerClass1\n" +
                                           "  +InnerClass2\n" +
                                           "  getValue(): int\n" +
                                           "  getClass(): Class<? extends Object>\n" +
                                           "  hashCode(): int\n" +
                                           "  equals(Object): boolean\n" +
                                           "  clone(): Object\n" +
                                           "  toString(): String\n" +
                                           "  notify(): void\n" +
                                           "  notifyAll(): void\n" +
                                           "  wait(long): void\n" +
                                           "  wait(long, int): void\n" +
                                           "  wait(): void\n" +
                                           "  finalize(): void\n" +
                                           "  myField1: boolean\n" +
                                           "  myField2: boolean\n" +
                                           "  myField3: boolean\n" +
                                           "  myField4: boolean\n");

        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, false);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class2.java\n" +
                                                                       " -Class2\n" +
                                                                       "  +InnerClass1\n" +
                                                                       "  +InnerClass2\n" +
                                                                       "  getValue(): int\n" +
                                                                       "  myField1: boolean\n" +
                                                                       "  myField2: boolean\n" +
                                                                       "  myField3: boolean\n" +
                                                                       "  myField4: boolean\n");
      }
    });
  }

  public void testVisibilitySorter() throws Exception {
    dotest(new CheckAction() {
      @Override
      public void testClassStructure(StructureViewComponent structureViewComponent) {
        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, false);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class2.java\n" +
                                                                 " -Class2\n" +
                                                                 "  __myPrivateFiield: int\n" +
                                                                 "  _myProtectedField: int\n" +
                                                                 "  myPublicField: int\n");

        structureViewComponent.setActionActive(VisibilitySorter.ID, true);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class2.java\n" +
                                                                 " -Class2\n" +
                                                                 "  myPublicField: int\n" +
                                                                 "  _myProtectedField: int\n" +
                                                                 "  __myPrivateFiield: int\n");
      }
    });
  }

  public void testMembersOrder() throws Exception {
    dotest(new CheckAction() {
      @Override
      public void testClassStructure(StructureViewComponent structureViewComponent) {
        structureViewComponent.setActionActive(JavaInheritedMembersNodeProvider.ID, false);

        IdeaTestUtil.assertTreeEqual(structureViewComponent.getTree(), "-Class2.java\n" +
                                                                 " -Class2\n" +
                                                                 "  Class2()\n" +
                                                                 "  af(): void\n" +
                                                                 "  zf(): void\n" +
                                                                 "  ab: int\n" +
                                                                 "  z: int\n"
        );
      }
    });
  }
  interface CheckAction {
    void testClassStructure(StructureViewComponent structureViewComponent);
  }
  private void dotest(CheckAction checkAction) throws Exception {
    final PsiClass psiClass = JavaDirectoryService.getInstance().getClasses(getPackageDirectory("com/package1"))[0];
    final VirtualFile virtualFile = psiClass.getContainingFile().getVirtualFile();
    final FileEditorManager fileEditorManager = FileEditorManager.getInstance(myProject);
    FileEditor[] fileEditors = fileEditorManager.openFile(virtualFile, false);
    final FileEditor fileEditor = fileEditors[0];
    try {
      final StructureViewComponent structureViewComponent =
        (StructureViewComponent)fileEditor.getStructureViewBuilder().createStructureView(fileEditor, myProject);

      checkAction.testClassStructure(structureViewComponent);
      Disposer.dispose(structureViewComponent);

    }
    finally {
      fileEditorManager.closeFile(virtualFile);
    }
  }

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