| /* |
| * Copyright 2000-2014 JetBrains s.r.o. |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| package com.intellij.projectView; |
| |
| import com.intellij.ide.DataManager; |
| import com.intellij.ide.projectView.ProjectView; |
| import com.intellij.ide.projectView.impl.AbstractProjectViewPSIPane; |
| import com.intellij.ide.projectView.impl.ProjectViewImpl; |
| import com.intellij.ide.projectView.impl.ProjectViewPane; |
| import com.intellij.ide.projectView.impl.ProjectViewToolWindowFactory; |
| import com.intellij.openapi.actionSystem.CommonDataKeys; |
| import com.intellij.openapi.actionSystem.DataContext; |
| import com.intellij.openapi.fileEditor.FileEditorManager; |
| import com.intellij.openapi.vfs.CharsetToolkit; |
| import com.intellij.openapi.vfs.VirtualFile; |
| import com.intellij.openapi.wm.ToolWindow; |
| import com.intellij.openapi.wm.ToolWindowId; |
| import com.intellij.openapi.wm.ToolWindowManager; |
| import com.intellij.psi.*; |
| import com.intellij.testFramework.PlatformTestUtil; |
| |
| import javax.swing.*; |
| import java.io.IOException; |
| |
| @SuppressWarnings({"HardCodedStringLiteral"}) |
| public class NavigateFromSourceTest extends BaseProjectViewTestCase { |
| public void testShowClassMembers() throws Exception { |
| useStandardProviders(); |
| final PsiClass[] classes = JavaDirectoryService.getInstance().getClasses(getPackageDirectory()); |
| sortClassesByName(classes); |
| PsiClass psiClass = classes[0]; |
| |
| final AbstractProjectViewPSIPane pane = myStructure.createPane(); |
| final PsiFile containingFile = psiClass.getContainingFile(); |
| final VirtualFile virtualFile = containingFile.getVirtualFile(); |
| |
| myStructure.checkNavigateFromSourceBehaviour(psiClass, virtualFile, pane); |
| |
| PlatformTestUtil.assertTreeEqual(pane.getTree(), "-Project\n" + |
| " -PsiDirectory: showClassMembers\n" + |
| " -PsiDirectory: src\n" + |
| " -PsiDirectory: com\n" + |
| " -PsiDirectory: package1\n" + |
| " [Class1]\n" + |
| " Class2\n" + |
| getRootFiles() + |
| " +External Libraries\n" |
| , true); |
| |
| changeClassTextAndTryToNavigate("class Class11 {}", (PsiJavaFile)containingFile, pane, "-Project\n" + |
| " -PsiDirectory: showClassMembers\n" + |
| " -PsiDirectory: src\n" + |
| " -PsiDirectory: com\n" + |
| " -PsiDirectory: package1\n" + |
| " -Class1.java\n" + |
| " [Class11]\n" + |
| " Class2\n" + |
| getRootFiles() + |
| " +External Libraries\n"); |
| |
| changeClassTextAndTryToNavigate("class Class1 {}", (PsiJavaFile)containingFile, pane, "-Project\n" + |
| " -PsiDirectory: showClassMembers\n" + |
| " -PsiDirectory: src\n" + |
| " -PsiDirectory: com\n" + |
| " -PsiDirectory: package1\n" + |
| " [Class1]\n" + |
| " Class2\n" + |
| getRootFiles() + |
| " +External Libraries\n"); |
| |
| doTestMultipleSelection(pane, ((PsiJavaFile)containingFile).getClasses()[0]); |
| } |
| |
| public void testAutoscrollFromSourceOnOpening() throws Exception { |
| final PsiClass[] classes = JavaDirectoryService.getInstance().getClasses(getPackageDirectory()); |
| PsiClass psiClass = classes[0]; |
| |
| FileEditorManager.getInstance(getProject()).openFile(psiClass.getContainingFile().getVirtualFile(), true); |
| |
| ProjectView projectView = ProjectView.getInstance(getProject()); |
| |
| ((ProjectViewImpl)projectView).setAutoscrollFromSource(true, ProjectViewPane.ID); |
| |
| ToolWindow toolWindow = ToolWindowManager.getInstance(getProject()).getToolWindow(ToolWindowId.PROJECT_VIEW); |
| |
| new ProjectViewToolWindowFactory().createToolWindowContent(getProject(), toolWindow); |
| |
| projectView.changeView(ProjectViewPane.ID); |
| |
| JComponent component = ((ProjectViewImpl)projectView).getComponent(); |
| DataContext context = DataManager.getInstance().getDataContext(component); |
| PsiElement element = CommonDataKeys.PSI_ELEMENT.getData(context); |
| assertEquals("Class1.java", ((PsiJavaFile)element).getName()); |
| } |
| |
| private static void doTestMultipleSelection(final AbstractProjectViewPSIPane pane, final PsiClass psiClass) { |
| JTree tree = pane.getTree(); |
| int rowCount = tree.getRowCount(); |
| for (int i = 0; i < rowCount; i++) { |
| tree.addSelectionRow(i); |
| } |
| |
| pane.select(psiClass, psiClass.getContainingFile().getVirtualFile(), true); |
| |
| assertEquals(9, tree.getSelectionCount()); |
| } |
| |
| private static void changeClassTextAndTryToNavigate(final String newClassString, |
| PsiJavaFile psiFile, |
| final AbstractProjectViewPSIPane pane, |
| final String expected) throws IOException, InterruptedException { |
| PsiClass psiClass = psiFile.getClasses()[0]; |
| final VirtualFile virtualFile = psiClass.getContainingFile().getVirtualFile(); |
| final JTree tree = pane.getTree(); |
| virtualFile.setBinaryContent(newClassString.getBytes(CharsetToolkit.UTF8_CHARSET)); |
| |
| PlatformTestUtil.waitForAlarm(600); |
| |
| psiClass = psiFile.getClasses()[0]; |
| pane.select(psiClass, virtualFile, true); |
| PlatformTestUtil.assertTreeEqual(tree, expected, true); |
| } |
| } |