orchid-orm-test-factory 0.0.1

package/README.md

@@ -0,0 +1,5 @@
+# Orchid ORM test factory
+
+This is a tool for building and creating mock objects basing on [Orchid ORM](https://github.com/romeerez/orchid-orm/tree/main/packages/orchid-orm) models.
+
+[Read docs here](https://orchid-orm.netlify.app/guide/orm-test-factories).

package/dist/index.d.ts

@@ -0,0 +1,73 @@
+import { AnyZodObject } from 'zod';
+import { Query, EmptyObject, CreateData } from 'pqb';
+import { InstanceToZod } from 'orchid-orm-schema-to-zod';
+
+declare type FactoryOptions = {
+    sequence?: number;
+    sequenceDistance?: number;
+    maxTextLength?: number;
+};
+declare type metaKey = typeof metaKey;
+declare const metaKey: unique symbol;
+declare type Result<T extends TestFactory, Data, Omitted = Omit<Data, keyof T[metaKey]['omit']>> = EmptyObject extends T[metaKey]['pick'] ? Omitted : Pick<Omitted, {
+    [K in keyof Omitted]: K extends keyof T[metaKey]['pick'] ? K : never;
+}[keyof Omitted]>;
+declare type BuildArg<T extends TestFactory> = {
+    [K in keyof T[metaKey]['type']]?: T[metaKey]['type'][K] | ((sequence: number) => T[metaKey]['type'][K]);
+} & Record<string, unknown>;
+declare type BuildResult<T extends TestFactory, Data extends BuildArg<T>> = Result<T, BuildArg<T> extends Data ? T[metaKey]['type'] : T[metaKey]['type'] & {
+    [K in keyof Data]: Data[K] extends () => void ? ReturnType<Data[K]> : Data[K];
+}>;
+declare type CreateArg<T extends TestFactory> = CreateData<Omit<T['model'], 'inputType'> & {
+    inputType: {
+        [K in keyof T['model']['type']]?: T['model']['type'][K] | ((sequence: number) => T['model']['type'][K]);
+    };
+}>;
+declare type CreateResult<T extends TestFactory> = Result<T, T['model']['type']>;
+declare class TestFactory<Q extends Query = Query, Schema extends AnyZodObject = AnyZodObject, Type extends EmptyObject = EmptyObject> {
+    model: Q;
+    schema: Schema;
+    private readonly data;
+    sequence: number;
+    private readonly omitValues;
+    private readonly pickValues;
+    [metaKey]: {
+        type: Type;
+        omit: EmptyObject;
+        pick: EmptyObject;
+    };
+    constructor(model: Q, schema: Schema, data?: Record<string, unknown>, options?: FactoryOptions);
+    set<T extends this, Meta extends {
+        type: EmptyObject;
+    }, Data extends {
+        [K in keyof Meta['type']]?: Meta['type'][K] | (() => Meta['type'][K]);
+    } & Record<string, unknown>>(this: T & {
+        [metaKey]: Meta;
+    }, data: Data): T & {
+        [metaKey]: Meta & {
+            type: Data;
+        };
+    };
+    omit<T extends this, Keys extends {
+        [K in keyof T[metaKey]['type']]?: true;
+    }>(this: T, keys: Keys): T & {
+        [metaKey]: T[metaKey] & {
+            omit: Keys;
+        };
+    };
+    pick<T extends this, Keys extends {
+        [K in keyof T[metaKey]['type']]?: true;
+    }>(this: T, keys: Keys): T & {
+        [metaKey]: T[metaKey] & {
+            pick: Keys;
+        };
+    };
+    build<T extends this, Data extends BuildArg<T>>(this: T, data?: Data): BuildResult<T, Data>;
+    buildList<T extends this, Data extends BuildArg<T>>(this: T, qty: number, data?: Data): BuildResult<T, Data>[];
+    create<T extends this, Data extends CreateArg<T>>(this: T, data?: Data): Promise<CreateResult<T>>;
+    createList<T extends this, Data extends CreateArg<T>>(this: T, qty: number, data?: Data): Promise<CreateResult<T>[]>;
+    extend<T extends this>(this: T): new () => TestFactory<Q, Schema, Type>;
+}
+declare const createFactory: <T extends Query>(model: T, options?: FactoryOptions) => TestFactory<T, InstanceToZod<T>, T["type"]>;
+
+export { TestFactory, createFactory };

package/dist/index.esm.js

@@ -0,0 +1,185 @@
+import { ZodNullable } from 'zod';
+import { DateBaseColumn, IntegerBaseColumn, TextBaseColumn } from 'pqb';
+import { instanceToZod } from 'orchid-orm-schema-to-zod';
+import { generateMock } from '@anatine/zod-mock';
+
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+const omit = (obj, keys) => {
+  const res = __spreadValues({}, obj);
+  Object.keys(keys).forEach((key) => {
+    delete res[key];
+  });
+  return res;
+};
+const pick = (obj, keys) => {
+  const res = {};
+  Object.keys(keys).forEach((key) => {
+    const value = obj[key];
+    if (value !== void 0) {
+      res[key] = value;
+    }
+  });
+  return res;
+};
+const processCreateData = (factory, data, arg) => {
+  const pick2 = {};
+  for (const key in factory.model.shape) {
+    pick2[key] = true;
+  }
+  factory.model.primaryKeys.forEach((key) => {
+    if (factory.model.shape[key].dataType.includes("serial")) {
+      delete pick2[key];
+    }
+  });
+  const result = {};
+  const fns = {};
+  const allData = arg ? __spreadValues(__spreadValues({}, data), arg) : data;
+  for (const key in allData) {
+    delete pick2[key];
+    const value = allData[key];
+    if (typeof value === "function") {
+      fns[key] = value;
+    } else {
+      result[key] = value;
+    }
+  }
+  const pickedSchema = factory.schema.pick(pick2);
+  return () => {
+    Object.assign(result, generateMock(pickedSchema));
+    for (const key in fns) {
+      result[key] = fns[key](factory.sequence);
+    }
+    factory.sequence++;
+    return __spreadValues({}, result);
+  };
+};
+const _TestFactory = class {
+  constructor(model, schema, data = {}, options = {}) {
+    this.model = model;
+    this.schema = schema;
+    this.data = data;
+    this.omitValues = {};
+    this.pickValues = {};
+    var _a;
+    if (options.sequence !== void 0) {
+      this.sequence = options.sequence;
+    } else {
+      let workerId = parseInt(process.env.JEST_WORKER_ID);
+      if (isNaN(workerId))
+        workerId = 1;
+      this.sequence = (workerId - 1) * ((_a = options.sequenceDistance) != null ? _a : 1e3) + 1;
+    }
+  }
+  set(data) {
+    return Object.assign(Object.create(this), {
+      data: __spreadValues(__spreadValues({}, this.data), data)
+    });
+  }
+  omit(keys) {
+    return Object.assign(Object.create(this), {
+      omitValues: __spreadValues(__spreadValues({}, this.omitValues), keys)
+    });
+  }
+  pick(keys) {
+    return Object.assign(Object.create(this), {
+      pickValues: __spreadValues(__spreadValues({}, this.pickValues), keys)
+    });
+  }
+  build(data) {
+    let schema = this.schema;
+    let arg = data ? __spreadValues(__spreadValues({}, this.data), data) : this.data;
+    if (this.omitValues) {
+      schema = schema.omit(this.omitValues);
+      arg = omit(arg, this.omitValues);
+    }
+    if (this.pickValues && Object.keys(this.pickValues).length) {
+      schema = schema.pick(this.pickValues);
+      arg = pick(arg, this.pickValues);
+    }
+    const result = generateMock(schema);
+    for (const key in arg) {
+      const value = arg[key];
+      if (typeof value === "function") {
+        result[key] = value(this.sequence);
+      } else {
+        result[key] = value;
+      }
+    }
+    this.sequence++;
+    return result;
+  }
+  buildList(qty, data) {
+    return [...Array(qty)].map(() => this.build(data));
+  }
+  async create(data) {
+    const getData = processCreateData(this, this.data, data);
+    return await this.model.create(getData());
+  }
+  async createList(qty, data) {
+    const getData = processCreateData(this, this.data, data);
+    const arr = [...Array(qty)].map(getData);
+    return await this.model.createMany(arr);
+  }
+  extend() {
+    const { model, schema } = this;
+    return class extends _TestFactory {
+      constructor() {
+        super(model, schema);
+      }
+    };
+  }
+};
+let TestFactory = _TestFactory;
+const nowString = new Date().toISOString();
+const createFactory = (model, options) => {
+  var _a, _b;
+  const schema = instanceToZod(model);
+  const data = {};
+  const now = Date.now();
+  for (const key in model.shape) {
+    const column = model.shape[key];
+    if (column instanceof DateBaseColumn) {
+      if (column.data.as instanceof IntegerBaseColumn) {
+        data[key] = (sequence) => now + sequence;
+      } else if (((_a = column.parseFn) == null ? void 0 : _a.call(column, nowString)) instanceof Date) {
+        data[key] = (sequence) => new Date(now + sequence);
+      } else {
+        data[key] = (sequence) => new Date(now + sequence).toISOString();
+      }
+    } else if (column instanceof TextBaseColumn) {
+      const max = (_b = options == null ? void 0 : options.maxTextLength) != null ? _b : 1e3;
+      const item = schema.shape[key];
+      const string = item instanceof ZodNullable ? item.unwrap() : item;
+      const maxCheck = string._def.checks.find(
+        (check) => check.kind === "max"
+      );
+      if (!maxCheck || maxCheck.value[0] > max) {
+        schema.shape[key] = item instanceof ZodNullable ? string.max(max).nullable() : string.max(max);
+      }
+    }
+  }
+  return new TestFactory(
+    model,
+    schema,
+    data,
+    options
+  );
+};
+
+export { TestFactory, createFactory };
+//# sourceMappingURL=index.esm.js.map

package/dist/index.esm.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.esm.js","sources":["../src/factory.ts"],"sourcesContent":["import { AnyZodObject, ZodNullable, ZodString, ZodTypeAny } from 'zod';\nimport {\n  CreateData,\n  DateBaseColumn,\n  EmptyObject,\n  IntegerBaseColumn,\n  Query,\n  TextBaseColumn,\n} from 'pqb';\nimport { instanceToZod, InstanceToZod } from 'orchid-orm-schema-to-zod';\nimport { generateMock } from '@anatine/zod-mock';\n\ntype FactoryOptions = {\n  sequence?: number;\n  sequenceDistance?: number;\n  maxTextLength?: number;\n};\n\ntype metaKey = typeof metaKey;\nconst metaKey = Symbol('meta');\n\ntype Result<\n  T extends TestFactory,\n  Data,\n  Omitted = Omit<Data, keyof T[metaKey]['omit']>,\n> = EmptyObject extends T[metaKey]['pick']\n  ? Omitted\n  : Pick<\n      Omitted,\n      {\n        [K in keyof Omitted]: K extends keyof T[metaKey]['pick'] ? K : never;\n      }[keyof Omitted]\n    >;\n\ntype BuildArg<T extends TestFactory> = {\n  [K in keyof T[metaKey]['type']]?:\n    | T[metaKey]['type'][K]\n    | ((sequence: number) => T[metaKey]['type'][K]);\n} & Record<string, unknown>;\n\ntype BuildResult<T extends TestFactory, Data extends BuildArg<T>> = Result<\n  T,\n  BuildArg<T> extends Data\n    ? T[metaKey]['type']\n    : T[metaKey]['type'] & {\n        [K in keyof Data]: Data[K] extends () => void\n          ? ReturnType<Data[K]>\n          : Data[K];\n      }\n>;\n\ntype CreateArg<T extends TestFactory> = CreateData<\n  Omit<T['model'], 'inputType'> & {\n    inputType: {\n      [K in keyof T['model']['type']]?:\n        | T['model']['type'][K]\n        | ((sequence: number) => T['model']['type'][K]);\n    };\n  }\n>;\n\ntype CreateResult<T extends TestFactory> = Result<T, T['model']['type']>;\n\nconst omit = <T, Keys extends Record<string, unknown>>(\n  obj: T,\n  keys: Keys,\n): Omit<T, keyof Keys> => {\n  const res = { ...obj };\n  Object.keys(keys).forEach((key) => {\n    delete (res as unknown as Record<string, unknown>)[key];\n  });\n  return res;\n};\n\nconst pick = <T, Keys extends Record<string, unknown>>(\n  obj: T,\n  keys: Keys,\n): Pick<T, { [K in keyof T]: K extends keyof Keys ? K : never }[keyof T]> => {\n  const res = {} as T;\n  Object.keys(keys).forEach((key) => {\n    const value = (obj as unknown as Record<string, unknown>)[key];\n    if (value !== undefined) {\n      (res as unknown as Record<string, unknown>)[key] = value;\n    }\n  });\n  return res;\n};\n\nconst processCreateData = <T extends TestFactory, Data extends CreateArg<T>>(\n  factory: T,\n  data: Record<string, unknown>,\n  arg?: Data,\n) => {\n  const pick: Record<string, true> = {};\n  for (const key in factory.model.shape) {\n    pick[key] = true;\n  }\n\n  factory.model.primaryKeys.forEach((key) => {\n    if (factory.model.shape[key].dataType.includes('serial')) {\n      delete pick[key];\n    }\n  });\n\n  const result: Record<string, unknown> = {};\n\n  const fns: Record<string, (sequence: number) => unknown> = {};\n\n  const allData = (arg ? { ...data, ...arg } : data) as Record<string, unknown>;\n\n  for (const key in allData) {\n    delete pick[key];\n    const value = allData[key];\n    if (typeof value === 'function') {\n      fns[key] = value as () => unknown;\n    } else {\n      result[key] = value;\n    }\n  }\n\n  const pickedSchema = factory.schema.pick(pick);\n\n  return () => {\n    Object.assign(result, generateMock(pickedSchema));\n\n    for (const key in fns) {\n      result[key] = fns[key](factory.sequence);\n    }\n\n    factory.sequence++;\n\n    return { ...result } as CreateData<T['model']>;\n  };\n};\n\nexport class TestFactory<\n  Q extends Query = Query,\n  Schema extends AnyZodObject = AnyZodObject,\n  Type extends EmptyObject = EmptyObject,\n> {\n  sequence: number;\n  private readonly omitValues: Record<PropertyKey, true> = {};\n  private readonly pickValues: Record<PropertyKey, true> = {};\n\n  [metaKey]!: {\n    type: Type;\n    omit: EmptyObject;\n    pick: EmptyObject;\n  };\n\n  constructor(\n    public model: Q,\n    public schema: Schema,\n    private readonly data: Record<string, unknown> = {},\n    options: FactoryOptions = {},\n  ) {\n    if (options.sequence !== undefined) {\n      this.sequence = options.sequence;\n    } else {\n      let workerId = parseInt(process.env.JEST_WORKER_ID as string);\n      if (isNaN(workerId)) workerId = 1;\n      this.sequence = (workerId - 1) * (options.sequenceDistance ?? 1000) + 1;\n    }\n  }\n\n  set<\n    T extends this,\n    Meta extends { type: EmptyObject },\n    Data extends {\n      [K in keyof Meta['type']]?: Meta['type'][K] | (() => Meta['type'][K]);\n    } & Record<string, unknown>,\n  >(\n    this: T & { [metaKey]: Meta },\n    data: Data,\n  ): T & { [metaKey]: Meta & { type: Data } } {\n    return Object.assign(Object.create(this), {\n      data: { ...this.data, ...data },\n    });\n  }\n\n  omit<T extends this, Keys extends { [K in keyof T[metaKey]['type']]?: true }>(\n    this: T,\n    keys: Keys,\n  ): T & { [metaKey]: T[metaKey] & { omit: Keys } } {\n    return Object.assign(Object.create(this), {\n      omitValues: { ...this.omitValues, ...keys },\n    });\n  }\n\n  pick<T extends this, Keys extends { [K in keyof T[metaKey]['type']]?: true }>(\n    this: T,\n    keys: Keys,\n  ): T & { [metaKey]: T[metaKey] & { pick: Keys } } {\n    return Object.assign(Object.create(this), {\n      pickValues: { ...this.pickValues, ...keys },\n    });\n  }\n\n  build<T extends this, Data extends BuildArg<T>>(\n    this: T,\n    data?: Data,\n  ): BuildResult<T, Data> {\n    let schema = this.schema as AnyZodObject;\n    let arg = data ? { ...this.data, ...data } : this.data;\n\n    if (this.omitValues) {\n      schema = schema.omit(this.omitValues);\n      arg = omit(arg, this.omitValues);\n    }\n\n    if (this.pickValues && Object.keys(this.pickValues).length) {\n      schema = schema.pick(this.pickValues);\n      arg = pick(arg, this.pickValues);\n    }\n\n    const result = generateMock(schema) as Record<string, unknown>;\n    for (const key in arg) {\n      const value = (arg as Record<string, unknown>)[key];\n      if (typeof value === 'function') {\n        result[key] = value(this.sequence);\n      } else {\n        result[key] = value;\n      }\n    }\n\n    this.sequence++;\n\n    return result as BuildResult<T, Data>;\n  }\n\n  buildList<T extends this, Data extends BuildArg<T>>(\n    this: T,\n    qty: number,\n    data?: Data,\n  ): BuildResult<T, Data>[] {\n    return [...Array(qty)].map(() => this.build(data));\n  }\n\n  async create<T extends this, Data extends CreateArg<T>>(\n    this: T,\n    data?: Data,\n  ): Promise<CreateResult<T>> {\n    const getData = processCreateData(this, this.data, data);\n    return (await this.model.create(getData())) as CreateResult<T>;\n  }\n\n  async createList<T extends this, Data extends CreateArg<T>>(\n    this: T,\n    qty: number,\n    data?: Data,\n  ): Promise<CreateResult<T>[]> {\n    const getData = processCreateData(this, this.data, data);\n    const arr = [...Array(qty)].map(getData);\n    return (await this.model.createMany(arr)) as CreateResult<T>[];\n  }\n\n  extend<T extends this>(this: T): new () => TestFactory<Q, Schema, Type> {\n    const { model, schema } = this;\n\n    return class extends TestFactory<Q, Schema, Type> {\n      constructor() {\n        super(model, schema);\n      }\n    };\n  }\n}\n\nconst nowString = new Date().toISOString();\n\nexport const createFactory = <T extends Query>(\n  model: T,\n  options?: FactoryOptions,\n) => {\n  const schema = instanceToZod(model);\n\n  const data: Record<string, unknown> = {};\n  const now = Date.now();\n\n  for (const key in model.shape) {\n    const column = model.shape[key];\n    if (column instanceof DateBaseColumn) {\n      if (column.data.as instanceof IntegerBaseColumn) {\n        data[key] = (sequence: number) => now + sequence;\n      } else if (column.parseFn?.(nowString) instanceof Date) {\n        data[key] = (sequence: number) => new Date(now + sequence);\n      } else {\n        data[key] = (sequence: number) =>\n          new Date(now + sequence).toISOString();\n      }\n    } else if (column instanceof TextBaseColumn) {\n      const max = options?.maxTextLength ?? 1000;\n      const item = schema.shape[key];\n      const string = (\n        item instanceof ZodNullable ? item.unwrap() : item\n      ) as ZodString;\n\n      const maxCheck = string._def.checks.find(\n        (check) => check.kind === 'max',\n      ) as { value: [number] } | undefined;\n\n      if (!maxCheck || maxCheck.value[0] > max) {\n        (schema.shape as Record<string, ZodTypeAny>)[key] =\n          item instanceof ZodNullable\n            ? string.max(max).nullable()\n            : string.max(max);\n      }\n    }\n  }\n\n  return new TestFactory<T, InstanceToZod<T>, T['type']>(\n    model,\n    schema,\n    data,\n    options,\n  );\n};\n"],"names":["pick"],"mappings":";;;;;;;;;;;;;;;;;;;;;AA+DA,MAAM,IAAA,GAAO,CACX,GAAA,EACA,IACwB,KAAA;AACxB,EAAA,MAAM,MAAM,cAAK,CAAA,EAAA,EAAA,GAAA,CAAA,CAAA;AACjB,EAAA,MAAA,CAAO,IAAK,CAAA,IAAI,CAAE,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACjC,IAAA,OAAQ,GAA2C,CAAA,GAAA,CAAA,CAAA;AAAA,GACpD,CAAA,CAAA;AACD,EAAO,OAAA,GAAA,CAAA;AACT,CAAA,CAAA;AAEA,MAAM,IAAA,GAAO,CACX,GAAA,EACA,IAC2E,KAAA;AAC3E,EAAA,MAAM,MAAM,EAAC,CAAA;AACb,EAAA,MAAA,CAAO,IAAK,CAAA,IAAI,CAAE,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACjC,IAAA,MAAM,QAAS,GAA2C,CAAA,GAAA,CAAA,CAAA;AAC1D,IAAA,IAAI,UAAU,KAAW,CAAA,EAAA;AACvB,MAAC,IAA2C,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,KACrD;AAAA,GACD,CAAA,CAAA;AACD,EAAO,OAAA,GAAA,CAAA;AACT,CAAA,CAAA;AAEA,MAAM,iBAAoB,GAAA,CACxB,OACA,EAAA,IAAA,EACA,GACG,KAAA;AACH,EAAA,MAAMA,QAA6B,EAAC,CAAA;AACpC,EAAW,KAAA,MAAA,GAAA,IAAO,OAAQ,CAAA,KAAA,CAAM,KAAO,EAAA;AACrC,IAAAA,MAAK,GAAO,CAAA,GAAA,IAAA,CAAA;AAAA,GACd;AAEA,EAAA,OAAA,CAAQ,KAAM,CAAA,WAAA,CAAY,OAAQ,CAAA,CAAC,GAAQ,KAAA;AACzC,IAAA,IAAI,QAAQ,KAAM,CAAA,KAAA,CAAM,KAAK,QAAS,CAAA,QAAA,CAAS,QAAQ,CAAG,EAAA;AACxD,MAAA,OAAOA,KAAK,CAAA,GAAA,CAAA,CAAA;AAAA,KACd;AAAA,GACD,CAAA,CAAA;AAED,EAAA,MAAM,SAAkC,EAAC,CAAA;AAEzC,EAAA,MAAM,MAAqD,EAAC,CAAA;AAE5D,EAAA,MAAM,OAAW,GAAA,GAAA,GAAM,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAA,EAAS,GAAQ,CAAA,GAAA,IAAA,CAAA;AAE7C,EAAA,KAAA,MAAW,OAAO,OAAS,EAAA;AACzB,IAAA,OAAOA,KAAK,CAAA,GAAA,CAAA,CAAA;AACZ,IAAA,MAAM,QAAQ,OAAQ,CAAA,GAAA,CAAA,CAAA;AACtB,IAAI,IAAA,OAAO,UAAU,UAAY,EAAA;AAC/B,MAAA,GAAA,CAAI,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,KACN,MAAA;AACL,MAAA,MAAA,CAAO,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,KAChB;AAAA,GACF;AAEA,EAAA,MAAM,YAAe,GAAA,OAAA,CAAQ,MAAO,CAAA,IAAA,CAAKA,KAAI,CAAA,CAAA;AAE7C,EAAA,OAAO,MAAM;AACX,IAAA,MAAA,CAAO,MAAO,CAAA,MAAA,EAAQ,YAAa,CAAA,YAAY,CAAC,CAAA,CAAA;AAEhD,IAAA,KAAA,MAAW,OAAO,GAAK,EAAA;AACrB,MAAA,MAAA,CAAO,GAAO,CAAA,GAAA,GAAA,CAAI,GAAK,CAAA,CAAA,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,KACzC;AAEA,IAAQ,OAAA,CAAA,QAAA,EAAA,CAAA;AAER,IAAA,OAAO,cAAK,CAAA,EAAA,EAAA,MAAA,CAAA,CAAA;AAAA,GACd,CAAA;AACF,CAAA,CAAA;AAEO,MAAM,eAAN,MAIL;AAAA,EAWA,WAAA,CACS,OACA,MACU,EAAA,IAAA,GAAgC,EACjD,EAAA,OAAA,GAA0B,EAC1B,EAAA;AAJO,IAAA,IAAA,CAAA,KAAA,GAAA,KAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACU,IAAA,IAAA,CAAA,IAAA,GAAA,IAAA,CAAA;AAZnB,IAAA,IAAA,CAAiB,aAAwC,EAAC,CAAA;AAC1D,IAAA,IAAA,CAAiB,aAAwC,EAAC,CAAA;AA9I5D,IAAA,IAAA,EAAA,CAAA;AA4JI,IAAI,IAAA,OAAA,CAAQ,aAAa,KAAW,CAAA,EAAA;AAClC,MAAA,IAAA,CAAK,WAAW,OAAQ,CAAA,QAAA,CAAA;AAAA,KACnB,MAAA;AACL,MAAA,IAAI,QAAW,GAAA,QAAA,CAAS,OAAQ,CAAA,GAAA,CAAI,cAAwB,CAAA,CAAA;AAC5D,MAAA,IAAI,MAAM,QAAQ,CAAA;AAAG,QAAW,QAAA,GAAA,CAAA,CAAA;AAChC,MAAA,IAAA,CAAK,YAAY,QAAW,GAAA,CAAA,KAAA,CAAM,EAAQ,GAAA,OAAA,CAAA,gBAAA,KAAR,YAA4B,GAAQ,CAAA,GAAA,CAAA,CAAA;AAAA,KACxE;AAAA,GACF;AAAA,EAEA,IAQE,IAC0C,EAAA;AAC1C,IAAA,OAAO,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AAAA,MACxC,IAAA,EAAM,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAK,IAAS,CAAA,EAAA,IAAA,CAAA;AAAA,KAC1B,CAAA,CAAA;AAAA,GACH;AAAA,EAEA,KAEE,IACgD,EAAA;AAChD,IAAA,OAAO,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AAAA,MACxC,UAAA,EAAY,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAK,UAAe,CAAA,EAAA,IAAA,CAAA;AAAA,KACtC,CAAA,CAAA;AAAA,GACH;AAAA,EAEA,KAEE,IACgD,EAAA;AAChD,IAAA,OAAO,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AAAA,MACxC,UAAA,EAAY,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAK,UAAe,CAAA,EAAA,IAAA,CAAA;AAAA,KACtC,CAAA,CAAA;AAAA,GACH;AAAA,EAEA,MAEE,IACsB,EAAA;AACtB,IAAA,IAAI,SAAS,IAAK,CAAA,MAAA,CAAA;AAClB,IAAA,IAAI,MAAM,IAAO,GAAA,cAAA,CAAA,cAAA,CAAA,EAAA,EAAK,IAAK,CAAA,IAAA,CAAA,EAAS,QAAS,IAAK,CAAA,IAAA,CAAA;AAElD,IAAA,IAAI,KAAK,UAAY,EAAA;AACnB,MAAS,MAAA,GAAA,MAAA,CAAO,IAAK,CAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AACpC,MAAM,GAAA,GAAA,IAAA,CAAK,GAAK,EAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AAAA,KACjC;AAEA,IAAA,IAAI,KAAK,UAAc,IAAA,MAAA,CAAO,KAAK,IAAK,CAAA,UAAU,EAAE,MAAQ,EAAA;AAC1D,MAAS,MAAA,GAAA,MAAA,CAAO,IAAK,CAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AACpC,MAAM,GAAA,GAAA,IAAA,CAAK,GAAK,EAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AAAA,KACjC;AAEA,IAAM,MAAA,MAAA,GAAS,aAAa,MAAM,CAAA,CAAA;AAClC,IAAA,KAAA,MAAW,OAAO,GAAK,EAAA;AACrB,MAAA,MAAM,QAAS,GAAgC,CAAA,GAAA,CAAA,CAAA;AAC/C,MAAI,IAAA,OAAO,UAAU,UAAY,EAAA;AAC/B,QAAO,MAAA,CAAA,GAAA,CAAA,GAAO,KAAM,CAAA,IAAA,CAAK,QAAQ,CAAA,CAAA;AAAA,OAC5B,MAAA;AACL,QAAA,MAAA,CAAO,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,OAChB;AAAA,KACF;AAEA,IAAK,IAAA,CAAA,QAAA,EAAA,CAAA;AAEL,IAAO,OAAA,MAAA,CAAA;AAAA,GACT;AAAA,EAEA,SAAA,CAEE,KACA,IACwB,EAAA;AACxB,IAAO,OAAA,CAAC,GAAG,KAAA,CAAM,GAAG,CAAC,CAAE,CAAA,GAAA,CAAI,MAAM,IAAA,CAAK,KAAM,CAAA,IAAI,CAAC,CAAA,CAAA;AAAA,GACnD;AAAA,EAEA,MAAM,OAEJ,IAC0B,EAAA;AAC1B,IAAA,MAAM,OAAU,GAAA,iBAAA,CAAkB,IAAM,EAAA,IAAA,CAAK,MAAM,IAAI,CAAA,CAAA;AACvD,IAAA,OAAQ,MAAM,IAAA,CAAK,KAAM,CAAA,MAAA,CAAO,SAAS,CAAA,CAAA;AAAA,GAC3C;AAAA,EAEA,MAAM,UAEJ,CAAA,GAAA,EACA,IAC4B,EAAA;AAC5B,IAAA,MAAM,OAAU,GAAA,iBAAA,CAAkB,IAAM,EAAA,IAAA,CAAK,MAAM,IAAI,CAAA,CAAA;AACvD,IAAM,MAAA,GAAA,GAAM,CAAC,GAAG,KAAA,CAAM,GAAG,CAAC,CAAA,CAAE,IAAI,OAAO,CAAA,CAAA;AACvC,IAAA,OAAQ,MAAM,IAAA,CAAK,KAAM,CAAA,UAAA,CAAW,GAAG,CAAA,CAAA;AAAA,GACzC;AAAA,EAEA,MAAwE,GAAA;AACtE,IAAM,MAAA,EAAE,KAAO,EAAA,MAAA,EAAW,GAAA,IAAA,CAAA;AAE1B,IAAA,OAAO,cAAc,YAA6B,CAAA;AAAA,MAChD,WAAc,GAAA;AACZ,QAAA,KAAA,CAAM,OAAO,MAAM,CAAA,CAAA;AAAA,OACrB;AAAA,KACF,CAAA;AAAA,GACF;AACF,CAAA,CAAA;AAlIO,IAAM,WAAN,GAAA,aAAA;AAoIP,MAAM,SAAY,GAAA,IAAI,IAAK,EAAA,CAAE,WAAY,EAAA,CAAA;AAE5B,MAAA,aAAA,GAAgB,CAC3B,KAAA,EACA,OACG,KAAA;AAhRL,EAAA,IAAA,EAAA,EAAA,EAAA,CAAA;AAiRE,EAAM,MAAA,MAAA,GAAS,cAAc,KAAK,CAAA,CAAA;AAElC,EAAA,MAAM,OAAgC,EAAC,CAAA;AACvC,EAAM,MAAA,GAAA,GAAM,KAAK,GAAI,EAAA,CAAA;AAErB,EAAW,KAAA,MAAA,GAAA,IAAO,MAAM,KAAO,EAAA;AAC7B,IAAM,MAAA,MAAA,GAAS,MAAM,KAAM,CAAA,GAAA,CAAA,CAAA;AAC3B,IAAA,IAAI,kBAAkB,cAAgB,EAAA;AACpC,MAAI,IAAA,MAAA,CAAO,IAAK,CAAA,EAAA,YAAc,iBAAmB,EAAA;AAC/C,QAAK,IAAA,CAAA,GAAA,CAAA,GAAO,CAAC,QAAA,KAAqB,GAAM,GAAA,QAAA,CAAA;AAAA,OAC/B,MAAA,IAAA,CAAA,CAAA,EAAA,GAAA,MAAA,CAAO,OAAP,KAAA,IAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,IAAA,CAAA,MAAA,EAAiB,uBAAsB,IAAM,EAAA;AACtD,QAAA,IAAA,CAAK,OAAO,CAAC,QAAA,KAAqB,IAAI,IAAA,CAAK,MAAM,QAAQ,CAAA,CAAA;AAAA,OACpD,MAAA;AACL,QAAK,IAAA,CAAA,GAAA,CAAA,GAAO,CAAC,QACX,KAAA,IAAI,KAAK,GAAM,GAAA,QAAQ,EAAE,WAAY,EAAA,CAAA;AAAA,OACzC;AAAA,KACF,MAAA,IAAW,kBAAkB,cAAgB,EAAA;AAC3C,MAAM,MAAA,GAAA,GAAA,CAAM,EAAS,GAAA,OAAA,IAAA,IAAA,GAAA,KAAA,CAAA,GAAA,OAAA,CAAA,aAAA,KAAT,IAA0B,GAAA,EAAA,GAAA,GAAA,CAAA;AACtC,MAAM,MAAA,IAAA,GAAO,OAAO,KAAM,CAAA,GAAA,CAAA,CAAA;AAC1B,MAAA,MAAM,MACJ,GAAA,IAAA,YAAgB,WAAc,GAAA,IAAA,CAAK,QAAW,GAAA,IAAA,CAAA;AAGhD,MAAM,MAAA,QAAA,GAAW,MAAO,CAAA,IAAA,CAAK,MAAO,CAAA,IAAA;AAAA,QAClC,CAAC,KAAU,KAAA,KAAA,CAAM,IAAS,KAAA,KAAA;AAAA,OAC5B,CAAA;AAEA,MAAA,IAAI,CAAC,QAAA,IAAY,QAAS,CAAA,KAAA,CAAM,KAAK,GAAK,EAAA;AACxC,QAAC,MAAO,CAAA,KAAA,CAAqC,GAC3C,CAAA,GAAA,IAAA,YAAgB,WACZ,GAAA,MAAA,CAAO,GAAI,CAAA,GAAG,CAAE,CAAA,QAAA,EAChB,GAAA,MAAA,CAAO,IAAI,GAAG,CAAA,CAAA;AAAA,OACtB;AAAA,KACF;AAAA,GACF;AAEA,EAAA,OAAO,IAAI,WAAA;AAAA,IACT,KAAA;AAAA,IACA,MAAA;AAAA,IACA,IAAA;AAAA,IACA,OAAA;AAAA,GACF,CAAA;AACF;;;;"}

package/dist/index.js

@@ -0,0 +1,190 @@
+'use strict';
+
+Object.defineProperty(exports, '__esModule', { value: true });
+
+var zod = require('zod');
+var pqb = require('pqb');
+var orchidOrmSchemaToZod = require('orchid-orm-schema-to-zod');
+var zodMock = require('@anatine/zod-mock');
+
+var __defProp = Object.defineProperty;
+var __getOwnPropSymbols = Object.getOwnPropertySymbols;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __propIsEnum = Object.prototype.propertyIsEnumerable;
+var __defNormalProp = (obj, key, value) => key in obj ? __defProp(obj, key, { enumerable: true, configurable: true, writable: true, value }) : obj[key] = value;
+var __spreadValues = (a, b) => {
+  for (var prop in b || (b = {}))
+    if (__hasOwnProp.call(b, prop))
+      __defNormalProp(a, prop, b[prop]);
+  if (__getOwnPropSymbols)
+    for (var prop of __getOwnPropSymbols(b)) {
+      if (__propIsEnum.call(b, prop))
+        __defNormalProp(a, prop, b[prop]);
+    }
+  return a;
+};
+const omit = (obj, keys) => {
+  const res = __spreadValues({}, obj);
+  Object.keys(keys).forEach((key) => {
+    delete res[key];
+  });
+  return res;
+};
+const pick = (obj, keys) => {
+  const res = {};
+  Object.keys(keys).forEach((key) => {
+    const value = obj[key];
+    if (value !== void 0) {
+      res[key] = value;
+    }
+  });
+  return res;
+};
+const processCreateData = (factory, data, arg) => {
+  const pick2 = {};
+  for (const key in factory.model.shape) {
+    pick2[key] = true;
+  }
+  factory.model.primaryKeys.forEach((key) => {
+    if (factory.model.shape[key].dataType.includes("serial")) {
+      delete pick2[key];
+    }
+  });
+  const result = {};
+  const fns = {};
+  const allData = arg ? __spreadValues(__spreadValues({}, data), arg) : data;
+  for (const key in allData) {
+    delete pick2[key];
+    const value = allData[key];
+    if (typeof value === "function") {
+      fns[key] = value;
+    } else {
+      result[key] = value;
+    }
+  }
+  const pickedSchema = factory.schema.pick(pick2);
+  return () => {
+    Object.assign(result, zodMock.generateMock(pickedSchema));
+    for (const key in fns) {
+      result[key] = fns[key](factory.sequence);
+    }
+    factory.sequence++;
+    return __spreadValues({}, result);
+  };
+};
+const _TestFactory = class {
+  constructor(model, schema, data = {}, options = {}) {
+    this.model = model;
+    this.schema = schema;
+    this.data = data;
+    this.omitValues = {};
+    this.pickValues = {};
+    var _a;
+    if (options.sequence !== void 0) {
+      this.sequence = options.sequence;
+    } else {
+      let workerId = parseInt(process.env.JEST_WORKER_ID);
+      if (isNaN(workerId))
+        workerId = 1;
+      this.sequence = (workerId - 1) * ((_a = options.sequenceDistance) != null ? _a : 1e3) + 1;
+    }
+  }
+  set(data) {
+    return Object.assign(Object.create(this), {
+      data: __spreadValues(__spreadValues({}, this.data), data)
+    });
+  }
+  omit(keys) {
+    return Object.assign(Object.create(this), {
+      omitValues: __spreadValues(__spreadValues({}, this.omitValues), keys)
+    });
+  }
+  pick(keys) {
+    return Object.assign(Object.create(this), {
+      pickValues: __spreadValues(__spreadValues({}, this.pickValues), keys)
+    });
+  }
+  build(data) {
+    let schema = this.schema;
+    let arg = data ? __spreadValues(__spreadValues({}, this.data), data) : this.data;
+    if (this.omitValues) {
+      schema = schema.omit(this.omitValues);
+      arg = omit(arg, this.omitValues);
+    }
+    if (this.pickValues && Object.keys(this.pickValues).length) {
+      schema = schema.pick(this.pickValues);
+      arg = pick(arg, this.pickValues);
+    }
+    const result = zodMock.generateMock(schema);
+    for (const key in arg) {
+      const value = arg[key];
+      if (typeof value === "function") {
+        result[key] = value(this.sequence);
+      } else {
+        result[key] = value;
+      }
+    }
+    this.sequence++;
+    return result;
+  }
+  buildList(qty, data) {
+    return [...Array(qty)].map(() => this.build(data));
+  }
+  async create(data) {
+    const getData = processCreateData(this, this.data, data);
+    return await this.model.create(getData());
+  }
+  async createList(qty, data) {
+    const getData = processCreateData(this, this.data, data);
+    const arr = [...Array(qty)].map(getData);
+    return await this.model.createMany(arr);
+  }
+  extend() {
+    const { model, schema } = this;
+    return class extends _TestFactory {
+      constructor() {
+        super(model, schema);
+      }
+    };
+  }
+};
+let TestFactory = _TestFactory;
+const nowString = new Date().toISOString();
+const createFactory = (model, options) => {
+  var _a, _b;
+  const schema = orchidOrmSchemaToZod.instanceToZod(model);
+  const data = {};
+  const now = Date.now();
+  for (const key in model.shape) {
+    const column = model.shape[key];
+    if (column instanceof pqb.DateBaseColumn) {
+      if (column.data.as instanceof pqb.IntegerBaseColumn) {
+        data[key] = (sequence) => now + sequence;
+      } else if (((_a = column.parseFn) == null ? void 0 : _a.call(column, nowString)) instanceof Date) {
+        data[key] = (sequence) => new Date(now + sequence);
+      } else {
+        data[key] = (sequence) => new Date(now + sequence).toISOString();
+      }
+    } else if (column instanceof pqb.TextBaseColumn) {
+      const max = (_b = options == null ? void 0 : options.maxTextLength) != null ? _b : 1e3;
+      const item = schema.shape[key];
+      const string = item instanceof zod.ZodNullable ? item.unwrap() : item;
+      const maxCheck = string._def.checks.find(
+        (check) => check.kind === "max"
+      );
+      if (!maxCheck || maxCheck.value[0] > max) {
+        schema.shape[key] = item instanceof zod.ZodNullable ? string.max(max).nullable() : string.max(max);
+      }
+    }
+  }
+  return new TestFactory(
+    model,
+    schema,
+    data,
+    options
+  );
+};
+
+exports.TestFactory = TestFactory;
+exports.createFactory = createFactory;
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sources":["../src/factory.ts"],"sourcesContent":["import { AnyZodObject, ZodNullable, ZodString, ZodTypeAny } from 'zod';\nimport {\n  CreateData,\n  DateBaseColumn,\n  EmptyObject,\n  IntegerBaseColumn,\n  Query,\n  TextBaseColumn,\n} from 'pqb';\nimport { instanceToZod, InstanceToZod } from 'orchid-orm-schema-to-zod';\nimport { generateMock } from '@anatine/zod-mock';\n\ntype FactoryOptions = {\n  sequence?: number;\n  sequenceDistance?: number;\n  maxTextLength?: number;\n};\n\ntype metaKey = typeof metaKey;\nconst metaKey = Symbol('meta');\n\ntype Result<\n  T extends TestFactory,\n  Data,\n  Omitted = Omit<Data, keyof T[metaKey]['omit']>,\n> = EmptyObject extends T[metaKey]['pick']\n  ? Omitted\n  : Pick<\n      Omitted,\n      {\n        [K in keyof Omitted]: K extends keyof T[metaKey]['pick'] ? K : never;\n      }[keyof Omitted]\n    >;\n\ntype BuildArg<T extends TestFactory> = {\n  [K in keyof T[metaKey]['type']]?:\n    | T[metaKey]['type'][K]\n    | ((sequence: number) => T[metaKey]['type'][K]);\n} & Record<string, unknown>;\n\ntype BuildResult<T extends TestFactory, Data extends BuildArg<T>> = Result<\n  T,\n  BuildArg<T> extends Data\n    ? T[metaKey]['type']\n    : T[metaKey]['type'] & {\n        [K in keyof Data]: Data[K] extends () => void\n          ? ReturnType<Data[K]>\n          : Data[K];\n      }\n>;\n\ntype CreateArg<T extends TestFactory> = CreateData<\n  Omit<T['model'], 'inputType'> & {\n    inputType: {\n      [K in keyof T['model']['type']]?:\n        | T['model']['type'][K]\n        | ((sequence: number) => T['model']['type'][K]);\n    };\n  }\n>;\n\ntype CreateResult<T extends TestFactory> = Result<T, T['model']['type']>;\n\nconst omit = <T, Keys extends Record<string, unknown>>(\n  obj: T,\n  keys: Keys,\n): Omit<T, keyof Keys> => {\n  const res = { ...obj };\n  Object.keys(keys).forEach((key) => {\n    delete (res as unknown as Record<string, unknown>)[key];\n  });\n  return res;\n};\n\nconst pick = <T, Keys extends Record<string, unknown>>(\n  obj: T,\n  keys: Keys,\n): Pick<T, { [K in keyof T]: K extends keyof Keys ? K : never }[keyof T]> => {\n  const res = {} as T;\n  Object.keys(keys).forEach((key) => {\n    const value = (obj as unknown as Record<string, unknown>)[key];\n    if (value !== undefined) {\n      (res as unknown as Record<string, unknown>)[key] = value;\n    }\n  });\n  return res;\n};\n\nconst processCreateData = <T extends TestFactory, Data extends CreateArg<T>>(\n  factory: T,\n  data: Record<string, unknown>,\n  arg?: Data,\n) => {\n  const pick: Record<string, true> = {};\n  for (const key in factory.model.shape) {\n    pick[key] = true;\n  }\n\n  factory.model.primaryKeys.forEach((key) => {\n    if (factory.model.shape[key].dataType.includes('serial')) {\n      delete pick[key];\n    }\n  });\n\n  const result: Record<string, unknown> = {};\n\n  const fns: Record<string, (sequence: number) => unknown> = {};\n\n  const allData = (arg ? { ...data, ...arg } : data) as Record<string, unknown>;\n\n  for (const key in allData) {\n    delete pick[key];\n    const value = allData[key];\n    if (typeof value === 'function') {\n      fns[key] = value as () => unknown;\n    } else {\n      result[key] = value;\n    }\n  }\n\n  const pickedSchema = factory.schema.pick(pick);\n\n  return () => {\n    Object.assign(result, generateMock(pickedSchema));\n\n    for (const key in fns) {\n      result[key] = fns[key](factory.sequence);\n    }\n\n    factory.sequence++;\n\n    return { ...result } as CreateData<T['model']>;\n  };\n};\n\nexport class TestFactory<\n  Q extends Query = Query,\n  Schema extends AnyZodObject = AnyZodObject,\n  Type extends EmptyObject = EmptyObject,\n> {\n  sequence: number;\n  private readonly omitValues: Record<PropertyKey, true> = {};\n  private readonly pickValues: Record<PropertyKey, true> = {};\n\n  [metaKey]!: {\n    type: Type;\n    omit: EmptyObject;\n    pick: EmptyObject;\n  };\n\n  constructor(\n    public model: Q,\n    public schema: Schema,\n    private readonly data: Record<string, unknown> = {},\n    options: FactoryOptions = {},\n  ) {\n    if (options.sequence !== undefined) {\n      this.sequence = options.sequence;\n    } else {\n      let workerId = parseInt(process.env.JEST_WORKER_ID as string);\n      if (isNaN(workerId)) workerId = 1;\n      this.sequence = (workerId - 1) * (options.sequenceDistance ?? 1000) + 1;\n    }\n  }\n\n  set<\n    T extends this,\n    Meta extends { type: EmptyObject },\n    Data extends {\n      [K in keyof Meta['type']]?: Meta['type'][K] | (() => Meta['type'][K]);\n    } & Record<string, unknown>,\n  >(\n    this: T & { [metaKey]: Meta },\n    data: Data,\n  ): T & { [metaKey]: Meta & { type: Data } } {\n    return Object.assign(Object.create(this), {\n      data: { ...this.data, ...data },\n    });\n  }\n\n  omit<T extends this, Keys extends { [K in keyof T[metaKey]['type']]?: true }>(\n    this: T,\n    keys: Keys,\n  ): T & { [metaKey]: T[metaKey] & { omit: Keys } } {\n    return Object.assign(Object.create(this), {\n      omitValues: { ...this.omitValues, ...keys },\n    });\n  }\n\n  pick<T extends this, Keys extends { [K in keyof T[metaKey]['type']]?: true }>(\n    this: T,\n    keys: Keys,\n  ): T & { [metaKey]: T[metaKey] & { pick: Keys } } {\n    return Object.assign(Object.create(this), {\n      pickValues: { ...this.pickValues, ...keys },\n    });\n  }\n\n  build<T extends this, Data extends BuildArg<T>>(\n    this: T,\n    data?: Data,\n  ): BuildResult<T, Data> {\n    let schema = this.schema as AnyZodObject;\n    let arg = data ? { ...this.data, ...data } : this.data;\n\n    if (this.omitValues) {\n      schema = schema.omit(this.omitValues);\n      arg = omit(arg, this.omitValues);\n    }\n\n    if (this.pickValues && Object.keys(this.pickValues).length) {\n      schema = schema.pick(this.pickValues);\n      arg = pick(arg, this.pickValues);\n    }\n\n    const result = generateMock(schema) as Record<string, unknown>;\n    for (const key in arg) {\n      const value = (arg as Record<string, unknown>)[key];\n      if (typeof value === 'function') {\n        result[key] = value(this.sequence);\n      } else {\n        result[key] = value;\n      }\n    }\n\n    this.sequence++;\n\n    return result as BuildResult<T, Data>;\n  }\n\n  buildList<T extends this, Data extends BuildArg<T>>(\n    this: T,\n    qty: number,\n    data?: Data,\n  ): BuildResult<T, Data>[] {\n    return [...Array(qty)].map(() => this.build(data));\n  }\n\n  async create<T extends this, Data extends CreateArg<T>>(\n    this: T,\n    data?: Data,\n  ): Promise<CreateResult<T>> {\n    const getData = processCreateData(this, this.data, data);\n    return (await this.model.create(getData())) as CreateResult<T>;\n  }\n\n  async createList<T extends this, Data extends CreateArg<T>>(\n    this: T,\n    qty: number,\n    data?: Data,\n  ): Promise<CreateResult<T>[]> {\n    const getData = processCreateData(this, this.data, data);\n    const arr = [...Array(qty)].map(getData);\n    return (await this.model.createMany(arr)) as CreateResult<T>[];\n  }\n\n  extend<T extends this>(this: T): new () => TestFactory<Q, Schema, Type> {\n    const { model, schema } = this;\n\n    return class extends TestFactory<Q, Schema, Type> {\n      constructor() {\n        super(model, schema);\n      }\n    };\n  }\n}\n\nconst nowString = new Date().toISOString();\n\nexport const createFactory = <T extends Query>(\n  model: T,\n  options?: FactoryOptions,\n) => {\n  const schema = instanceToZod(model);\n\n  const data: Record<string, unknown> = {};\n  const now = Date.now();\n\n  for (const key in model.shape) {\n    const column = model.shape[key];\n    if (column instanceof DateBaseColumn) {\n      if (column.data.as instanceof IntegerBaseColumn) {\n        data[key] = (sequence: number) => now + sequence;\n      } else if (column.parseFn?.(nowString) instanceof Date) {\n        data[key] = (sequence: number) => new Date(now + sequence);\n      } else {\n        data[key] = (sequence: number) =>\n          new Date(now + sequence).toISOString();\n      }\n    } else if (column instanceof TextBaseColumn) {\n      const max = options?.maxTextLength ?? 1000;\n      const item = schema.shape[key];\n      const string = (\n        item instanceof ZodNullable ? item.unwrap() : item\n      ) as ZodString;\n\n      const maxCheck = string._def.checks.find(\n        (check) => check.kind === 'max',\n      ) as { value: [number] } | undefined;\n\n      if (!maxCheck || maxCheck.value[0] > max) {\n        (schema.shape as Record<string, ZodTypeAny>)[key] =\n          item instanceof ZodNullable\n            ? string.max(max).nullable()\n            : string.max(max);\n      }\n    }\n  }\n\n  return new TestFactory<T, InstanceToZod<T>, T['type']>(\n    model,\n    schema,\n    data,\n    options,\n  );\n};\n"],"names":["pick","generateMock","instanceToZod","DateBaseColumn","IntegerBaseColumn","TextBaseColumn","ZodNullable"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;AA+DA,MAAM,IAAA,GAAO,CACX,GAAA,EACA,IACwB,KAAA;AACxB,EAAA,MAAM,MAAM,cAAK,CAAA,EAAA,EAAA,GAAA,CAAA,CAAA;AACjB,EAAA,MAAA,CAAO,IAAK,CAAA,IAAI,CAAE,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACjC,IAAA,OAAQ,GAA2C,CAAA,GAAA,CAAA,CAAA;AAAA,GACpD,CAAA,CAAA;AACD,EAAO,OAAA,GAAA,CAAA;AACT,CAAA,CAAA;AAEA,MAAM,IAAA,GAAO,CACX,GAAA,EACA,IAC2E,KAAA;AAC3E,EAAA,MAAM,MAAM,EAAC,CAAA;AACb,EAAA,MAAA,CAAO,IAAK,CAAA,IAAI,CAAE,CAAA,OAAA,CAAQ,CAAC,GAAQ,KAAA;AACjC,IAAA,MAAM,QAAS,GAA2C,CAAA,GAAA,CAAA,CAAA;AAC1D,IAAA,IAAI,UAAU,KAAW,CAAA,EAAA;AACvB,MAAC,IAA2C,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,KACrD;AAAA,GACD,CAAA,CAAA;AACD,EAAO,OAAA,GAAA,CAAA;AACT,CAAA,CAAA;AAEA,MAAM,iBAAoB,GAAA,CACxB,OACA,EAAA,IAAA,EACA,GACG,KAAA;AACH,EAAA,MAAMA,QAA6B,EAAC,CAAA;AACpC,EAAW,KAAA,MAAA,GAAA,IAAO,OAAQ,CAAA,KAAA,CAAM,KAAO,EAAA;AACrC,IAAAA,MAAK,GAAO,CAAA,GAAA,IAAA,CAAA;AAAA,GACd;AAEA,EAAA,OAAA,CAAQ,KAAM,CAAA,WAAA,CAAY,OAAQ,CAAA,CAAC,GAAQ,KAAA;AACzC,IAAA,IAAI,QAAQ,KAAM,CAAA,KAAA,CAAM,KAAK,QAAS,CAAA,QAAA,CAAS,QAAQ,CAAG,EAAA;AACxD,MAAA,OAAOA,KAAK,CAAA,GAAA,CAAA,CAAA;AAAA,KACd;AAAA,GACD,CAAA,CAAA;AAED,EAAA,MAAM,SAAkC,EAAC,CAAA;AAEzC,EAAA,MAAM,MAAqD,EAAC,CAAA;AAE5D,EAAA,MAAM,OAAW,GAAA,GAAA,GAAM,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAA,EAAS,GAAQ,CAAA,GAAA,IAAA,CAAA;AAE7C,EAAA,KAAA,MAAW,OAAO,OAAS,EAAA;AACzB,IAAA,OAAOA,KAAK,CAAA,GAAA,CAAA,CAAA;AACZ,IAAA,MAAM,QAAQ,OAAQ,CAAA,GAAA,CAAA,CAAA;AACtB,IAAI,IAAA,OAAO,UAAU,UAAY,EAAA;AAC/B,MAAA,GAAA,CAAI,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,KACN,MAAA;AACL,MAAA,MAAA,CAAO,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,KAChB;AAAA,GACF;AAEA,EAAA,MAAM,YAAe,GAAA,OAAA,CAAQ,MAAO,CAAA,IAAA,CAAKA,KAAI,CAAA,CAAA;AAE7C,EAAA,OAAO,MAAM;AACX,IAAA,MAAA,CAAO,MAAO,CAAA,MAAA,EAAQC,oBAAa,CAAA,YAAY,CAAC,CAAA,CAAA;AAEhD,IAAA,KAAA,MAAW,OAAO,GAAK,EAAA;AACrB,MAAA,MAAA,CAAO,GAAO,CAAA,GAAA,GAAA,CAAI,GAAK,CAAA,CAAA,OAAA,CAAQ,QAAQ,CAAA,CAAA;AAAA,KACzC;AAEA,IAAQ,OAAA,CAAA,QAAA,EAAA,CAAA;AAER,IAAA,OAAO,cAAK,CAAA,EAAA,EAAA,MAAA,CAAA,CAAA;AAAA,GACd,CAAA;AACF,CAAA,CAAA;AAEO,MAAM,eAAN,MAIL;AAAA,EAWA,WAAA,CACS,OACA,MACU,EAAA,IAAA,GAAgC,EACjD,EAAA,OAAA,GAA0B,EAC1B,EAAA;AAJO,IAAA,IAAA,CAAA,KAAA,GAAA,KAAA,CAAA;AACA,IAAA,IAAA,CAAA,MAAA,GAAA,MAAA,CAAA;AACU,IAAA,IAAA,CAAA,IAAA,GAAA,IAAA,CAAA;AAZnB,IAAA,IAAA,CAAiB,aAAwC,EAAC,CAAA;AAC1D,IAAA,IAAA,CAAiB,aAAwC,EAAC,CAAA;AA9I5D,IAAA,IAAA,EAAA,CAAA;AA4JI,IAAI,IAAA,OAAA,CAAQ,aAAa,KAAW,CAAA,EAAA;AAClC,MAAA,IAAA,CAAK,WAAW,OAAQ,CAAA,QAAA,CAAA;AAAA,KACnB,MAAA;AACL,MAAA,IAAI,QAAW,GAAA,QAAA,CAAS,OAAQ,CAAA,GAAA,CAAI,cAAwB,CAAA,CAAA;AAC5D,MAAA,IAAI,MAAM,QAAQ,CAAA;AAAG,QAAW,QAAA,GAAA,CAAA,CAAA;AAChC,MAAA,IAAA,CAAK,YAAY,QAAW,GAAA,CAAA,KAAA,CAAM,EAAQ,GAAA,OAAA,CAAA,gBAAA,KAAR,YAA4B,GAAQ,CAAA,GAAA,CAAA,CAAA;AAAA,KACxE;AAAA,GACF;AAAA,EAEA,IAQE,IAC0C,EAAA;AAC1C,IAAA,OAAO,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AAAA,MACxC,IAAA,EAAM,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAK,IAAS,CAAA,EAAA,IAAA,CAAA;AAAA,KAC1B,CAAA,CAAA;AAAA,GACH;AAAA,EAEA,KAEE,IACgD,EAAA;AAChD,IAAA,OAAO,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AAAA,MACxC,UAAA,EAAY,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAK,UAAe,CAAA,EAAA,IAAA,CAAA;AAAA,KACtC,CAAA,CAAA;AAAA,GACH;AAAA,EAEA,KAEE,IACgD,EAAA;AAChD,IAAA,OAAO,MAAO,CAAA,MAAA,CAAO,MAAO,CAAA,MAAA,CAAO,IAAI,CAAG,EAAA;AAAA,MACxC,UAAA,EAAY,cAAK,CAAA,cAAA,CAAA,EAAA,EAAA,IAAA,CAAK,UAAe,CAAA,EAAA,IAAA,CAAA;AAAA,KACtC,CAAA,CAAA;AAAA,GACH;AAAA,EAEA,MAEE,IACsB,EAAA;AACtB,IAAA,IAAI,SAAS,IAAK,CAAA,MAAA,CAAA;AAClB,IAAA,IAAI,MAAM,IAAO,GAAA,cAAA,CAAA,cAAA,CAAA,EAAA,EAAK,IAAK,CAAA,IAAA,CAAA,EAAS,QAAS,IAAK,CAAA,IAAA,CAAA;AAElD,IAAA,IAAI,KAAK,UAAY,EAAA;AACnB,MAAS,MAAA,GAAA,MAAA,CAAO,IAAK,CAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AACpC,MAAM,GAAA,GAAA,IAAA,CAAK,GAAK,EAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AAAA,KACjC;AAEA,IAAA,IAAI,KAAK,UAAc,IAAA,MAAA,CAAO,KAAK,IAAK,CAAA,UAAU,EAAE,MAAQ,EAAA;AAC1D,MAAS,MAAA,GAAA,MAAA,CAAO,IAAK,CAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AACpC,MAAM,GAAA,GAAA,IAAA,CAAK,GAAK,EAAA,IAAA,CAAK,UAAU,CAAA,CAAA;AAAA,KACjC;AAEA,IAAM,MAAA,MAAA,GAASA,qBAAa,MAAM,CAAA,CAAA;AAClC,IAAA,KAAA,MAAW,OAAO,GAAK,EAAA;AACrB,MAAA,MAAM,QAAS,GAAgC,CAAA,GAAA,CAAA,CAAA;AAC/C,MAAI,IAAA,OAAO,UAAU,UAAY,EAAA;AAC/B,QAAO,MAAA,CAAA,GAAA,CAAA,GAAO,KAAM,CAAA,IAAA,CAAK,QAAQ,CAAA,CAAA;AAAA,OAC5B,MAAA;AACL,QAAA,MAAA,CAAO,GAAO,CAAA,GAAA,KAAA,CAAA;AAAA,OAChB;AAAA,KACF;AAEA,IAAK,IAAA,CAAA,QAAA,EAAA,CAAA;AAEL,IAAO,OAAA,MAAA,CAAA;AAAA,GACT;AAAA,EAEA,SAAA,CAEE,KACA,IACwB,EAAA;AACxB,IAAO,OAAA,CAAC,GAAG,KAAA,CAAM,GAAG,CAAC,CAAE,CAAA,GAAA,CAAI,MAAM,IAAA,CAAK,KAAM,CAAA,IAAI,CAAC,CAAA,CAAA;AAAA,GACnD;AAAA,EAEA,MAAM,OAEJ,IAC0B,EAAA;AAC1B,IAAA,MAAM,OAAU,GAAA,iBAAA,CAAkB,IAAM,EAAA,IAAA,CAAK,MAAM,IAAI,CAAA,CAAA;AACvD,IAAA,OAAQ,MAAM,IAAA,CAAK,KAAM,CAAA,MAAA,CAAO,SAAS,CAAA,CAAA;AAAA,GAC3C;AAAA,EAEA,MAAM,UAEJ,CAAA,GAAA,EACA,IAC4B,EAAA;AAC5B,IAAA,MAAM,OAAU,GAAA,iBAAA,CAAkB,IAAM,EAAA,IAAA,CAAK,MAAM,IAAI,CAAA,CAAA;AACvD,IAAM,MAAA,GAAA,GAAM,CAAC,GAAG,KAAA,CAAM,GAAG,CAAC,CAAA,CAAE,IAAI,OAAO,CAAA,CAAA;AACvC,IAAA,OAAQ,MAAM,IAAA,CAAK,KAAM,CAAA,UAAA,CAAW,GAAG,CAAA,CAAA;AAAA,GACzC;AAAA,EAEA,MAAwE,GAAA;AACtE,IAAM,MAAA,EAAE,KAAO,EAAA,MAAA,EAAW,GAAA,IAAA,CAAA;AAE1B,IAAA,OAAO,cAAc,YAA6B,CAAA;AAAA,MAChD,WAAc,GAAA;AACZ,QAAA,KAAA,CAAM,OAAO,MAAM,CAAA,CAAA;AAAA,OACrB;AAAA,KACF,CAAA;AAAA,GACF;AACF,CAAA,CAAA;AAlIO,IAAM,WAAN,GAAA,aAAA;AAoIP,MAAM,SAAY,GAAA,IAAI,IAAK,EAAA,CAAE,WAAY,EAAA,CAAA;AAE5B,MAAA,aAAA,GAAgB,CAC3B,KAAA,EACA,OACG,KAAA;AAhRL,EAAA,IAAA,EAAA,EAAA,EAAA,CAAA;AAiRE,EAAM,MAAA,MAAA,GAASC,mCAAc,KAAK,CAAA,CAAA;AAElC,EAAA,MAAM,OAAgC,EAAC,CAAA;AACvC,EAAM,MAAA,GAAA,GAAM,KAAK,GAAI,EAAA,CAAA;AAErB,EAAW,KAAA,MAAA,GAAA,IAAO,MAAM,KAAO,EAAA;AAC7B,IAAM,MAAA,MAAA,GAAS,MAAM,KAAM,CAAA,GAAA,CAAA,CAAA;AAC3B,IAAA,IAAI,kBAAkBC,kBAAgB,EAAA;AACpC,MAAI,IAAA,MAAA,CAAO,IAAK,CAAA,EAAA,YAAcC,qBAAmB,EAAA;AAC/C,QAAK,IAAA,CAAA,GAAA,CAAA,GAAO,CAAC,QAAA,KAAqB,GAAM,GAAA,QAAA,CAAA;AAAA,OAC/B,MAAA,IAAA,CAAA,CAAA,EAAA,GAAA,MAAA,CAAO,OAAP,KAAA,IAAA,GAAA,KAAA,CAAA,GAAA,EAAA,CAAA,IAAA,CAAA,MAAA,EAAiB,uBAAsB,IAAM,EAAA;AACtD,QAAA,IAAA,CAAK,OAAO,CAAC,QAAA,KAAqB,IAAI,IAAA,CAAK,MAAM,QAAQ,CAAA,CAAA;AAAA,OACpD,MAAA;AACL,QAAK,IAAA,CAAA,GAAA,CAAA,GAAO,CAAC,QACX,KAAA,IAAI,KAAK,GAAM,GAAA,QAAQ,EAAE,WAAY,EAAA,CAAA;AAAA,OACzC;AAAA,KACF,MAAA,IAAW,kBAAkBC,kBAAgB,EAAA;AAC3C,MAAM,MAAA,GAAA,GAAA,CAAM,EAAS,GAAA,OAAA,IAAA,IAAA,GAAA,KAAA,CAAA,GAAA,OAAA,CAAA,aAAA,KAAT,IAA0B,GAAA,EAAA,GAAA,GAAA,CAAA;AACtC,MAAM,MAAA,IAAA,GAAO,OAAO,KAAM,CAAA,GAAA,CAAA,CAAA;AAC1B,MAAA,MAAM,MACJ,GAAA,IAAA,YAAgBC,eAAc,GAAA,IAAA,CAAK,QAAW,GAAA,IAAA,CAAA;AAGhD,MAAM,MAAA,QAAA,GAAW,MAAO,CAAA,IAAA,CAAK,MAAO,CAAA,IAAA;AAAA,QAClC,CAAC,KAAU,KAAA,KAAA,CAAM,IAAS,KAAA,KAAA;AAAA,OAC5B,CAAA;AAEA,MAAA,IAAI,CAAC,QAAA,IAAY,QAAS,CAAA,KAAA,CAAM,KAAK,GAAK,EAAA;AACxC,QAAC,MAAO,CAAA,KAAA,CAAqC,GAC3C,CAAA,GAAA,IAAA,YAAgBA,eACZ,GAAA,MAAA,CAAO,GAAI,CAAA,GAAG,CAAE,CAAA,QAAA,EAChB,GAAA,MAAA,CAAO,IAAI,GAAG,CAAA,CAAA;AAAA,OACtB;AAAA,KACF;AAAA,GACF;AAEA,EAAA,OAAO,IAAI,WAAA;AAAA,IACT,KAAA;AAAA,IACA,MAAA;AAAA,IACA,IAAA;AAAA,IACA,OAAA;AAAA,GACF,CAAA;AACF;;;;;"}

package/jest-setup.ts

@@ -0,0 +1,23 @@
+import {
+  patchPgForTransactions,
+  startTransaction,
+  rollbackTransaction,
+} from 'pg-transactional-tests';
+import { db } from './src/test-utils';
+
+jest.mock('pqb', () => require('../pqb/src'), {
+  virtual: true,
+});
+jest.mock('orchid-orm', () => require('../orchid-orm/src'), {
+  virtual: true,
+});
+
+patchPgForTransactions();
+
+beforeAll(startTransaction);
+beforeEach(startTransaction);
+afterEach(rollbackTransaction);
+afterAll(async () => {
+  await rollbackTransaction();
+  await db.$close();
+});

package/package.json

@@ -0,0 +1,69 @@
+{
+  "name": "orchid-orm-test-factory",
+  "version": "0.0.1",
+  "description": "Orchid ORM factories for tests",
+  "homepage": "https://orchid-orm.netlify.app/guide/orm-test-factories",
+  "repository": {
+    "type": "git",
+    "url": "https://github.com/romeerez/orchid-orm/tree/main/packages/test-factory"
+  },
+  "main": "dist/index.js",
+  "module": "dist/index.esm.js",
+  "typings": "dist/index.d.ts",
+  "exports": {
+    ".": {
+      "require": "./dist/index.js",
+      "import": "./dist/index.esm.js",
+      "types": "./dist/index.d.ts"
+    }
+  },
+  "jest": {
+    "setupFiles": [
+      "dotenv/config"
+    ],
+    "globalSetup": "../../jest-global-setup.ts",
+    "setupFilesAfterEnv": [
+      "../../jest-setup.ts",
+      "./jest-setup.ts"
+    ],
+    "transform": {
+      "^.+\\.ts$": "@swc/jest"
+    }
+  },
+  "keywords": [
+    "orchid-orm",
+    "test",
+    "factory"
+  ],
+  "author": "Roman Kushyn",
+  "license": "ISC",
+  "dependencies": {
+    "orchid-orm": "0.0.1",
+    "pqb": "0.5.4",
+    "orchid-orm-schema-to-zod": "0.0.1",
+    "@faker-js/faker": "^7.6.0",
+    "@anatine/zod-mock": "^3.6.0",
+    "zod": "^3.19.1"
+  },
+  "devDependencies": {
+    "rollup": "^2.79.0",
+    "rollup-plugin-dts": "^4.2.2",
+    "rollup-plugin-esbuild": "^4.10.1",
+    "@swc/core": "^1.2.210",
+    "@swc/jest": "^0.2.21",
+    "@types/jest": "^28.1.2",
+    "@types/node": "^18.0.1",
+    "jest": "^28.1.2",
+    "rimraf": "^3.0.2",
+    "tslib": "^2.4.0",
+    "typescript": "^4.7.4",
+    "dotenv": "^16.0.1",
+    "pg-transactional-tests": "^1.0.5",
+    "pg": "^8.8.0",
+    "@types/pg": "^8.6.5"
+  },
+  "scripts": {
+    "test": "jest",
+    "build": "rimraf ./dist/ && rollup -c --rollup.config"
+  }
+}

package/rollup.config.js

@@ -0,0 +1,3 @@
+import config from '../../rollup.config';
+
+export default config;

package/src/factory.test.ts

@@ -0,0 +1,304 @@
+import { createFactory } from './factory';
+import { assertType, db, User, Model, adapter } from './test-utils';
+import { z } from 'zod';
+import { orchidORM } from 'orchid-orm';
+
+describe('factory', () => {
+  describe('sequence and sequenceDistance', () => {
+    beforeAll(() => {
+      process.env.JEST_WORKER_ID = '5';
+    });
+
+    afterAll(() => {
+      process.env.JEST_WORKER_ID = '1';
+    });
+
+    it('should depend on process.env.JEST_WORKER_ID when it is defined', () => {
+      const factory = createFactory(db.user);
+      expect(factory.sequence).toBe(4001);
+    });
+
+    it('should allow to override sequence', () => {
+      const factory = createFactory(db.user, {
+        sequence: 123,
+      });
+      expect(factory.sequence).toBe(123);
+    });
+
+    it('should allow to override sequence distance', () => {
+      const factory = createFactory(db.user, {
+        sequenceDistance: 100,
+      });
+      expect(factory.sequence).toBe(401);
+    });
+  });
+
+  const userFactory = createFactory(db.user);
+
+  describe('build', () => {
+    it('should build an object for the model', () => {
+      const data = userFactory.build();
+
+      assertType<typeof data, User>();
+
+      expect(() => userFactory.schema.parse(data)).not.toThrow();
+    });
+
+    it('should accept data with values to override result', () => {
+      const data = userFactory.build({
+        age: 18,
+        name: 'name',
+        extra: true,
+      });
+
+      assertType<
+        typeof data,
+        User & { name: 'name'; age: 18; extra: boolean }
+      >();
+
+      expect(data).toMatchObject({ age: 18, name: 'name', extra: true });
+    });
+
+    it('should accept data with functions to override result', () => {
+      const data = userFactory.build({
+        age: () => 18,
+        name: () => 'name',
+        extra: () => true,
+      });
+
+      assertType<typeof data, User & { age: number; extra: true }>();
+
+      expect(data).toMatchObject({ age: 18, name: 'name', extra: true });
+    });
+
+    it('should limit long strings with 1000 by default', () => {
+      const profileFactory = createFactory(db.profile);
+      const data = profileFactory.build();
+
+      expect(data.bio.length).toBeLessThanOrEqual(1000);
+    });
+
+    it('should respect max which is set on column', () => {
+      class ProfileModel extends Model {
+        table = 'profile';
+        columns = this.setColumns((t) => ({
+          bio: t.text().min(100).max(120),
+        }));
+      }
+
+      const db = orchidORM(
+        {
+          adapter,
+        },
+        {
+          profile: ProfileModel,
+        },
+      );
+
+      const profileFactory = createFactory(db.profile);
+      const data = profileFactory.build();
+
+      expect(data.bio.length).toBeLessThanOrEqual(120);
+    });
+
+    it('should allow to override maxTextLength', () => {
+      const profileFactory = createFactory(db.profile, {
+        maxTextLength: 500,
+      });
+      const data = profileFactory.build();
+
+      expect(data.bio.length).toBeLessThanOrEqual(500);
+    });
+  });
+
+  describe('buildList', () => {
+    const original = userFactory.build;
+    const buildMock = jest.fn();
+
+    beforeAll(() => {
+      userFactory.build = buildMock;
+    });
+
+    afterAll(() => {
+      userFactory.build = original;
+    });
+
+    it('should call build provided number of times, pass the argument, return array', () => {
+      const arg = { extra: true };
+      const arr = userFactory.buildList(3, arg);
+
+      assertType<typeof arr, (User & { extra: boolean })[]>();
+
+      expect(buildMock).toHaveBeenCalledTimes(3);
+      expect(buildMock).toHaveBeenCalledWith(arg);
+    });
+  });
+
+  describe('omit', () => {
+    it('should allow to build data with omitted fields', () => {
+      const data = userFactory.omit({ id: true, name: true }).build();
+
+      assertType<typeof data, Omit<User, 'id' | 'name'>>();
+
+      expect(() =>
+        userFactory.schema.strict().omit({ id: true, name: true }).parse(data),
+      ).not.toThrow();
+    });
+  });
+
+  describe('pick', () => {
+    it('should allow to build data with picked fields', () => {
+      const data = userFactory.pick({ id: true, name: true }).build();
+
+      assertType<typeof data, Pick<User, 'id' | 'name'>>();
+
+      expect(() =>
+        userFactory.schema.strict().pick({ id: true, name: true }).parse(data),
+      ).not.toThrow();
+    });
+  });
+
+  describe('create', () => {
+    it('should create record with generated data, except serial primary keys, datetime numbers should be the same in the record and to be around now', async () => {
+      const item = await userFactory.create();
+      const now = Date.now();
+
+      expect(item.createdAt).toBe(item.updatedAt);
+
+      expect(Math.floor(item.createdAt / 1000)).toEqual(Math.floor(now / 1000));
+
+      assertType<typeof item, User>();
+
+      expect(() => userFactory.schema.parse(item)).not.toThrow();
+    });
+
+    it('should create record with overridden data', async () => {
+      const item = await userFactory.create({ name: 'name' });
+
+      assertType<typeof item, User>();
+
+      expect(item.name).toBe('name');
+    });
+
+    it('should create record with nested create', async () => {
+      const user = await userFactory.create({
+        profile: {
+          create: {
+            bio: 'bio',
+          },
+        },
+      });
+
+      assertType<typeof user, User>();
+    });
+  });
+
+  describe('createList', () => {
+    it('should create a list of records, datetime numbers should be the same in one record and increase for each next record', async () => {
+      const items = await userFactory.createList(2);
+
+      assertType<typeof items, User[]>();
+
+      expect(items[0].name).not.toBe(items[1].name);
+
+      expect(items[0].createdAt).toEqual(items[0].updatedAt);
+      expect(items[1].createdAt).toEqual(items[1].updatedAt);
+
+      expect(items[0].createdAt).toBeLessThan(items[1].createdAt);
+
+      expect(() => z.array(userFactory.schema).parse(items)).not.toThrow();
+    });
+
+    it('should create a list of records with overridden data', async () => {
+      const items = await userFactory.createList(2, { name: 'name' });
+
+      assertType<typeof items, User[]>();
+
+      expect(items.map((item) => item.name)).toEqual(['name', 'name']);
+    });
+  });
+
+  describe('set', () => {
+    it('should set data to override result and work with build', () => {
+      const data = userFactory
+        .set({
+          age: 18,
+        })
+        .build({
+          name: 'name',
+        });
+
+      assertType<typeof data, User & { age: number; name: 'name' }>();
+
+      expect(data).toMatchObject({ age: 18, name: 'name' });
+    });
+
+    it('should set data to override result and work with buildList', () => {
+      const arr = userFactory
+        .set({
+          age: 18,
+        })
+        .buildList(2, {
+          name: 'name',
+        });
+
+      assertType<typeof arr, (User & { age: number; name: 'name' })[]>();
+    });
+
+    it('should set data to override result and work with create', async () => {
+      const item = await userFactory.set({ age: 18 }).create();
+
+      assertType<typeof item, User>();
+
+      expect(() => userFactory.schema.parse(item)).not.toThrow();
+      expect(item.age).toBe(18);
+    });
+
+    it('should set data to override result and work with createList', async () => {
+      const items = await userFactory.set({ age: 18 }).createList(2);
+
+      assertType<typeof items, User[]>();
+
+      expect(() => z.array(userFactory.schema).parse(items)).not.toThrow();
+      expect(items.map((item) => item.age)).toEqual([18, 18]);
+    });
+  });
+
+  describe('custom methods', () => {
+    class ExtendedFactory extends createFactory(db.user).extend() {
+      specificUser(age: number) {
+        return this.otherMethod().set({
+          age,
+          name: 'specific',
+        });
+      }
+      otherMethod() {
+        return this.set({ extra: true });
+      }
+    }
+
+    const extendedFactory = new ExtendedFactory();
+
+    it('should respect omitted fields and build a proper object', async () => {
+      const data = extendedFactory.omit({ id: true }).specificUser(42).build();
+
+      assertType<
+        typeof data,
+        Omit<User, 'id'> & { age: number; extra: boolean }
+      >();
+
+      expect(data).toMatchObject({ age: 42, name: 'specific', extra: true });
+    });
+
+    it('should respect picked fields and build a proper object', async () => {
+      const data = extendedFactory
+        .pick({ age: true, name: true })
+        .specificUser(42)
+        .build();
+
+      assertType<typeof data, Pick<User, 'name'> & { age: number }>();
+
+      expect(data).toEqual({ age: 42, name: 'specific' });
+    });
+  });
+});

package/src/factory.ts

@@ -0,0 +1,316 @@
+import { AnyZodObject, ZodNullable, ZodString, ZodTypeAny } from 'zod';
+import {
+  CreateData,
+  DateBaseColumn,
+  EmptyObject,
+  IntegerBaseColumn,
+  Query,
+  TextBaseColumn,
+} from 'pqb';
+import { instanceToZod, InstanceToZod } from 'orchid-orm-schema-to-zod';
+import { generateMock } from '@anatine/zod-mock';
+
+type FactoryOptions = {
+  sequence?: number;
+  sequenceDistance?: number;
+  maxTextLength?: number;
+};
+
+type metaKey = typeof metaKey;
+const metaKey = Symbol('meta');
+
+type Result<
+  T extends TestFactory,
+  Data,
+  Omitted = Omit<Data, keyof T[metaKey]['omit']>,
+> = EmptyObject extends T[metaKey]['pick']
+  ? Omitted
+  : Pick<
+      Omitted,
+      {
+        [K in keyof Omitted]: K extends keyof T[metaKey]['pick'] ? K : never;
+      }[keyof Omitted]
+    >;
+
+type BuildArg<T extends TestFactory> = {
+  [K in keyof T[metaKey]['type']]?:
+    | T[metaKey]['type'][K]
+    | ((sequence: number) => T[metaKey]['type'][K]);
+} & Record<string, unknown>;
+
+type BuildResult<T extends TestFactory, Data extends BuildArg<T>> = Result<
+  T,
+  BuildArg<T> extends Data
+    ? T[metaKey]['type']
+    : T[metaKey]['type'] & {
+        [K in keyof Data]: Data[K] extends () => void
+          ? ReturnType<Data[K]>
+          : Data[K];
+      }
+>;
+
+type CreateArg<T extends TestFactory> = CreateData<
+  Omit<T['model'], 'inputType'> & {
+    inputType: {
+      [K in keyof T['model']['type']]?:
+        | T['model']['type'][K]
+        | ((sequence: number) => T['model']['type'][K]);
+    };
+  }
+>;
+
+type CreateResult<T extends TestFactory> = Result<T, T['model']['type']>;
+
+const omit = <T, Keys extends Record<string, unknown>>(
+  obj: T,
+  keys: Keys,
+): Omit<T, keyof Keys> => {
+  const res = { ...obj };
+  Object.keys(keys).forEach((key) => {
+    delete (res as unknown as Record<string, unknown>)[key];
+  });
+  return res;
+};
+
+const pick = <T, Keys extends Record<string, unknown>>(
+  obj: T,
+  keys: Keys,
+): Pick<T, { [K in keyof T]: K extends keyof Keys ? K : never }[keyof T]> => {
+  const res = {} as T;
+  Object.keys(keys).forEach((key) => {
+    const value = (obj as unknown as Record<string, unknown>)[key];
+    if (value !== undefined) {
+      (res as unknown as Record<string, unknown>)[key] = value;
+    }
+  });
+  return res;
+};
+
+const processCreateData = <T extends TestFactory, Data extends CreateArg<T>>(
+  factory: T,
+  data: Record<string, unknown>,
+  arg?: Data,
+) => {
+  const pick: Record<string, true> = {};
+  for (const key in factory.model.shape) {
+    pick[key] = true;
+  }
+
+  factory.model.primaryKeys.forEach((key) => {
+    if (factory.model.shape[key].dataType.includes('serial')) {
+      delete pick[key];
+    }
+  });
+
+  const result: Record<string, unknown> = {};
+
+  const fns: Record<string, (sequence: number) => unknown> = {};
+
+  const allData = (arg ? { ...data, ...arg } : data) as Record<string, unknown>;
+
+  for (const key in allData) {
+    delete pick[key];
+    const value = allData[key];
+    if (typeof value === 'function') {
+      fns[key] = value as () => unknown;
+    } else {
+      result[key] = value;
+    }
+  }
+
+  const pickedSchema = factory.schema.pick(pick);
+
+  return () => {
+    Object.assign(result, generateMock(pickedSchema));
+
+    for (const key in fns) {
+      result[key] = fns[key](factory.sequence);
+    }
+
+    factory.sequence++;
+
+    return { ...result } as CreateData<T['model']>;
+  };
+};
+
+export class TestFactory<
+  Q extends Query = Query,
+  Schema extends AnyZodObject = AnyZodObject,
+  Type extends EmptyObject = EmptyObject,
+> {
+  sequence: number;
+  private readonly omitValues: Record<PropertyKey, true> = {};
+  private readonly pickValues: Record<PropertyKey, true> = {};
+
+  [metaKey]!: {
+    type: Type;
+    omit: EmptyObject;
+    pick: EmptyObject;
+  };
+
+  constructor(
+    public model: Q,
+    public schema: Schema,
+    private readonly data: Record<string, unknown> = {},
+    options: FactoryOptions = {},
+  ) {
+    if (options.sequence !== undefined) {
+      this.sequence = options.sequence;
+    } else {
+      let workerId = parseInt(process.env.JEST_WORKER_ID as string);
+      if (isNaN(workerId)) workerId = 1;
+      this.sequence = (workerId - 1) * (options.sequenceDistance ?? 1000) + 1;
+    }
+  }
+
+  set<
+    T extends this,
+    Meta extends { type: EmptyObject },
+    Data extends {
+      [K in keyof Meta['type']]?: Meta['type'][K] | (() => Meta['type'][K]);
+    } & Record<string, unknown>,
+  >(
+    this: T & { [metaKey]: Meta },
+    data: Data,
+  ): T & { [metaKey]: Meta & { type: Data } } {
+    return Object.assign(Object.create(this), {
+      data: { ...this.data, ...data },
+    });
+  }
+
+  omit<T extends this, Keys extends { [K in keyof T[metaKey]['type']]?: true }>(
+    this: T,
+    keys: Keys,
+  ): T & { [metaKey]: T[metaKey] & { omit: Keys } } {
+    return Object.assign(Object.create(this), {
+      omitValues: { ...this.omitValues, ...keys },
+    });
+  }
+
+  pick<T extends this, Keys extends { [K in keyof T[metaKey]['type']]?: true }>(
+    this: T,
+    keys: Keys,
+  ): T & { [metaKey]: T[metaKey] & { pick: Keys } } {
+    return Object.assign(Object.create(this), {
+      pickValues: { ...this.pickValues, ...keys },
+    });
+  }
+
+  build<T extends this, Data extends BuildArg<T>>(
+    this: T,
+    data?: Data,
+  ): BuildResult<T, Data> {
+    let schema = this.schema as AnyZodObject;
+    let arg = data ? { ...this.data, ...data } : this.data;
+
+    if (this.omitValues) {
+      schema = schema.omit(this.omitValues);
+      arg = omit(arg, this.omitValues);
+    }
+
+    if (this.pickValues && Object.keys(this.pickValues).length) {
+      schema = schema.pick(this.pickValues);
+      arg = pick(arg, this.pickValues);
+    }
+
+    const result = generateMock(schema) as Record<string, unknown>;
+    for (const key in arg) {
+      const value = (arg as Record<string, unknown>)[key];
+      if (typeof value === 'function') {
+        result[key] = value(this.sequence);
+      } else {
+        result[key] = value;
+      }
+    }
+
+    this.sequence++;
+
+    return result as BuildResult<T, Data>;
+  }
+
+  buildList<T extends this, Data extends BuildArg<T>>(
+    this: T,
+    qty: number,
+    data?: Data,
+  ): BuildResult<T, Data>[] {
+    return [...Array(qty)].map(() => this.build(data));
+  }
+
+  async create<T extends this, Data extends CreateArg<T>>(
+    this: T,
+    data?: Data,
+  ): Promise<CreateResult<T>> {
+    const getData = processCreateData(this, this.data, data);
+    return (await this.model.create(getData())) as CreateResult<T>;
+  }
+
+  async createList<T extends this, Data extends CreateArg<T>>(
+    this: T,
+    qty: number,
+    data?: Data,
+  ): Promise<CreateResult<T>[]> {
+    const getData = processCreateData(this, this.data, data);
+    const arr = [...Array(qty)].map(getData);
+    return (await this.model.createMany(arr)) as CreateResult<T>[];
+  }
+
+  extend<T extends this>(this: T): new () => TestFactory<Q, Schema, Type> {
+    const { model, schema } = this;
+
+    return class extends TestFactory<Q, Schema, Type> {
+      constructor() {
+        super(model, schema);
+      }
+    };
+  }
+}
+
+const nowString = new Date().toISOString();
+
+export const createFactory = <T extends Query>(
+  model: T,
+  options?: FactoryOptions,
+) => {
+  const schema = instanceToZod(model);
+
+  const data: Record<string, unknown> = {};
+  const now = Date.now();
+
+  for (const key in model.shape) {
+    const column = model.shape[key];
+    if (column instanceof DateBaseColumn) {
+      if (column.data.as instanceof IntegerBaseColumn) {
+        data[key] = (sequence: number) => now + sequence;
+      } else if (column.parseFn?.(nowString) instanceof Date) {
+        data[key] = (sequence: number) => new Date(now + sequence);
+      } else {
+        data[key] = (sequence: number) =>
+          new Date(now + sequence).toISOString();
+      }
+    } else if (column instanceof TextBaseColumn) {
+      const max = options?.maxTextLength ?? 1000;
+      const item = schema.shape[key];
+      const string = (
+        item instanceof ZodNullable ? item.unwrap() : item
+      ) as ZodString;
+
+      const maxCheck = string._def.checks.find(
+        (check) => check.kind === 'max',
+      ) as { value: [number] } | undefined;
+
+      if (!maxCheck || maxCheck.value[0] > max) {
+        (schema.shape as Record<string, ZodTypeAny>)[key] =
+          item instanceof ZodNullable
+            ? string.max(max).nullable()
+            : string.max(max);
+      }
+    }
+  }
+
+  return new TestFactory<T, InstanceToZod<T>, T['type']>(
+    model,
+    schema,
+    data,
+    options,
+  );
+};

package/src/index.ts

@@ -0,0 +1 @@
+export * from './factory';

package/src/test-utils.ts

@@ -0,0 +1,87 @@
+import { createModel, orchidORM } from 'orchid-orm';
+import { Adapter, columnTypes } from 'pqb';
+
+type AssertEqual<T, Expected> = [T] extends [Expected]
+  ? [Expected] extends [T]
+    ? true
+    : false
+  : false;
+
+export const assertType = <T, Expected>(
+  ..._: AssertEqual<T, Expected> extends true ? [] : ['invalid type']
+) => {
+  // noop
+};
+
+export const Model = createModel({
+  columnTypes: {
+    ...columnTypes,
+    timestamp: () => columnTypes.timestamp().asNumber(),
+  },
+});
+
+export type User = UserModel['columns']['type'];
+class UserModel extends Model {
+  table = 'user';
+  columns = this.setColumns((t) => ({
+    id: t.serial().primaryKey(),
+    name: t.text(),
+    password: t.text(),
+    picture: t.text().nullable(),
+    data: t
+      .json((j) =>
+        j.object({
+          name: j.string(),
+          tags: j.string().array(),
+        }),
+      )
+      .nullable(),
+    age: t.integer().nullable(),
+    active: t.boolean().nullable(),
+    ...t.timestamps(),
+  }));
+
+  relations = {
+    profile: this.hasOne(() => ProfileModel, {
+      required: true,
+      primaryKey: 'id',
+      foreignKey: 'userId',
+    }),
+  };
+}
+
+export class ProfileModel extends Model {
+  table = 'profile';
+  columns = this.setColumns((t) => ({
+    id: t.serial().primaryKey(),
+    userId: t
+      .integer()
+      .nullable()
+      .foreignKey(() => UserModel, 'id'),
+    bio: t.text().min(100).max(100000),
+    ...t.timestamps(),
+  }));
+
+  relations = {
+    user: this.belongsTo(() => UserModel, {
+      required: true,
+      primaryKey: 'id',
+      foreignKey: 'userId',
+    }),
+  };
+}
+
+export const adapter = new Adapter({
+  connectionString: process.env.DATABASE_URL,
+});
+
+export const db = orchidORM(
+  {
+    adapter,
+    log: false,
+  },
+  {
+    user: UserModel,
+    profile: ProfileModel,
+  },
+);

package/tsconfig.json

@@ -0,0 +1,13 @@
+{
+  "extends": "../../tsconfig.json",
+  "include": ["./src", "./jest-setup.ts"],
+  "compilerOptions": {
+    "outDir": "./dist",
+    "noEmit": false,
+    "baseUrl": ".",
+    "paths": {
+      "orchid-orm": ["../orchid-orm/src"],
+      "orchid-orm-schema-to-zod": ["../schema-to-zod/src"]
+    }
+  }
+}