Add tests for middleware

Urigo · Nov 29, 2018 · 24eac3d · 24eac3d
1 parent d43271b
commit 24eac3d
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/packages/core/tests/graphql-module.spec.ts b/packages/core/tests/graphql-module.spec.ts
@@ -886,4 +886,26 @@ describe('GraphQLModule', () => {
       expect(result.data['bar'].content).toBe('BAR');
     });
   });
+  it('should mutate schema using middleware', async () => {
+    const { schema, context } = new GraphQLModule({
+      typeDefs: gql`
+        type Query {
+          foo: Boolean
+        }
+      `,
+      resolvers: {
+        Query: {
+          foo: (root, args, context, info) => !!info.schema['__DIRTY__'],
+        },
+      },
+      middleware: ({schema}) => { schema['__DIRTY__'] = true; return { schema }; },
+    });
+    const result = await execute({
+      schema,
+      document: gql`query { foo }`,
+      contextValue: await context({ req: {} }),
+    });
+    expect(result.errors).toBeFalsy();
+    expect(result.data['foo']).toBeTruthy();
+  });
 });