chore: add withMocks function

apollographql · Aug 29, 2024 · 674da60 · 674da60 · github-actions · Aug 29, 2024
1 parent d476674
commit 674da60
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/src/handlers.ts b/src/handlers.ts
@@ -146,6 +146,25 @@ function createHandlerFromSchema<TResolvers>(
     });
   }
 
+  function withMocks(mocks: IMocks<TResolvers>) {
+    const oldSchema = testSchema;
+
+    testSchema = addMocksToSchema({
+      schema: oldSchema,
+      mocks: mocks,
+    });
+
+    function restore() {
+      testSchema = oldSchema;
+    }
+
+    return Object.assign(restore, {
+      [Symbol.dispose]() {
+        restore();
+      },
+    });
+  }
+
   function replaceDelay(newDelay: Delay) {
     const oldDelay = _delay;
     _delay = newDelay;
@@ -269,6 +288,7 @@ function createHandlerFromSchema<TResolvers>(
       replaceSchema,
       replaceDelay,
       withResolvers,
+      withMocks,
     },
   );
 }