graphql-data-generator 0.3.0-alpha.2 → 0.3.0-alpha.20

package/esm/index.js

@@ -1,4 +1,4 @@
 export { init } from "./init.js";
 export { codegen } from "./codegen.js";
-export { proxy } from "./proxy.js";
+export { clear, proxy } from "./proxy.js";
 export { formatCode, toObject } from "./util.js";

package/esm/init.js

@@ -1,11 +1,24 @@
+import * as dntShim from "./_dnt.shims.js";
+import { createRequire } from "node:module";
 import { readFileSync } from "node:fs";
 import { Kind, parse } from "graphql";
-import { dirname, join } from "node:path";
 import { gqlPluckFromCodeStringSync } from "@graphql-tools/graphql-tag-pluck";
 import { operation, proxy, withGetDefaultPatch } from "./proxy.js";
 import { toObject } from "./util.js";
+import { dirname, resolve } from "node:path";
+globalThis.require ??= createRequire(dntShim.Deno.cwd());
 const files = {};
-const loadFile = (path) => files[path] || (files[path] = readFileSync(path, "utf-8").replace(/#import "(.*)"/, (_, fragmentPath) => loadFile(join(dirname(path), fragmentPath))));
+const loadFile = (path) => {
+    if (files[path])
+        return files[path];
+    const raw = files[path] = readFileSync(path, "utf-8");
+    const imports = Array.from(raw.matchAll(/#import "(.*)"/gm), ([, importPath]) => loadFile(require.resolve(importPath.startsWith(".")
+        ? resolve(dntShim.Deno.cwd(), dirname(path), importPath)
+        : importPath, { paths: [dntShim.Deno.cwd()] })));
+    if (!imports.length)
+        return raw;
+    return [raw, ...imports].join("\n\n");
+};
 const getOperationContentMap = {};
 const getOperationContent = (path, operationName) => {
     const existing = getOperationContentMap[path];
@@ -14,14 +27,14 @@ const getOperationContent = (path, operationName) => {
             return existing;
         return getOperationContentMap[path][operationName];
     }
-    const fileContent = readFileSync(path, "utf-8").replace(/#import "(.*)"/, (_, fragmentPath) => loadFile(join(dirname(path), fragmentPath)));
+    const fileContent = loadFile(path);
     try {
         const sources = gqlPluckFromCodeStringSync(path, fileContent);
         getOperationContentMap[path] = Object.fromEntries(sources.map((s) => {
             const document = parse(s);
             const firstOp = document.definitions.find((d) => d.kind === Kind.OPERATION_DEFINITION);
             if (!firstOp)
-                throw new Error(`Cound not find an operation in ${path}`);
+                throw new Error(`Could not find an operation in ${path}`);
             return [firstOp.name?.value, document];
         }));
     }
@@ -30,7 +43,16 @@ const getOperationContent = (path, operationName) => {
     }
     return getOperationContent(path, operationName);
 };
-// - Types will be suffixed with their type: fooQuery or fooMutation
+/**
+ * Initialize the data builder.
+ * @param schema The plain text of your schema.
+ * @param queries List of queries exported from generated types.
+ * @param mutations List of mutations exported from generated types.
+ * @param subscriptions List of subscriptions exported from generated types.
+ * @param types List of types exported from generated types.
+ * @param inputs List of types exported from generated types.
+ * @param scalars A mapping to generate scalar values. Function values will be invoked with their `__typename`.
+ */
 export const init = (schema, queries, mutations, subscriptions, types, inputs, scalars, options) => (fn) => {
     const doc = parse(schema);
     const build = {};
@@ -104,7 +126,6 @@ export const init = (schema, queries, mutations, subscriptions, types, inputs, s
                             : variables,
                     });
                     Error.captureStackTrace(mock, obj.variables);
-                    mock.stack = mock.stack?.slice(6);
                     return mock;
                 },
             },
@@ -125,7 +146,6 @@ export const init = (schema, queries, mutations, subscriptions, types, inputs, s
                             : data,
                     });
                     Error.captureStackTrace(mock, obj.data);
-                    mock.stack = mock.stack?.slice(6);
                     return mock;
                 },
             },
@@ -146,7 +166,6 @@ export const init = (schema, queries, mutations, subscriptions, types, inputs, s
                     const builder = build[operation];
                     const mock = builder(prevInput, patch);
                     Error.captureStackTrace(mock, obj[name]);
-                    mock.stack = mock.stack?.slice(6);
                     return mock;
                 },
             }])));
