import { LocalGraphQLDataSource } from '../LocalGraphQLDataSource';
import { buildSubgraphSchema } from '@apollo/subgraph';
import gql from 'graphql-tag';
import { GraphQLResolverMap } from 'apollo-graphql';
import { GraphQLRequestContext } from 'apollo-server-types';
import { GraphQLDataSourceRequestKind } from '../types';

describe('constructing requests', () => {
  it('accepts context', async () => {
    const typeDefs = gql`
      type Query {
        me: User
      }
      type User {
        id: ID
        name: String!
      }
    `;
    const resolvers: GraphQLResolverMap<{ userId: number }> = {
      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 = buildSubgraphSchema([{ typeDefs, resolvers }]);

    const DataSource = new LocalGraphQLDataSource(schema);

    const { data } = await DataSource.process({
      kind: GraphQLDataSourceRequestKind.INCOMING_OPERATION,
      request: {
        query: '{ me { name } }',
      },
      incomingRequestContext: {
        context: { userId: 2 },
      } as GraphQLRequestContext<{userId: number}>,
      context: { userId: 2 },
    });

    expect(data).toEqual({ me: { name: 'james' } });
  });
});