Handle recursive input types when pruning schemas

ardatan · Sep 11, 2020 · 42156d1 · 42156d1
1 parent 65b41f0
commit 42156d1
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 2 deletions.
diff --git a/packages/utils/src/prune.ts b/packages/utils/src/prune.ts
@@ -167,6 +167,7 @@ function visitInputType(
   }
 
   pruningContext.unusedTypes[type.name] = false;
+  visitedTypes[type.name] = true;
 
   if (isInputObjectType(type)) {
     const fields = type.getFields();
@@ -176,8 +177,6 @@ function visitInputType(
       visitInputType(visitedTypes, pruningContext, namedType);
     });
   }
-
-  visitedTypes[type.name] = true;
 }
 
 function visitTypes(pruningContext: PruningContext, schema: GraphQLSchema): void {

diff --git a/packages/utils/tests/prune.test.ts b/packages/utils/tests/prune.test.ts
@@ -0,0 +1,18 @@
+import { buildSchema } from 'graphql';
+
+import { pruneSchema } from '../src/prune';
+
+describe('prune', () => {
+  test('can handle recursive input types', () => {
+    const schema = buildSchema(`
+      input Input {
+        moreInput: Input
+      }
+
+      type Query {
+        someQuery(input: Input): Boolean
+      }
+      `);
+    pruneSchema(schema);
+  });
+});