diff --git a/python/testSrc/com/jetbrains/python/fixtures/PyLightFixtureTestCase.java b/python/testSrc/com/jetbrains/python/fixtures/PyLightFixtureTestCase.java index 3cab76ca553d..1f76476ab157 100644 --- a/python/testSrc/com/jetbrains/python/fixtures/PyLightFixtureTestCase.java +++ b/python/testSrc/com/jetbrains/python/fixtures/PyLightFixtureTestCase.java @@ -6,6 +6,8 @@ import com.intellij.openapi.module.ModuleType; import com.intellij.openapi.projectRoots.Sdk; import com.intellij.openapi.roots.ContentEntry; import com.intellij.openapi.roots.ModifiableRootModel; +import com.intellij.psi.PsiDocumentManager; +import com.intellij.psi.PsiReference; import com.intellij.testFramework.LightProjectDescriptor; import com.intellij.testFramework.UsefulTestCase; import com.intellij.testFramework.fixtures.CodeInsightTestFixture; @@ -55,6 +57,11 @@ public abstract class PyLightFixtureTestCase extends UsefulTestCase { return ourPyDescriptor; } + protected PsiReference findReferenceBySignature(final String signature) { + int pos = PsiDocumentManager.getInstance(myFixture.getProject()).getDocument(myFixture.getFile()).getText().indexOf(signature); + return myFixture.getFile().findReferenceAt(pos); + } + protected static class PyLightProjectDescriptor implements LightProjectDescriptor { public ModuleType getModuleType() { return EmptyModuleType.getInstance();