add support for Context in test runners

docs/pages/docs/apis/config.mdx

@@ -447,7 +447,7 @@ S3 options:
 - `pathPrefix`: The prefix for the file, used to set the subfolder of your bucket files will be stored in.
 - `endpoint`: The endpoint to use - if provided, this endpoint will be used instead of the default amazon s3 endpoint
 - `forcePathStyle`: Force the old pathstyle of using the bucket name after the host
-- `signed.expiry`: Use S3 URL signing to keep S3 assets private. `expiry` is in seconds 
+- `signed.expiry`: Use S3 URL signing to keep S3 assets private. `expiry` is in seconds
 
 ```typescript
 import { config } from '@keystone-6/core';

examples/ecommerce/tests/mutations.test.ts

@@ -5,8 +5,7 @@ import { Context } from '.keystone/types';
 const FAKE_ID = 'cinjfgbkjnfg';
 
 const asUser = (context: Context, itemId?: string) => context.withSession({ itemId, data: {} });
-
-const runner = setupTestRunner({ config });
+const runner = setupTestRunner<Context>({ config });
 
 describe(`Custom mutations`, () => {
   describe('checkout(token)', () => {

packages/core/src/testing.ts

@@ -89,13 +89,29 @@ export async function setupTestEnv({
   };
 }
 
-export function setupTestRunner({ config }: { config: KeystoneConfig }) {
+export function setupTestRunner<Context extends KeystoneContext>({
+  config,
+}: {
+  config: KeystoneConfig;
+}) {
+  type TestArgs = {
+    context: Context;
+    graphQLRequest: GraphQLRequest;
+    app: express.Express;
+    server: Server;
+  };
+
   return (testFn: (testArgs: TestArgs) => Promise<void>) => async () => {
     // Reset the database to be empty for every test.
     const { connect, disconnect, testArgs } = await setupTestEnv({ config });
     await connect();
+
+    const { context } = testArgs;
     try {
-      return await testFn(testArgs);
+      return await testFn({
+        ...testArgs,
+        context: context as Context,
+      });
     } finally {
       await disconnect();
     }