@@ -158,14 +177,16 @@ export const init = (schema, queries, mutations, subscriptions, types, inputs, s
             if (transforms[name] && "default" in transforms[name]) {
                 patches = [transforms[name].default, ...patches];
             }
-            const { request: { query: parsedQuery, ...request }, ...raw } = operation(doc.definitions, scalars, query, ...patches);
-            const mock = toObject({
-                request: { ...request },
-                ...raw,
+            const { mock, parsedQuery } = withGetDefaultPatch((type) => transforms[type]?.default, () => {
+                const { request: { query: parsedQuery, ...request }, ...raw } = operation(doc.definitions, scalars, query, ...patches);
+                const mock = toObject({
+                    request: { ...request },
+                    ...raw,
+                });
+                return { mock, parsedQuery };
             });
             mock.request.query = parsedQuery;
             Error.captureStackTrace(mock, builder);
-            mock.stack = mock.stack?.slice(6);
             return addOperationTransforms(name, options?.finalizeOperation
                 ? options.finalizeOperation(mock)
                 : mock);

package/esm/jest.js

@@ -1,10 +1,10 @@
 import * as dntShim from "./_dnt.shims.js";
 import React, { useMemo } from "react";
-import { ApolloLink } from "@apollo/client";
-import { ErrorLink } from "@apollo/client/link/error/index.js";
-import { MockedProvider, MockLink, } from "@apollo/client/testing/index.js";
+import { ApolloLink, useApolloClient, } from "@apollo/client";
+import { onError } from "@apollo/client/link/error";
+import { MockedProvider as ApolloMockedProvider, MockLink, } from "@apollo/client/testing";
 import { waitFor } from "@testing-library/dom";
-import { print } from "graphql";
+import { Kind, print } from "graphql";
 import { diff as jestDiff } from "jest-diff";
 let currentSpecResult;
 jasmine.getEnv().addReporter({
@@ -12,25 +12,26 @@ jasmine.getEnv().addReporter({
 });
 const afterTest = [];
 afterEach(async () => {
-    for (const hook of afterTest)
+    const hooks = afterTest.splice(0);
+    for (const hook of hooks)
         await hook();
-    afterTest.splice(0);
 });
-const diff = (a, b) => jestDiff(a, b, {
-    omitAnnotationLines: true,
-    // aColor: green,
-    // bColor: red,
-    // changeColor: inverse,
-    // commonColor: dim,
-    // patchColor: yellow,
-})
+const diff = (a, b) => jestDiff(a, b, { omitAnnotationLines: true })
     ?.replace(/\w+ \{/g, "{") // Remove class names
     .replace(/\w+ \[/g, "["); // Remove array class names
+const getOperationDefinition = (document) => document.definitions.find((d) => d.kind === Kind.OPERATION_DEFINITION);
+const getOperationType = (operation) => getOperationDefinition(operation.query)?.operation ??
+    "<unknown operation type>";
+const getOperationName = (document) => getOperationDefinition(document)?.name?.value;
+const getOperationInfo = (document) => {
+    const def = getOperationDefinition(document);
+    return {
+        name: def?.name?.value ?? "<unknown operation>",
+        operationType: def?.operation ?? "<unknown operation type>",
+    };
+};
 const getErrorMessage = (operation, mockLink) => {
-    const definition = operation.query.definitions[0];
-    const operationType = definition.kind === "OperationDefinition"
-        ? definition.operation
-        : "<unknown operation type>";
+    const operationType = getOperationType(operation);
     const key = JSON.stringify({
         query: print(operation.query),
     });
@@ -38,7 +39,7 @@ const getErrorMessage = (operation, mockLink) => {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     const alts = (mockLink
         .mockedResponsesByKey[key] ?? []);
-    let errorMessage = `Expected GraphQL ${operationType} ${operation.operationName} to have been mocked`;
+    let errorMessage = `Expected ${operationType} ${operation.operationName} to have been mocked`;
     if (alts.length || Object.keys(operation.variables).length > 0) {
         errorMessage += ` with variables ${dntShim.Deno.inspect(operation.variables, { depth: Infinity, colors: true })}`;
     }
@@ -61,59 +62,162 @@ const getErrorMessage = (operation, mockLink) => {
             : undefined,
     };
 };
-export const MockProvider = ({ mocks, stack, children }) => {
-    const observableMocks = useMemo(() => {
-        const observableMocks = mocks.map((m) => ({
-            ...m,
-            stack: m.stack,
-            result: Object.assign(jest.fn(() => m.result), m.result),
-        }));
-        afterTest.push(async () => {
+let _failRefetchWarnings = false;
+/**
+ * In older versions of `@apollo/client`, refetches with missing mocks trigger
+ * warnings instead of being treated as standard missing mocks.
+ * This utility converts those warnings into failures and ensures watch queries
+ * are installed for queries with `watch: true`. This must be set when
+ * `MockProvider` is mounted.
+ *
+ * This is not required on modern version of `@apollo/client`.
+ */
+export const failRefetchWarnings = (value = true) => _failRefetchWarnings = value;
+let _allowMissingMocks = false;
+/**
+ * Allows missing mocks, resulting in tests passing. Usage is intended to ease
+ * migration.
+ */
+export const allowMissingMocks = (value) => _allowMissingMocks = value;
+const AutoWatch = ({ mocks }) => {
+    const client = useApolloClient();
+    for (const mock of mocks)
+        if (mock.watch)
+            client.watchQuery(mock.request);
+    return null;
+};
+let lastMocks = [];
+// deno-lint-ignore ban-types
+const getStack = (to) => {
+    const obj = {};
+    Error.captureStackTrace(obj, to);
+    return obj.stack;
+};
+const _waitForMocks = async (mocks, cause) => {
+    for (const mock of mocks) {
+        if (mock.optional || mock.error)
+            continue;
+        await waitFor(() => {
             if (currentSpecResult.failedExpectations.length)
                 return;
-            for (const mock of observableMocks) {
-                if ("optional" in mock && mock.optional || mock.error)
-                    continue;
-                await waitFor(() => {
-                    if (currentSpecResult.failedExpectations.length)
-                        return;
-                    if (mock.result.mock.calls.length === 0) {
-                        const err = new Error(`Expected to have used mock ${mock.request.variables
-                            ? ` with variables ${dntShim.Deno.inspect(mock.request.variables, {
-                                depth: Infinity,
-                                colors: true,
-                            })}`
-                            : ""}`);
-                        err.stack = `${err.message}\n${mock.stack ?? stack ?? err.stack?.slice(6)}`;
-                        throw err;
-                    }
-                }, { onTimeout: (e) => e });
+            if (mock.result.mock.calls.length === 0) {
+                const { name, operationType } = getOperationInfo(mock.request.query);
+                const err = new Error(`Expected to have used ${operationType} ${name}${mock.request.variables
+                    ? ` with variables ${dntShim.Deno.inspect(mock.request.variables, {
+                        depth: Infinity,
+                        colors: true,
+                    })}`
+                    : ""}`);
+                if (mock.stack) {
+                    err.stack = `${mock.stack}${cause ? `\nCaused by: ${cause}` : ""}`;
+                }
+                else if (cause)
+                    err.stack = cause;
+                throw err;
             }
         });
+    }
+};
+/**
+ * Wait for mocks to have been used.
+ * @param mock If `undefined`, waits for all mocks. If a number, waits fort he first `mocks` mocks. If a string, waits for all mocks up until and including that mock.
+ * @param offset If `mocks` is a string, grabs the `offset`th mock of that name (e.g., the third `getReport` mock)
+ */
+export const waitForMocks = async (mock = lastMocks.length, offset = 0) => {
+    if (typeof mock === "string") {
+        const matches = lastMocks.map((m, i) => [m, i])
+            .filter(([m]) => getOperationName(m.request.query) === mock);
+        if (matches.length <= offset) {
+            fail({
+                name: "Error",
+                message: `Expected mock ${mock} to have been mocked`,
+                stack: getStack(waitForMocks),
+            });
+        }
+        expect(matches.length).toBeGreaterThan(offset);
+        mock = matches[offset][1] + 1;
+    }
+    await _waitForMocks(lastMocks.slice(0, mock), getStack(waitForMocks));
+};
+/**
+ * A wrapper for `@apollo/client/testing`, this component will assert all
+ * requests have matching mocks and all defined mocks are used unless marked
+ * `optional`.
+ */
+export const MockedProvider = ({ mocks, stack: renderStack, children, link: passedLink, ...rest }) => {
+    const observableMocks = useMemo(() => {
+        const observableMocks = mocks.flatMap((m) => [
+            {
+                ...m,
+                stack: m.stack,
+                result: Object.assign(jest.fn(() => m.result), m.result),
+            },
+            ...(m.watch
+                ? [{
+                        ...m,
+                        stack: m.stack,
+                        result: Object.assign(jest.fn(() => m.result), m.result),
+                        watch: false,
+                    }]
+                : []),
+        ]);
+        lastMocks = observableMocks;
+        afterTest.push(() => _waitForMocks(lastMocks, renderStack));
         return observableMocks;
     }, [mocks]);
     const link = useMemo(() => {
         const mockLink = new MockLink(observableMocks);
         mockLink.showWarnings = false;
-        const errorLoggingLink = new ErrorLink(({ networkError, operation }) => {
-            if (!networkError?.message.includes("No more mocked responses"))
+        const errorLoggingLink = onError(({ networkError, operation }) => {
+            if (_allowMissingMocks ||
+                !networkError?.message?.includes("No more mocked responses"))
                 return;
             const { message, stack: altStack } = getErrorMessage(operation, mockLink);
             try {
                 networkError.message = message;
-                const finalStack = altStack ?? stack
-                    ? `${message}\n${altStack ?? stack}`
-                    : undefined;
-                if (finalStack)
-                    networkError.stack = finalStack;
-                fail({ name: "Error", message, stack: finalStack });
+                if (altStack) {
+                    networkError.stack = renderStack
+                        ? `${altStack}\nCaused By: ${renderStack}`
+                        : altStack;
+                }
+                else if (renderStack)
+                    networkError.stack = renderStack;
+                fail({ name: "Error", message, stack: networkError.stack });
             }
             catch {
                 // fail both throws and marks the test as failed in jest; we only need the latter
             }
         });
-        // @ts-ignore It's fine
-        return ApolloLink.from([errorLoggingLink, mockLink]);
-    }, [observableMocks, stack]);
-    return React.createElement(MockedProvider, { link: link }, children);
+        return ApolloLink.from([
+            errorLoggingLink,
+            mockLink,
+            ...(passedLink ? [passedLink] : []),
+        ]);
+    }, [observableMocks, renderStack]);
+    if (_failRefetchWarnings) {
+        const oldWarn = console.warn.bind(console.warn);
+        console.warn = (message, operation, ...etc) => {
+            if (message !==
+                'Unknown query named "%s" requested in refetchQueries in options.include array') {
+                return oldWarn(message, operation, ...etc);
+            }
+            try {
+                fail({
+                    name: "Error",
+                    message: `Expected query ${operation} requested in refetchQueries options.include array to have bene mocked`,
+                    stack: renderStack,
+                });
+            }
+            catch {
+                // eat
+            }
+        };
+        afterTest.push(() => {
+            console.warn = oldWarn;
+        });
+    }
+    return (React.createElement(ApolloMockedProvider, { ...rest, link: link },
+        React.createElement(React.Fragment, null,
+            React.createElement(AutoWatch, { mocks: observableMocks }),
+            children)));
 };

package/esm/proxy.js

@@ -1,5 +1,14 @@
 import { Kind, Location, parse } from "graphql";
 import { absurd } from "./util.js";
+/**
+ * Updates the value to the base value, bypassing default transformations.
+ * Useful to clear optional inputs, resulting in `undefined` rather than `null`.
+ */
+export const clear = Symbol("clear");
+/**
+ * Called to insert default patches at the front of nested objects (props &
+ * arrays). `build` will insert default patches for top-level objects.
+ */
 let getDefaultPatch = (__typename) => undefined;
 export const withGetDefaultPatch = (newGetDefaultPatch, fn) => {
     const prev = getDefaultPatch;
@@ -131,7 +140,7 @@ const resolveConcreteType = (definitions, definition, patch, prev) => {
 };
 const resolveValue = (definitions, scalars, type, ctx) => {
     if (type.kind !== Kind.NON_NULL_TYPE)
-        return null;
+        return ctx.input ? undefined : null;
     if (type.type.kind === Kind.LIST_TYPE)
         return [];
     type = type.type;
@@ -161,7 +170,7 @@ const resolveValue = (definitions, scalars, type, ctx) => {
             ? rawDefaultPatch(base)
             : rawDefaultPatch;
         if (defaultPatch) {
-            return _proxy(definitions, scalars, fieldTypeDefinitions[0].name.value, [base, defaultPatch]);
+            return _proxy(definitions, scalars, fieldTypeDefinitions[0].name.value, [base, defaultPatch], { selectionSet: ctx.selectionSet });
         }
         return base;
     }
@@ -258,7 +267,7 @@ const selectionSetToKeys = (definitions, selectionSet, typename) => {
     }
     return keys;
 };
-const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = resolveType(definitions, path), selectionSet, } = {}) => {
+let _proxy = (definitions, scalars, path, patches, { prev, resolvedType = resolveType(definitions, path), selectionSet, } = {}) => {
     const { parent = path, definition } = resolvedType;
     let type = resolvedType.type;
     if (!prev && patches.length) {
@@ -270,9 +279,18 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
     const patch = typeof rawPatch === "function"
         ? prev ? rawPatch(prev) : undefined
         : rawPatch;
+    if (patch === clear) {
+        return _proxy(definitions, scalars, path, [], { selectionSet });
+    }
     if (type.kind !== Kind.NON_NULL_TYPE) {
-        if (!patches.length)
-            return (prev ?? null);
+        if (!patches.length) {
+            return (prev ??
+                (resolveType(definitions, path.split(".").slice(0, -1).join("."))
+                    .definition?.kind ===
+                    Kind.INPUT_OBJECT_TYPE_DEFINITION
+                    ? undefined
+                    : null));
+        }
     }
     else
         type = type.type;
@@ -280,6 +298,9 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
         if (!patches.length)
             return (prev ?? []);
         const value = (patches.at(-1) ?? []);
+        if (value === clear) {
+            return _proxy(definitions, scalars, path, [], { selectionSet });
+        }
         const previousArray = (prev ?? []);
         const lastIndex = Math.max(previousArray.length - 1, 0);
         const nextIndex = previousArray.length;
@@ -297,7 +318,14 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
         }));
         const arr = [];
         for (let i = 0; i < length; i++) {
+            const nestType = type.type.kind === Kind.NAMED_TYPE
+                ? type.type.name.value
+                : type.type.kind === Kind.NON_NULL_TYPE &&
+                    type.type.type.kind === Kind.NAMED_TYPE
+                    ? type.type.type.name.value
+                    : undefined;
             arr[i] = _proxy(definitions, scalars, `${path}.${i}`, [
+                nestType && !previousArray[i] ? getDefaultPatch(nestType) : undefined,
                 value[i],
                 i === lastIndex ? value.last : undefined,
                 i === nextIndex ? value.next : undefined,
@@ -335,6 +363,9 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                     const value = typeof rawValue === "function"
                         ? rawValue(prev)
                         : rawValue;
+                    if (value === clear) {
+                        return target[prop] = _proxy(definitions, scalars, `${path}.${prop}`, [], { selectionSet });
+                    }
                     if (value && typeof value === "object") {
                         const nonNullFieldType = field.type.kind === Kind.NON_NULL_TYPE
                             ? field.type.type
@@ -373,6 +404,7 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                     selectionSet: selectionSet
                         ? getSelectionSetSelection(definitions, selectionSet, prop)
                         : undefined,
+                    input: definition.kind === Kind.INPUT_OBJECT_TYPE_DEFINITION,
                 });
             };
             const keys = [
@@ -466,7 +498,7 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                     mock.variables[name] = mockPrev.variables[name];
                     continue;
                 }
-                const result = resolveValue(definitions, scalars, variableDefinition.type, { hostType: `${path}Variables`, prop: name });
+                const result = resolveValue(definitions, scalars, variableDefinition.type, { hostType: `${path}Variables`, prop: name, input: false });
                 if (result != null) {
                     if (!mock.variables)
                         mock.variables = {};
@@ -489,6 +521,10 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                         // Query, Mutation, Subscription
                         const operationKey = definition.operation[0].toUpperCase() +
                             definition.operation.slice(1);
+                        if (value === clear) {
+                            mock.data[prop] = _proxy(definitions, scalars, `${operationKey}.${selection.name.value}`, [], { selectionSet: selection.selectionSet });
+                            continue;
+                        }
                         mock.data[prop] = _proxy(definitions, scalars, `${operationKey}.${selection.name.value}`, value != null ? [value] : [], {
                             prev: mockPrev?.data?.[prop],
                             selectionSet: selection.selectionSet,
@@ -534,6 +570,14 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
             const value = typeof scalar === "function" ? scalar(parent) : scalar;
             return value;
         }
+        case Kind.ENUM_TYPE_DEFINITION: {
+            const patch = patches[patches.length - 1];
+            if (patch != null)
+                return patch;
+            if (prev != null)
+                return prev;
+            return definition.values?.[0]?.name.value;
+        }
         default:
             throw new Error(`Unhandled definition kind '${definition.kind}'`);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "graphql-data-generator",
-  "version": "0.3.0-alpha.2",
+  "version": "0.3.0-alpha.20",
   "repository": {
     "type": "git",
     "url": "git+https://github.com/voces/graphql-data-generator.git"
@@ -35,6 +35,16 @@
   "bin": {
     "graphql-data-generator": "./esm/cli.js"
   },
+  "typesVersions": {
+    "*": {
+      "script/jest": [
+        "types/jest.d.ts"
+      ],
+      "script/plugin": [
+        "types/plugin.d.ts"
+      ]
+    }
+  },
   "dependencies": {
     "@graphql-codegen/visitor-plugin-common": "*",
     "fast-glob": "*",

package/script/index.js

@@ -1,11 +1,12 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.toObject = exports.formatCode = exports.proxy = exports.codegen = exports.init = void 0;
+exports.toObject = exports.formatCode = exports.proxy = exports.clear = exports.codegen = exports.init = void 0;
 var init_js_1 = require("./init.js");
 Object.defineProperty(exports, "init", { enumerable: true, get: function () { return init_js_1.init; } });
 var codegen_js_1 = require("./codegen.js");
 Object.defineProperty(exports, "codegen", { enumerable: true, get: function () { return codegen_js_1.codegen; } });
 var proxy_js_1 = require("./proxy.js");
+Object.defineProperty(exports, "clear", { enumerable: true, get: function () { return proxy_js_1.clear; } });
 Object.defineProperty(exports, "proxy", { enumerable: true, get: function () { return proxy_js_1.proxy; } });
 var util_js_1 = require("./util.js");
 Object.defineProperty(exports, "formatCode", { enumerable: true, get: function () { return util_js_1.formatCode; } });

package/script/init.js

@@ -1,14 +1,50 @@
 "use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || function (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.init = void 0;
+const dntShim = __importStar(require("./_dnt.shims.js"));
+const node_module_1 = require("node:module");
 const node_fs_1 = require("node:fs");
 const graphql_1 = require("graphql");
-const node_path_1 = require("node:path");
 const graphql_tag_pluck_1 = require("@graphql-tools/graphql-tag-pluck");
 const proxy_js_1 = require("./proxy.js");
 const util_js_1 = require("./util.js");
+const node_path_1 = require("node:path");
+globalThis.require ??= (0, node_module_1.createRequire)(dntShim.Deno.cwd());
 const files = {};
-const loadFile = (path) => files[path] || (files[path] = (0, node_fs_1.readFileSync)(path, "utf-8").replace(/#import "(.*)"/, (_, fragmentPath) => loadFile((0, node_path_1.join)((0, node_path_1.dirname)(path), fragmentPath))));
+const loadFile = (path) => {
+    if (files[path])
+        return files[path];
+    const raw = files[path] = (0, node_fs_1.readFileSync)(path, "utf-8");
+    const imports = Array.from(raw.matchAll(/#import "(.*)"/gm), ([, importPath]) => loadFile(require.resolve(importPath.startsWith(".")
+        ? (0, node_path_1.resolve)(dntShim.Deno.cwd(), (0, node_path_1.dirname)(path), importPath)
+        : importPath, { paths: [dntShim.Deno.cwd()] })));
+    if (!imports.length)
+        return raw;
+    return [raw, ...imports].join("\n\n");
+};
 const getOperationContentMap = {};
 const getOperationContent = (path, operationName) => {
     const existing = getOperationContentMap[path];
@@ -17,14 +53,14 @@ const getOperationContent = (path, operationName) => {
             return existing;
         return getOperationContentMap[path][operationName];
     }
-    const fileContent = (0, node_fs_1.readFileSync)(path, "utf-8").replace(/#import "(.*)"/, (_, fragmentPath) => loadFile((0, node_path_1.join)((0, node_path_1.dirname)(path), fragmentPath)));
+    const fileContent = loadFile(path);
     try {
         const sources = (0, graphql_tag_pluck_1.gqlPluckFromCodeStringSync)(path, fileContent);
         getOperationContentMap[path] = Object.fromEntries(sources.map((s) => {
             const document = (0, graphql_1.parse)(s);
             const firstOp = document.definitions.find((d) => d.kind === graphql_1.Kind.OPERATION_DEFINITION);
             if (!firstOp)
-                throw new Error(`Cound not find an operation in ${path}`);
+                throw new Error(`Could not find an operation in ${path}`);
             return [firstOp.name?.value, document];
         }));
     }
@@ -33,7 +69,16 @@ const getOperationContent = (path, operationName) => {
     }
     return getOperationContent(path, operationName);
 };
-// - Types will be suffixed with their type: fooQuery or fooMutation
+/**
+ * Initialize the data builder.
+ * @param schema The plain text of your schema.
+ * @param queries List of queries exported from generated types.
+ * @param mutations List of mutations exported from generated types.
+ * @param subscriptions List of subscriptions exported from generated types.
+ * @param types List of types exported from generated types.
+ * @param inputs List of types exported from generated types.
+ * @param scalars A mapping to generate scalar values. Function values will be invoked with their `__typename`.
+ */
 const init = (schema, queries, mutations, subscriptions, types, inputs, scalars, options) => (fn) => {
     const doc = (0, graphql_1.parse)(schema);
     const build = {};
@@ -107,7 +152,6 @@ const init = (schema, queries, mutations, subscriptions, types, inputs, scalars,
                             : variables,
                     });
                     Error.captureStackTrace(mock, obj.variables);
-                    mock.stack = mock.stack?.slice(6);
                     return mock;
                 },
             },
@@ -128,7 +172,6 @@ const init = (schema, queries, mutations, subscriptions, types, inputs, scalars,
                             : data,
                     });
                     Error.captureStackTrace(mock, obj.data);
-                    mock.stack = mock.stack?.slice(6);
                     return mock;
                 },
             },
@@ -149,7 +192,6 @@ const init = (schema, queries, mutations, subscriptions, types, inputs, scalars,
                     const builder = build[operation];
                     const mock = builder(prevInput, patch);
                     Error.captureStackTrace(mock, obj[name]);
-                    mock.stack = mock.stack?.slice(6);
                     return mock;
                 },
             }])));
@@ -161,14 +203,16 @@ const init = (schema, queries, mutations, subscriptions, types, inputs, scalars,
             if (transforms[name] && "default" in transforms[name]) {
                 patches = [transforms[name].default, ...patches];
             }
-            const { request: { query: parsedQuery, ...request }, ...raw } = (0, proxy_js_1.operation)(doc.definitions, scalars, query, ...patches);
-            const mock = (0, util_js_1.toObject)({
-                request: { ...request },
-                ...raw,
+            const { mock, parsedQuery } = (0, proxy_js_1.withGetDefaultPatch)((type) => transforms[type]?.default, () => {
+                const { request: { query: parsedQuery, ...request }, ...raw } = (0, proxy_js_1.operation)(doc.definitions, scalars, query, ...patches);
+                const mock = (0, util_js_1.toObject)({
+                    request: { ...request },
+                    ...raw,
+                });
+                return { mock, parsedQuery };
             });
             mock.request.query = parsedQuery;
             Error.captureStackTrace(mock, builder);
-            mock.stack = mock.stack?.slice(6);
             return addOperationTransforms(name, options?.finalizeOperation
                 ? options.finalizeOperation(mock)
                 : mock);

package/script/jest.js

@@ -23,12 +23,12 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.MockProvider = void 0;
+exports.MockedProvider = exports.waitForMocks = exports.allowMissingMocks = exports.failRefetchWarnings = void 0;
 const dntShim = __importStar(require("./_dnt.shims.js"));
 const react_1 = __importStar(require("react"));
 const client_1 = require("@apollo/client");
-const index_js_1 = require("@apollo/client/link/error/index.js");
-const index_js_2 = require("@apollo/client/testing/index.js");
+const error_1 = require("@apollo/client/link/error");
+const testing_1 = require("@apollo/client/testing");
 const dom_1 = require("@testing-library/dom");
 const graphql_1 = require("graphql");
 const jest_diff_1 = require("jest-diff");
@@ -38,25 +38,26 @@ jasmine.getEnv().addReporter({
 });
 const afterTest = [];
 afterEach(async () => {
-    for (const hook of afterTest)
+    const hooks = afterTest.splice(0);
+    for (const hook of hooks)
         await hook();
-    afterTest.splice(0);
 });
-const diff = (a, b) => (0, jest_diff_1.diff)(a, b, {
-    omitAnnotationLines: true,
-    // aColor: green,
-    // bColor: red,
-    // changeColor: inverse,
-    // commonColor: dim,
-    // patchColor: yellow,
-})
+const diff = (a, b) => (0, jest_diff_1.diff)(a, b, { omitAnnotationLines: true })
     ?.replace(/\w+ \{/g, "{") // Remove class names
     .replace(/\w+ \[/g, "["); // Remove array class names
+const getOperationDefinition = (document) => document.definitions.find((d) => d.kind === graphql_1.Kind.OPERATION_DEFINITION);
+const getOperationType = (operation) => getOperationDefinition(operation.query)?.operation ??
+    "<unknown operation type>";
+const getOperationName = (document) => getOperationDefinition(document)?.name?.value;
+const getOperationInfo = (document) => {
+    const def = getOperationDefinition(document);
+    return {
+        name: def?.name?.value ?? "<unknown operation>",
+        operationType: def?.operation ?? "<unknown operation type>",
+    };
+};
 const getErrorMessage = (operation, mockLink) => {
-    const definition = operation.query.definitions[0];
-    const operationType = definition.kind === "OperationDefinition"
-        ? definition.operation
-        : "<unknown operation type>";
+    const operationType = getOperationType(operation);
     const key = JSON.stringify({
         query: (0, graphql_1.print)(operation.query),
     });
@@ -64,7 +65,7 @@ const getErrorMessage = (operation, mockLink) => {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     const alts = (mockLink
         .mockedResponsesByKey[key] ?? []);
-    let errorMessage = `Expected GraphQL ${operationType} ${operation.operationName} to have been mocked`;
+    let errorMessage = `Expected ${operationType} ${operation.operationName} to have been mocked`;
     if (alts.length || Object.keys(operation.variables).length > 0) {
         errorMessage += ` with variables ${dntShim.Deno.inspect(operation.variables, { depth: Infinity, colors: true })}`;
     }
@@ -87,60 +88,166 @@ const getErrorMessage = (operation, mockLink) => {
             : undefined,
     };
 };
-const MockProvider = ({ mocks, stack, children }) => {
-    const observableMocks = (0, react_1.useMemo)(() => {
-        const observableMocks = mocks.map((m) => ({
-            ...m,
-            stack: m.stack,
-            result: Object.assign(jest.fn(() => m.result), m.result),
-        }));
-        afterTest.push(async () => {
+let _failRefetchWarnings = false;
+/**
+ * In older versions of `@apollo/client`, refetches with missing mocks trigger
+ * warnings instead of being treated as standard missing mocks.
+ * This utility converts those warnings into failures and ensures watch queries
+ * are installed for queries with `watch: true`. This must be set when
+ * `MockProvider` is mounted.
+ *
+ * This is not required on modern version of `@apollo/client`.
+ */
+const failRefetchWarnings = (value = true) => _failRefetchWarnings = value;
+exports.failRefetchWarnings = failRefetchWarnings;
+let _allowMissingMocks = false;
+/**
+ * Allows missing mocks, resulting in tests passing. Usage is intended to ease
+ * migration.
+ */
+const allowMissingMocks = (value) => _allowMissingMocks = value;
+exports.allowMissingMocks = allowMissingMocks;
+const AutoWatch = ({ mocks }) => {
+    const client = (0, client_1.useApolloClient)();
+    for (const mock of mocks)
+        if (mock.watch)
+            client.watchQuery(mock.request);
+    return null;
+};
+let lastMocks = [];
+// deno-lint-ignore ban-types
+const getStack = (to) => {
+    const obj = {};
+    Error.captureStackTrace(obj, to);
+    return obj.stack;
+};
+const _waitForMocks = async (mocks, cause) => {
+    for (const mock of mocks) {
+        if (mock.optional || mock.error)
+            continue;
+        await (0, dom_1.waitFor)(() => {
             if (currentSpecResult.failedExpectations.length)
                 return;
-            for (const mock of observableMocks) {
-                if ("optional" in mock && mock.optional || mock.error)
-                    continue;
-                await (0, dom_1.waitFor)(() => {
-                    if (currentSpecResult.failedExpectations.length)
-                        return;
-                    if (mock.result.mock.calls.length === 0) {
-                        const err = new Error(`Expected to have used mock ${mock.request.variables
-                            ? ` with variables ${dntShim.Deno.inspect(mock.request.variables, {
-                                depth: Infinity,
-                                colors: true,
-                            })}`
-                            : ""}`);
-                        err.stack = `${err.message}\n${mock.stack ?? stack ?? err.stack?.slice(6)}`;
-                        throw err;
-                    }
-                }, { onTimeout: (e) => e });
+            if (mock.result.mock.calls.length === 0) {
+                const { name, operationType } = getOperationInfo(mock.request.query);
+                const err = new Error(`Expected to have used ${operationType} ${name}${mock.request.variables
+                    ? ` with variables ${dntShim.Deno.inspect(mock.request.variables, {
+                        depth: Infinity,
+                        colors: true,
+                    })}`
+                    : ""}`);
+                if (mock.stack) {
+                    err.stack = `${mock.stack}${cause ? `\nCaused by: ${cause}` : ""}`;
+                }
+                else if (cause)
+                    err.stack = cause;
+                throw err;
             }
         });
+    }
+};
+/**
+ * Wait for mocks to have been used.
+ * @param mock If `undefined`, waits for all mocks. If a number, waits fort he first `mocks` mocks. If a string, waits for all mocks up until and including that mock.
+ * @param offset If `mocks` is a string, grabs the `offset`th mock of that name (e.g., the third `getReport` mock)
+ */
+const waitForMocks = async (mock = lastMocks.length, offset = 0) => {
+    if (typeof mock === "string") {
+        const matches = lastMocks.map((m, i) => [m, i])
+            .filter(([m]) => getOperationName(m.request.query) === mock);
+        if (matches.length <= offset) {
+            fail({
+                name: "Error",
+                message: `Expected mock ${mock} to have been mocked`,
+                stack: getStack(exports.waitForMocks),
+            });
+        }
+        expect(matches.length).toBeGreaterThan(offset);
+        mock = matches[offset][1] + 1;
+    }
+    await _waitForMocks(lastMocks.slice(0, mock), getStack(exports.waitForMocks));
+};
+exports.waitForMocks = waitForMocks;
+/**
+ * A wrapper for `@apollo/client/testing`, this component will assert all
+ * requests have matching mocks and all defined mocks are used unless marked
+ * `optional`.
+ */
+const MockedProvider = ({ mocks, stack: renderStack, children, link: passedLink, ...rest }) => {
+    const observableMocks = (0, react_1.useMemo)(() => {
+        const observableMocks = mocks.flatMap((m) => [
+            {
+                ...m,
+                stack: m.stack,
+                result: Object.assign(jest.fn(() => m.result), m.result),
+            },
+            ...(m.watch
+                ? [{
+                        ...m,
+                        stack: m.stack,
+                        result: Object.assign(jest.fn(() => m.result), m.result),
+                        watch: false,
+                    }]
+                : []),
+        ]);
+        lastMocks = observableMocks;
+        afterTest.push(() => _waitForMocks(lastMocks, renderStack));
         return observableMocks;
     }, [mocks]);
     const link = (0, react_1.useMemo)(() => {
-        const mockLink = new index_js_2.MockLink(observableMocks);
+        const mockLink = new testing_1.MockLink(observableMocks);
         mockLink.showWarnings = false;
-        const errorLoggingLink = new index_js_1.ErrorLink(({ networkError, operation }) => {
-            if (!networkError?.message.includes("No more mocked responses"))
+        const errorLoggingLink = (0, error_1.onError)(({ networkError, operation }) => {
+            if (_allowMissingMocks ||
+                !networkError?.message?.includes("No more mocked responses"))
                 return;
             const { message, stack: altStack } = getErrorMessage(operation, mockLink);
             try {
                 networkError.message = message;
-                const finalStack = altStack ?? stack
-                    ? `${message}\n${altStack ?? stack}`
-                    : undefined;
-                if (finalStack)
-                    networkError.stack = finalStack;
-                fail({ name: "Error", message, stack: finalStack });
+                if (altStack) {
+                    networkError.stack = renderStack
+                        ? `${altStack}\nCaused By: ${renderStack}`
+                        : altStack;
+                }
+                else if (renderStack)
+                    networkError.stack = renderStack;
+                fail({ name: "Error", message, stack: networkError.stack });
             }
             catch {
                 // fail both throws and marks the test as failed in jest; we only need the latter
             }
         });
-        // @ts-ignore It's fine
-        return client_1.ApolloLink.from([errorLoggingLink, mockLink]);
-    }, [observableMocks, stack]);
-    return react_1.default.createElement(index_js_2.MockedProvider, { link: link }, children);
+        return client_1.ApolloLink.from([
+            errorLoggingLink,
+            mockLink,
+            ...(passedLink ? [passedLink] : []),
+        ]);
+    }, [observableMocks, renderStack]);
+    if (_failRefetchWarnings) {
+        const oldWarn = console.warn.bind(console.warn);
+        console.warn = (message, operation, ...etc) => {
+            if (message !==
+                'Unknown query named "%s" requested in refetchQueries in options.include array') {
+                return oldWarn(message, operation, ...etc);
+            }
+            try {
+                fail({
+                    name: "Error",
+                    message: `Expected query ${operation} requested in refetchQueries options.include array to have bene mocked`,
+                    stack: renderStack,
+                });
+            }
+            catch {
+                // eat
+            }
+        };
+        afterTest.push(() => {
+            console.warn = oldWarn;
+        });
+    }
+    return (react_1.default.createElement(testing_1.MockedProvider, { ...rest, link: link },
+        react_1.default.createElement(react_1.default.Fragment, null,
+            react_1.default.createElement(AutoWatch, { mocks: observableMocks }),
+            children)));
 };
-exports.MockProvider = MockProvider;
+exports.MockedProvider = MockedProvider;

package/script/proxy.js

@@ -1,8 +1,17 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.operation = exports.proxy = exports.withGetDefaultPatch = void 0;
+exports.operation = exports.proxy = exports.withGetDefaultPatch = exports.clear = void 0;
 const graphql_1 = require("graphql");
 const util_js_1 = require("./util.js");
+/**
+ * Updates the value to the base value, bypassing default transformations.
+ * Useful to clear optional inputs, resulting in `undefined` rather than `null`.
+ */
+exports.clear = Symbol("clear");
+/**
+ * Called to insert default patches at the front of nested objects (props &
+ * arrays). `build` will insert default patches for top-level objects.
+ */
 let getDefaultPatch = (__typename) => undefined;
 const withGetDefaultPatch = (newGetDefaultPatch, fn) => {
     const prev = getDefaultPatch;
@@ -135,7 +144,7 @@ const resolveConcreteType = (definitions, definition, patch, prev) => {
 };
 const resolveValue = (definitions, scalars, type, ctx) => {
     if (type.kind !== graphql_1.Kind.NON_NULL_TYPE)
-        return null;
+        return ctx.input ? undefined : null;
     if (type.type.kind === graphql_1.Kind.LIST_TYPE)
         return [];
     type = type.type;
@@ -165,7 +174,7 @@ const resolveValue = (definitions, scalars, type, ctx) => {
             ? rawDefaultPatch(base)
             : rawDefaultPatch;
         if (defaultPatch) {
-            return _proxy(definitions, scalars, fieldTypeDefinitions[0].name.value, [base, defaultPatch]);
+            return _proxy(definitions, scalars, fieldTypeDefinitions[0].name.value, [base, defaultPatch], { selectionSet: ctx.selectionSet });
         }
         return base;
     }
@@ -262,7 +271,7 @@ const selectionSetToKeys = (definitions, selectionSet, typename) => {
     }
     return keys;
 };
-const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = resolveType(definitions, path), selectionSet, } = {}) => {
+let _proxy = (definitions, scalars, path, patches, { prev, resolvedType = resolveType(definitions, path), selectionSet, } = {}) => {
     const { parent = path, definition } = resolvedType;
     let type = resolvedType.type;
     if (!prev && patches.length) {
@@ -274,9 +283,18 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
     const patch = typeof rawPatch === "function"
         ? prev ? rawPatch(prev) : undefined
         : rawPatch;
+    if (patch === exports.clear) {
+        return _proxy(definitions, scalars, path, [], { selectionSet });
+    }
     if (type.kind !== graphql_1.Kind.NON_NULL_TYPE) {
-        if (!patches.length)
-            return (prev ?? null);
+        if (!patches.length) {
+            return (prev ??
+                (resolveType(definitions, path.split(".").slice(0, -1).join("."))
+                    .definition?.kind ===
+                    graphql_1.Kind.INPUT_OBJECT_TYPE_DEFINITION
+                    ? undefined
+                    : null));
+        }
     }
     else
         type = type.type;
@@ -284,6 +302,9 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
         if (!patches.length)
             return (prev ?? []);
         const value = (patches.at(-1) ?? []);
+        if (value === exports.clear) {
+            return _proxy(definitions, scalars, path, [], { selectionSet });
+        }
         const previousArray = (prev ?? []);
         const lastIndex = Math.max(previousArray.length - 1, 0);
         const nextIndex = previousArray.length;
@@ -301,7 +322,14 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
         }));
         const arr = [];
         for (let i = 0; i < length; i++) {
+            const nestType = type.type.kind === graphql_1.Kind.NAMED_TYPE
+                ? type.type.name.value
+                : type.type.kind === graphql_1.Kind.NON_NULL_TYPE &&
+                    type.type.type.kind === graphql_1.Kind.NAMED_TYPE
+                    ? type.type.type.name.value
+                    : undefined;
             arr[i] = _proxy(definitions, scalars, `${path}.${i}`, [
+                nestType && !previousArray[i] ? getDefaultPatch(nestType) : undefined,
                 value[i],
                 i === lastIndex ? value.last : undefined,
                 i === nextIndex ? value.next : undefined,
@@ -339,6 +367,9 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                     const value = typeof rawValue === "function"
                         ? rawValue(prev)
                         : rawValue;
+                    if (value === exports.clear) {
+                        return target[prop] = _proxy(definitions, scalars, `${path}.${prop}`, [], { selectionSet });
+                    }
                     if (value && typeof value === "object") {
                         const nonNullFieldType = field.type.kind === graphql_1.Kind.NON_NULL_TYPE
                             ? field.type.type
@@ -377,6 +408,7 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                     selectionSet: selectionSet
                         ? getSelectionSetSelection(definitions, selectionSet, prop)
                         : undefined,
+                    input: definition.kind === graphql_1.Kind.INPUT_OBJECT_TYPE_DEFINITION,
                 });
             };
             const keys = [
@@ -470,7 +502,7 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                     mock.variables[name] = mockPrev.variables[name];
                     continue;
                 }
-                const result = resolveValue(definitions, scalars, variableDefinition.type, { hostType: `${path}Variables`, prop: name });
+                const result = resolveValue(definitions, scalars, variableDefinition.type, { hostType: `${path}Variables`, prop: name, input: false });
                 if (result != null) {
                     if (!mock.variables)
                         mock.variables = {};
@@ -493,6 +525,10 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
                         // Query, Mutation, Subscription
                         const operationKey = definition.operation[0].toUpperCase() +
                             definition.operation.slice(1);
+                        if (value === exports.clear) {
+                            mock.data[prop] = _proxy(definitions, scalars, `${operationKey}.${selection.name.value}`, [], { selectionSet: selection.selectionSet });
+                            continue;
+                        }
                         mock.data[prop] = _proxy(definitions, scalars, `${operationKey}.${selection.name.value}`, value != null ? [value] : [], {
                             prev: mockPrev?.data?.[prop],
                             selectionSet: selection.selectionSet,
@@ -538,6 +574,14 @@ const _proxy = (definitions, scalars, path, patches, { prev, resolvedType = reso
             const value = typeof scalar === "function" ? scalar(parent) : scalar;
             return value;
         }
+        case graphql_1.Kind.ENUM_TYPE_DEFINITION: {
+            const patch = patches[patches.length - 1];
+            if (patch != null)
+                return patch;
+            if (prev != null)
+                return prev;
+            return definition.values?.[0]?.name.value;
+        }
         default:
             throw new Error(`Unhandled definition kind '${definition.kind}'`);
     }

package/types/index.d.ts

@@ -1,5 +1,5 @@
 export { init } from "./init.js";
-export type { DeepPartial, Patch } from "./types.js";
+export type { DeepPartial, OperationMock, Patch } from "./types.js";
 export { codegen } from "./codegen.js";
-export { proxy } from "./proxy.js";
+export { clear, proxy } from "./proxy.js";
 export { formatCode, toObject } from "./util.js";

package/types/init.d.ts

@@ -1,5 +1,15 @@
 import type { Build, MapObjectsToTransforms, MapOperationsToTransforms } from "./extendedTypes.js";
 import type { ContravariantEmpty, OperationMock } from "./types.js";
+/**
+ * Initialize the data builder.
+ * @param schema The plain text of your schema.
+ * @param queries List of queries exported from generated types.
+ * @param mutations List of mutations exported from generated types.
+ * @param subscriptions List of subscriptions exported from generated types.
+ * @param types List of types exported from generated types.
+ * @param inputs List of types exported from generated types.
+ * @param scalars A mapping to generate scalar values. Function values will be invoked with their `__typename`.
+ */
 export declare const init: <Query extends Record<string, {
     data: Record<string, unknown>;
     variables?: Record<string, unknown>;

package/types/jest.d.ts

@@ -1,7 +1,33 @@
 import React from "react";
+import { MockedProviderProps } from "@apollo/client/testing";
 import type { OperationMock } from "./types.js";
-export declare const MockProvider: ({ mocks, stack, children }: {
-    mocks: OperationMock[];
+/**
+ * In older versions of `@apollo/client`, refetches with missing mocks trigger
+ * warnings instead of being treated as standard missing mocks.
+ * This utility converts those warnings into failures and ensures watch queries
+ * are installed for queries with `watch: true`. This must be set when
+ * `MockProvider` is mounted.
+ *
+ * This is not required on modern version of `@apollo/client`.
+ */
+export declare const failRefetchWarnings: (value?: boolean) => boolean;
+/**
+ * Allows missing mocks, resulting in tests passing. Usage is intended to ease
+ * migration.
+ */
+export declare const allowMissingMocks: (value: true) => true;
+/**
+ * Wait for mocks to have been used.
+ * @param mock If `undefined`, waits for all mocks. If a number, waits fort he first `mocks` mocks. If a string, waits for all mocks up until and including that mock.
+ * @param offset If `mocks` is a string, grabs the `offset`th mock of that name (e.g., the third `getReport` mock)
+ */
+export declare const waitForMocks: (mock?: number | string, offset?: number) => Promise<void>;
+/**
+ * A wrapper for `@apollo/client/testing`, this component will assert all
+ * requests have matching mocks and all defined mocks are used unless marked
+ * `optional`.
+ */
+export declare const MockedProvider: ({ mocks, stack: renderStack, children, link: passedLink, ...rest }: Omit<MockedProviderProps, "mocks"> & {
+    mocks: ReadonlyArray<OperationMock>;
     stack?: string;
-    children?: React.ReactNode;
 }) => React.JSX.Element;

package/types/proxy.d.ts

@@ -1,5 +1,10 @@
 import type { DefinitionNode, DocumentNode, GraphQLError } from "graphql";
 import type { OperationMock, Patch, SimpleOperationMock } from "./types.js";
+/**
+ * Updates the value to the base value, bypassing default transformations.
+ * Useful to clear optional inputs, resulting in `undefined` rather than `null`.
+ */
+export declare const clear: undefined;
 export declare const withGetDefaultPatch: <T>(newGetDefaultPatch: <U>(__typename: string) => Patch<U> | ((prev: U) => Patch<U> | undefined) | undefined, fn: () => T) => T;
 export declare const proxy: <T>(definitions: readonly DefinitionNode[], scalars: Record<string, unknown | ((typename: string) => unknown)>, type: string, ...patches: (Patch<T> | ((prev: T) => Patch<T>))[]) => T;
 export declare const operation: <O extends SimpleOperationMock, Extra = object>(definitions: readonly DefinitionNode[], scalars: Record<string, unknown | ((typename: string) => unknown)>, query: string | DocumentNode, ...patches: (Patch<Omit<O, "error" | "errors">> & {

package/types/types.d.ts

@@ -28,12 +28,16 @@ export type OperationMock<Data extends Record<string, unknown> = Record<string,
     };
     error?: Error;
     stack?: string;
+    watch?: boolean;
+    optional?: boolean;
 };
 export type SimpleOperationMock<Data extends Record<string, unknown> = Record<string, unknown>, Variables = Record<string, unknown> | never> = {
     data: Data;
     variables?: Variables;
     error?: Error;
     errors?: GraphQLError[];
+    watch?: boolean;
+    optional?: boolean;
 };
 export type SimpleOperationPatch<Data extends Record<string, unknown> = Record<string, unknown>, Variables = Record<string, unknown> | never> = Patch<{
     data: Data;
@@ -41,5 +45,7 @@ export type SimpleOperationPatch<Data extends Record<string, unknown> = Record<s
 }> & {
     error?: Error;
     errors?: GraphQLError[];
+    watch?: boolean;
+    optional?: boolean;
 };
 export {};