Merge pull request #2821 from lgandecki/passContextToLocalDataStore

pass context to localGraphQLDatasource for testing purposes

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ### vNext
 
+- `apollo-gateway`: Pass `context` through to the `graphql` command in `LocalGraphQLDatasource` `process` method [PR #2821](https://github.com/apollographql/apollo-server/pull/2821)
 - `apollo-engine-reporting`: Set `forbiddenOperation` and `registeredOperation` later in the request lifecycle [PR #2828](https://github.com/apollographql/apollo-server/pull/2828)
 
 ### v2.6.2

packages/apollo-gateway/src/datasources/LocalGraphQLDatasource.ts

@@ -18,6 +18,7 @@ export class LocalGraphQLDataSource implements GraphQLDataSource {
 
   async process<TContext>({
     request,
+    context,
   }: Pick<GraphQLRequestContext<TContext>, 'request' | 'context'>): Promise<
     GraphQLResponse
   > {
@@ -26,6 +27,7 @@ export class LocalGraphQLDataSource implements GraphQLDataSource {
       source: request.query!,
       variableValues: request.variables,
       operationName: request.operationName,
+      contextValue: context,
     });
   }
 

packages/apollo-gateway/src/datasources/__tests__/LocalGraphQLDatasource.test.ts

@@ -0,0 +1,44 @@
+import { LocalGraphQLDataSource } from '../LocalGraphQLDatasource';
+import { buildFederatedSchema } from '@apollo/federation';
+import gql from 'graphql-tag';
+
+describe('constructing requests', () => {
+  it('accepts context', async () => {
+    const typeDefs = gql`
+      type Query {
+        me: User
+      }
+      type User {
+        id: ID
+        name: String!
+      }
+    `;
+    const resolvers = {
+      Query: {
+        me(_, __, { userId }) {
+          const users = [
+            { id: 1, name: 'otherGuy' },
+            { id: 2, name: 'james' },
+            {
+              id: 3,
+              name: 'someoneElse',
+            },
+          ];
+          return users.find(user => user.id === userId);
+        },
+      },
+    };
+    const schema = buildFederatedSchema([{ typeDefs, resolvers }]);
+
+    const DataSource = new LocalGraphQLDataSource(schema);
+
+    const { data } = await DataSource.process({
+      request: {
+        query: '{ me { name } }',
+      },
+      context: { userId: 2 },
+    });
+
+    expect(data).toEqual({ me: { name: 'james' } });
+  });
+});