java ast indices: add injected code find usages test

java/java-tests/testData/compiler/compilerReferenceFindUsages/testFindUsagesInInjectedCode/Foo.java

@@ -0,0 +1,7 @@
+class A {
+
+  void m() {
+    String java = "class A { void m() { System.out.println(); }}";
+  }
+
+}

java/java-tests/testSrc/com/intellij/compiler/CompilerReferencesFindUsagesTest.java

@@ -0,0 +1,41 @@
+/*
+ * Copyright 2000-2016 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.compiler;
+
+import com.intellij.JavaTestUtil;
+import com.intellij.lang.injection.InjectedLanguageManager;
+import com.intellij.psi.PsiClass;
+import com.intellij.psi.PsiReference;
+import com.intellij.psi.impl.source.tree.injected.MyTestInjector;
+import com.intellij.psi.impl.source.tree.java.PsiReferenceExpressionImpl;
+import com.intellij.psi.search.searches.ReferencesSearch;
+import com.intellij.testFramework.SkipSlowTestLocally;
+
+@SkipSlowTestLocally
+public class CompilerReferencesFindUsagesTest extends CompilerReferencesTestBase {
+  protected String getTestDataPath() {
+    return JavaTestUtil.getJavaTestDataPath() + "/compiler/compilerReferenceFindUsages/";
+  }
+
+  public void testFindUsagesInInjectedCode() {
+    new MyTestInjector(getPsiManager()).injectAll(getTestRootDisposable());
+    myFixture.configureByFile(getName() + "/Foo.java");
+    rebuildProject();
+    PsiClass classForSearch = myFixture.getJavaFacade().findClass("java.lang.System");
+    PsiReference reference = assertOneElement(ReferencesSearch.search(classForSearch).findAll());
+    assertTrue(InjectedLanguageManager.getInstance(getProject()).isInjectedFragment(((PsiReferenceExpressionImpl)reference).getContainingFile()));
+  }
+}

java/java-tests/testSrc/com/intellij/compiler/CompilerReferencesTest.java

@@ -36,22 +36,7 @@ import java.util.Set;
 import java.util.stream.Collectors;
 
 @SkipSlowTestLocally
-public class CompilerReferencesTest extends AbstractCompilerAwareTest {
-  private boolean myDefaultEnableState;
-
-  @Override
-  public void setUp() throws Exception {
-    myDefaultEnableState = CompilerReferenceService.IS_ENABLED_KEY.asBoolean();
-    CompilerReferenceService.IS_ENABLED_KEY.setValue(true);
-    super.setUp();
-  }
-
-  @Override
-  public void tearDown() throws Exception {
-    CompilerReferenceService.IS_ENABLED_KEY.setValue(myDefaultEnableState);
-    super.tearDown();
-  }
-
+public class CompilerReferencesTest extends CompilerReferencesTestBase {
   protected String getTestDataPath() {
     return JavaTestUtil.getJavaTestDataPath() + "/compiler/bytecodeReferences/";
   }

java/java-tests/testSrc/com/intellij/compiler/CompilerReferencesTestBase.java

@@ -0,0 +1,35 @@
+/*
+ * Copyright 2000-2016 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.compiler;
+
+import com.intellij.codeInsight.completion.AbstractCompilerAwareTest;
+
+abstract class CompilerReferencesTestBase extends AbstractCompilerAwareTest {
+  private boolean myDefaultEnableState;
+
+  @Override
+  public void setUp() throws Exception {
+    myDefaultEnableState = CompilerReferenceService.IS_ENABLED_KEY.asBoolean();
+    CompilerReferenceService.IS_ENABLED_KEY.setValue(true);
+    super.setUp();
+  }
+
+  @Override
+  public void tearDown() throws Exception {
+    CompilerReferenceService.IS_ENABLED_KEY.setValue(myDefaultEnableState);
+    super.tearDown();
+  }
+}