Function wrap

packages/core/sdk/src/plugin-storage.test.ts:125–161

(source: FumaDb)

// disarmed test.
const failPluginStorageBulkWriteAfterFirstRow = (db: FumaDb): FumaDb => {
  const wrap = (source: FumaDb): FumaDb =>
    new Proxy(source, {
      get(target, property, receiver) {
        if (property === "withContext") {
          const withContext = target.withContext;
          return withContext === undefined
            ? undefined
            : (context: unknown) => wrap(withContext(context));
        }
        if (property === "transaction") {
          const transaction: FumaDb["transaction"] = (run) =>
            target.transaction((transactionDb) => run(wrap(transactionDb)));
          return transaction;
        }
        if (property === "upsertMany") {
          const upsertMany: FumaDb["upsertMany"] = async (table, options) => {
            if (table !== "plugin_storage" || options.values.length < 2) {
              return target.upsertMany(table, options);
            }

            await target.upsertMany(table, {
              ...options,
              values: options.values.slice(0, 1),
            });
            // oxlint-disable-next-line executor/no-promise-reject -- boundary: fault-injecting FumaDB adapter must reject to exercise transaction rollback
            return Promise.reject(
              new StorageError({
                message: "Injected plugin storage bulk-write failure.",
                cause: undefined,
              }),
            );
          };
          return upsertMany;
        }
        return Reflect.get(target, property, receiver);
      },
    });

  return wrap(db);
};