plugins/groovy/test/org/jetbrains/plugins/groovy/lang/overriding/FindOverridingMethodsAndClassesTest.groovy

/*
 * Copyright 2000-2009 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 org.jetbrains.plugins.groovy.lang.overriding
import com.intellij.psi.PsiClass
import com.intellij.psi.PsiMethod
import com.intellij.psi.search.searches.ClassInheritorsSearch
import com.intellij.psi.search.searches.OverridingMethodsSearch
import com.intellij.testFramework.fixtures.LightCodeInsightFixtureTestCase
import org.jetbrains.plugins.groovy.lang.psi.GroovyFile
import org.jetbrains.plugins.groovy.util.TestUtils
/**
 * @author Maxim.Medvedev
 */
public class FindOverridingMethodsAndClassesTest extends LightCodeInsightFixtureTestCase {
  @Override
  protected String getBasePath() {
    TestUtils.testDataPath + "overriding/findOverridingMethodsAndClasses"
  }

  void testSimpleCase() {
    doTest(1, 1)
  }

  void testAnonymousClass() {
    doTest(2, 2)
  }

  void testTraitImplementsInterface() {
    doTest(4, 4)
  }

  void testClassExtendsTrait() {
    doTest(2, 2)
  }

  void testTraitExtendsTrait() { doTest 2, 2 }
  void testTraitImplementsTrait() { doTest 2, 2 }

  private void doTest(int methodCount, int classCount) {
    myFixture.configureByFile(getTestName(false) + ".groovy")
    final GroovyFile groovyFile = (GroovyFile)myFixture.file
    final PsiClass psiClass = groovyFile.classes[0]
    final PsiMethod method = psiClass.methods[0]

    final Collection<PsiMethod> methods = OverridingMethodsSearch.search(method, psiClass.resolveScope, true).findAll()
    assertEquals("Method count is wrong", methodCount, methods.size())

    final Collection<PsiClass> classes = ClassInheritorsSearch.search(psiClass).findAll()
    assertEquals("Class count is wrong", classCount, classes.size())
  }
}