find usages of fragments #656

Author: vepanimas

src/main/com/intellij/lang/jsgraphql/ide/resolve/GraphQLScopeProvider.kt

@@ -4,6 +4,7 @@ import com.intellij.ide.scratch.ScratchUtil
 import com.intellij.lang.jsgraphql.ide.config.GraphQLConfigProvider
 import com.intellij.lang.jsgraphql.ide.resolve.scope.GraphQLMetaInfSchemaSearchScope
 import com.intellij.lang.jsgraphql.ide.resolve.scope.GraphQLRestrictedFileTypesScope
+import com.intellij.lang.jsgraphql.psi.GraphQLFragmentDefinition
 import com.intellij.lang.jsgraphql.psi.GraphQLFragmentSpread
 import com.intellij.lang.jsgraphql.psi.GraphQLIdentifier
 import com.intellij.lang.jsgraphql.schema.library.GraphQLLibraryRootsProvider
@@ -53,7 +54,7 @@ class GraphQLScopeProvider(private val project: Project) {
 
         fun isResolvedInNonStrictScope(element: PsiElement?): Boolean {
             val context = if (element is GraphQLIdentifier) element.parent else element
-            return context is GraphQLFragmentSpread
+            return context is GraphQLFragmentSpread || context is GraphQLFragmentDefinition
         }
     }
 

src/test/com/intellij/lang/jsgraphql/GraphQLTestCaseBase.java

@@ -58,6 +58,11 @@ protected void reloadConfiguration() {
         PlatformTestUtil.dispatchAllEventsInIdeEventQueue();
     }
 
+    protected void copyProject() {
+        myFixture.copyDirectoryToProject(getTestName(true), "");
+        reloadConfiguration();
+    }
+
     protected void doHighlightingTest() {
         doHighlightingTest("graphql");
     }

src/test/com/intellij/lang/jsgraphql/config/GraphQLConfigOverrideTest.kt

@@ -205,11 +205,6 @@ class GraphQLConfigOverrideTest : GraphQLTestCaseBase() {
         return projectConfig
     }
 
-    private fun copyProject() {
-        myFixture.copyDirectoryToProject(getTestName(true), "")
-        reloadConfiguration()
-    }
-
     private fun resolveConfig(file: VirtualFile): GraphQLProjectConfig {
         val config = GraphQLConfigProvider.getInstance(project).resolveConfig(file)
         TestCase.assertNotNull("config is not found", config)

src/test/com/intellij/lang/jsgraphql/config/GraphQLConfigResolveTest.kt

@@ -130,11 +130,6 @@ class GraphQLConfigResolveTest : GraphQLTestCaseBase() {
         )
     }
 
-    private fun copyProject() {
-        myFixture.copyDirectoryToProject(getTestName(true), "")
-        reloadConfiguration()
-    }
-
     private fun resolveConfig(filePath: String): GraphQLProjectConfig {
         val context = myFixture.configureFromTempProjectFile(filePath)
         TestCase.assertNotNull("source file is not found", context)

src/test/com/intellij/lang/jsgraphql/config/GraphQLConfigScopeTest.kt

@@ -184,11 +184,6 @@ class GraphQLConfigScopeTest : GraphQLTestCaseBase() {
         assertSameElements(message, actualFiles, expectedFiles)
     }
 
-    private fun copyProject() {
-        myFixture.copyDirectoryToProject(getTestName(true), "")
-        reloadConfiguration()
-    }
-
     private fun loadConfig(configPath: String): GraphQLConfig {
         val file = myFixture.findFileInTempDir(configPath)!!
         val config = GraphQLConfigProvider.getInstance(project).getForConfigFile(file)

src/test/com/intellij/lang/jsgraphql/findUsages/GraphQLFindUsagesTest.kt

@@ -0,0 +1,28 @@
+package com.intellij.lang.jsgraphql.findUsages
+
+import com.intellij.lang.jsgraphql.GraphQLTestCaseBase
+import com.intellij.lang.jsgraphql.ide.resolve.GraphQLResolveUtil
+import com.intellij.psi.PsiNamedElement
+import com.intellij.testFramework.UsefulTestCase
+
+class GraphQLFindUsagesTest : GraphQLTestCaseBase() {
+
+    override fun getBasePath(): String = "/findUsages"
+
+    override fun setUp() {
+        super.setUp()
+        copyProject()
+    }
+
+    fun testFragments() {
+        myFixture.configureFromTempProjectFile("${getTestName(true)}.graphql")
+        val usageInfos = myFixture.findUsages(myFixture.elementAtCaret)
+
+        UsefulTestCase.assertSize(3, usageInfos)
+        val containingDeclarationNames = usageInfos
+            .map { GraphQLResolveUtil.findContainingDefinition(it.element) }
+            .filterIsInstance<PsiNamedElement>()
+            .map { it.name }
+        assertSameElements(containingDeclarationNames, "Frag2", "Frag3", "Query1")
+    }
+}

test-resources/testData/graphql/findUsages/fragments/excluded.graphql

@@ -0,0 +1,4 @@
+fragment FragExcluded on Character {
+    ...Frag1
+}
+

test-resources/testData/graphql/findUsages/fragments/fragments.graphql

@@ -0,0 +1,13 @@
+fragment Fr<caret>ag1 on Character {
+    age
+    name
+}
+
+fragment Frag2 on Character {
+    ...Frag1
+}
+
+fragment Frag3 on Character {
+    ...Frag1
+}
+

test-resources/testData/graphql/findUsages/fragments/graphql.config.yml

@@ -0,0 +1,3 @@
+schema: schema.graphql
+documents: fragments.graphql
+exclude: excluded.graphql

test-resources/testData/graphql/findUsages/fragments/schema.graphql

@@ -0,0 +1,15 @@
+type Character {
+    name: String
+    age: Int
+    lang: String
+}
+
+type Query {
+    character: Character
+}
+
+query Query1 {
+    character {
+        ...Frag1
+    }
+}