diff --git a/idea/tests/org/jetbrains/jet/plugin/libraries/LibraryNavigationRegressionTest.java b/idea/tests/org/jetbrains/jet/plugin/libraries/LibraryNavigationRegressionTest.java index ac5873a882e..f492df76dda 100644 --- a/idea/tests/org/jetbrains/jet/plugin/libraries/LibraryNavigationRegressionTest.java +++ b/idea/tests/org/jetbrains/jet/plugin/libraries/LibraryNavigationRegressionTest.java @@ -37,36 +37,25 @@ public class LibraryNavigationRegressionTest extends LightCodeInsightFixtureTest * Regression test against KT-1652 */ public void testRefToStdlib() { - String text = "fun foo() { println() }"; - myFixture.configureByText(JetFileType.INSTANCE, text); - PsiReference ref = myFixture.getFile().findReferenceAt(text.indexOf("println")); - //noinspection ConstantConditions - assertSame(JetLanguage.INSTANCE, ref.resolve().getNavigationElement().getLanguage()); + PsiElement navigationElement = configureAndResolve("fun foo() { println() }"); + assertSame(JetLanguage.INSTANCE, navigationElement.getLanguage()); } /** * Regression test against KT-1652 */ public void testRefToJdk() { - String text = "val x = java.util.HashMap().get(\"\")"; - myFixture.configureByText(JetFileType.INSTANCE, text); - PsiReference ref = myFixture.getFile().findReferenceAt(text.indexOf("get")); - //noinspection ConstantConditions - ref.resolve().getNavigationElement(); + configureAndResolve("val x = java.util.HashMap().get(\"\")"); } /** * Regression test against KT-1815 */ public void testRefToAltHeaders() { - String text = "fun foo(e : java.util.Map.Entry) { e.getKey(); }"; - myFixture.configureByText(JetFileType.INSTANCE, text); - PsiReference ref = myFixture.getFile().findReferenceAt(text.indexOf("getKey")); + PsiElement navigationElement = configureAndResolve("fun foo(e : java.util.Map.Entry) { e.getKey(); }"); PsiClass expectedClass = JavaPsiFacade.getInstance(getProject()).findClass("java.util.Map.Entry", GlobalSearchScope.allScope(getProject())); - //noinspection ConstantConditions - PsiElement actualClass = ref.resolve().getNavigationElement().getParent(); - assertSame(expectedClass, actualClass); + assertSame(expectedClass, navigationElement.getParent()); } @NotNull @@ -74,4 +63,11 @@ public class LibraryNavigationRegressionTest extends LightCodeInsightFixtureTest protected LightProjectDescriptor getProjectDescriptor() { return JetWithJdkAndRuntimeLightProjectDescriptor.INSTANCE; } + + private PsiElement configureAndResolve(String text) { + myFixture.configureByText(JetFileType.INSTANCE, text); + PsiReference ref = myFixture.getFile().findReferenceAt(myFixture.getCaretOffset()); + //noinspection ConstantConditions + return ref.resolve().getNavigationElement(); + } }