Autocompletion inside GraphQL injections #662

Author: vepanimas

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

@@ -90,7 +90,8 @@ class GraphQLScopeProvider(private val project: Project) {
         element: PsiElement,
         key: Key<CachedValue<GlobalSearchScope>>
     ): GlobalSearchScope {
-        val file = element.containingFile
+        val file = element.containingFile.originalFile
+
         return CachedValuesManager.getCachedValue(file, key) {
             val projectConfig = configProvider.resolveProjectConfig(file)
             var scope: GlobalSearchScope =

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

@@ -29,6 +29,19 @@ public abstract class GraphQLCompletionTestCaseBase extends GraphQLTestCaseBase
         return doTest();
     }
 
+    protected LookupElement @Nullable [] doTestWithProject() {
+        return doTestWithProject(".graphql");
+    }
+
+    protected LookupElement @Nullable [] doTestWithProject(@NotNull String ext) {
+        String dirName = getTestName(false);
+        myFixture.copyDirectoryToProject(dirName, "");
+        reloadConfiguration();
+        String filePath = dirName + ext;
+        myFixture.configureFromTempProjectFile(filePath);
+        return myFixture.complete(CompletionType.BASIC, 1);
+    }
+
     protected static void checkEqualsOrdered(LookupElement @Nullable [] items, String @NotNull ... variants) {
         assertNotNull(items);
         assertOrderedEquals(ContainerUtil.map(items, LookupElement::getLookupString), variants);

src/test/com/intellij/lang/jsgraphql/completion/GraphQLOperationsCompletionTest.java

@@ -398,4 +398,8 @@ public void testSubscriptionOperationDirectives() {
         checkEqualsOrdered(lookupElements, "SubscriptionDir", "SubscriptionDir1");
     }
 
+    public void testEnumArgumentInInjection() {
+        LookupElement[] lookupElements = doTestWithProject(".js");
+        checkEqualsOrdered(lookupElements, "ADMIN", "GUEST", "USER");
+    }
 }

test-resources/testData/graphql/completion/operations/EnumArgumentInInjection/EnumArgumentInInjection.js

@@ -0,0 +1,7 @@
+const QUERY = gql`
+    {
+        field {
+            user(role: <caret>)
+        }
+    }
+`;

test-resources/testData/graphql/completion/operations/EnumArgumentInInjection/graphql.config.yml

@@ -0,0 +1,2 @@
+schema: schema.graphql
+documents: '**/*.graphql'

test-resources/testData/graphql/completion/operations/EnumArgumentInInjection/schema.graphql

@@ -0,0 +1,13 @@
+type Query {
+    field: FieldType
+}
+
+type FieldType {
+   user(role: UserRole): ID
+}
+
+enum UserRole {
+    ADMIN
+    USER
+    GUEST
+}