| /* |
| * Copyright 2000-2012 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.psi.search; |
| |
| import com.intellij.JavaTestUtil; |
| import com.intellij.openapi.diagnostic.Logger; |
| import com.intellij.openapi.progress.ProgressManager; |
| import com.intellij.psi.JavaPsiFacade; |
| import com.intellij.psi.PsiClass; |
| import com.intellij.psi.PsiPackage; |
| import com.intellij.psi.search.searches.ClassInheritorsSearch; |
| import com.intellij.testFramework.IdeaTestUtil; |
| import com.intellij.testFramework.PsiTestCase; |
| import com.intellij.testFramework.PsiTestUtil; |
| import org.jetbrains.annotations.NotNull; |
| |
| import java.util.ArrayList; |
| |
| public class InheritorsTest extends PsiTestCase{ |
| private static final Logger LOG = Logger.getInstance("#com.intellij.psi.search.InheritorsTest"); |
| |
| @Override |
| protected void setUp() throws Exception { |
| super.setUp(); |
| |
| String root = JavaTestUtil.getJavaTestDataPath() + "/psi/search/inheritors/" + getTestName(true); |
| PsiTestUtil.removeAllRoots(myModule, IdeaTestUtil.getMockJdk17()); |
| PsiTestUtil.createTestProjectStructure(myProject, myModule, root, myFilesToDelete); |
| } |
| |
| public void testScope() throws Exception { |
| doTest("pack1.Base", "pack1", true, "pack1.Derived1", "pack1.Derived3"); |
| } |
| |
| public void testNoScanJdk() throws Exception { |
| doTest("javax.swing.JPanel", "", false); |
| } |
| |
| public void testSameNamedClasses() throws Exception { |
| doTest("x.Test", "", true, "x.Goo", "x.Zoo"); |
| } |
| |
| public void testMultipleInheritance() throws Exception { |
| doTest("I", "", true, "A", "B"); |
| } |
| |
| private void doTest(String className, String packageScopeName, final boolean deep, String... inheritorNames) throws Exception { |
| final PsiClass aClass = myJavaFacade.findClass(className); |
| assertNotNull(aClass); |
| |
| final SearchScope scope; |
| if (packageScopeName != null){ |
| PsiPackage aPackage = JavaPsiFacade.getInstance(myPsiManager.getProject()).findPackage(packageScopeName); |
| scope = PackageScope.packageScope(aPackage, true).intersectWith(GlobalSearchScope.projectScope(myProject)); |
| } |
| else{ |
| scope = GlobalSearchScope.projectScope(myProject); |
| } |
| |
| final ArrayList<String> inheritorsList = new ArrayList<String>(); |
| ProgressManager.getInstance().runProcess( |
| new Runnable() { |
| @Override |
| public void run() { |
| ClassInheritorsSearch.search(aClass, scope, deep).forEach(new PsiElementProcessorAdapter<PsiClass>(new PsiElementProcessor<PsiClass>() { |
| @Override |
| public boolean execute(@NotNull PsiClass element) { |
| inheritorsList.add(element.getQualifiedName()); |
| return true; |
| } |
| })); |
| } |
| }, |
| null |
| ); |
| |
| assertSameElements(inheritorsList, inheritorNames); |
| } |
| } |