diff --git a/packages/wobe-graphql-yoga/src/index.test.ts b/packages/wobe-graphql-yoga/src/index.test.ts index 11a2bf6..1e184ae 100644 --- a/packages/wobe-graphql-yoga/src/index.test.ts +++ b/packages/wobe-graphql-yoga/src/index.test.ts @@ -20,12 +20,16 @@ describe('Wobe GraphQL Yoga plugin', () => { Query: { hello: (_, __, context) => { context.response.setCookie('tata', 'tata') + expect(context.test).toBeDefined() expect(context.response).toBeDefined() expect(context.request.headers).toBeDefined() return 'Hello from Yoga!' }, }, }, + context: () => { + return { test: 'test' } + }, }), ) diff --git a/packages/wobe-graphql-yoga/src/index.ts b/packages/wobe-graphql-yoga/src/index.ts index aba0e35..2f4a66c 100644 --- a/packages/wobe-graphql-yoga/src/index.ts +++ b/packages/wobe-graphql-yoga/src/index.ts @@ -3,13 +3,15 @@ import { createYoga, type GraphQLSchemaWithContext, type YogaServerOptions, - type YogaInitialContext, } from 'graphql-yoga' import type { MaybePromise, Wobe, WobePlugin, WobeResponse } from 'wobe' export type GraphqlYogaContext = | MaybePromise> - | ((initialContext: YogaInitialContext) => MaybePromise) + | ((context: { + request: Request + response: WobeResponse + }) => MaybePromise) export interface GraphqlYogaPluginOptions { graphqlMiddleware?: (