From 86c8bfc864195f2070b97f19d4fda652da9851d1 Mon Sep 17 00:00:00 2001 From: Alexey Sedunov Date: Tue, 24 Jun 2014 16:07:42 +0400 Subject: [PATCH] Find Usages: Add support of searching by reference in AbstractFindUsagesTest --- .../jet/findUsages/AbstractJetFindUsagesTest.java | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/idea/tests/org/jetbrains/jet/findUsages/AbstractJetFindUsagesTest.java b/idea/tests/org/jetbrains/jet/findUsages/AbstractJetFindUsagesTest.java index d747e42bf62..0a63fc33543 100644 --- a/idea/tests/org/jetbrains/jet/findUsages/AbstractJetFindUsagesTest.java +++ b/idea/tests/org/jetbrains/jet/findUsages/AbstractJetFindUsagesTest.java @@ -21,6 +21,7 @@ import com.google.common.base.Joiner; import com.google.common.base.Predicates; import com.google.common.collect.Collections2; import com.google.common.collect.Ordering; +import com.intellij.codeInsight.TargetElementUtilBase; import com.intellij.find.FindManager; import com.intellij.find.findUsages.*; import com.intellij.find.impl.FindManagerImpl; @@ -301,7 +302,11 @@ public abstract class AbstractJetFindUsagesTest extends LightCodeInsightFixtureT } myFixture.configureByFile(path); - T caretElement = PsiTreeUtil.getParentOfType(myFixture.getElementAtCaret(), caretElementClass, false); + PsiElement originalElement = + InTextDirectivesUtils.isDirectiveDefined(mainFileText, "// FIND_BY_REF") + ? TargetElementUtilBase.findTargetElement(myFixture.getEditor(), TargetElementUtilBase.REFERENCED_ELEMENT_ACCEPTED) + : myFixture.getElementAtCaret(); + T caretElement = PsiTreeUtil.getParentOfType(originalElement, caretElementClass, false); assertNotNull(String.format("Element with type '%s' wasn't found at caret position", caretElementClass), caretElement); FindUsagesOptions options = parser != null ? parser.parse(mainFileText, getProject()) : null;