UNPKG

zod

Version:

TypeScript-first schema declaration and validation library with static type inference

352 lines (315 loc) • 9.3 kB
import { expect, expectTypeOf, test } from "vitest"; import * as z from "zod/v4"; test("preprocess", () => { const schema = z.preprocess((data) => [data], z.string().array()); const value = schema.parse("asdf"); expect(value).toEqual(["asdf"]); expectTypeOf<(typeof schema)["_input"]>().toEqualTypeOf<unknown>(); }); test("async preprocess", async () => { const schema = z.preprocess(async (data) => { return [data]; }, z.string().array()); const value = await schema.safeParseAsync("asdf"); expect(value.data).toEqual(["asdf"]); expect(value).toMatchInlineSnapshot(` { "data": [ "asdf", ], "success": true, } `); }); test("ctx.addIssue accepts string", () => { const schema = z.preprocess((_, ctx) => { ctx.addIssue("bad stuff"); }, z.string()); const result = schema.safeParse("asdf"); expect(result.error!.issues).toHaveLength(1); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "message": "bad stuff", "code": "custom", "path": [] } ]], "success": false, } `); }); test("preprocess ctx.addIssue with parse", () => { const a = z.preprocess((data, ctx) => { ctx.addIssue({ input: data, code: "custom", message: `${data} is not one of our allowed strings`, }); return data; }, z.string()); const result = a.safeParse("asdf"); // expect(result.error!.toJSON()).toContain("not one of our allowed strings"); expect(result.error!.issues).toHaveLength(1); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "code": "custom", "message": "asdf is not one of our allowed strings", "path": [] } ]], "success": false, } `); }); test("preprocess ctx.addIssue fatal by default", () => { const schema = z.preprocess((data, ctx) => { ctx.addIssue({ code: "custom", message: `custom error`, }); return data; }, z.string()); const result = schema.safeParse(1234); expect(result.error!.issues).toHaveLength(1); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "code": "custom", "message": "custom error", "path": [] } ]], "success": false, } `); }); test("preprocess ctx.addIssue fatal true", () => { const schema = z.preprocess((data, ctx) => { ctx.addIssue({ input: data, code: "custom", origin: "custom", message: `custom error`, fatal: true, }); return data; }, z.string()); const result = schema.safeParse(1234); expect(result.error!.issues).toHaveLength(1); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "code": "custom", "origin": "custom", "message": "custom error", "fatal": true, "path": [] } ]], "success": false, } `); }); test("async preprocess ctx.addIssue with parseAsync", async () => { const schema = z.preprocess(async (data, ctx) => { ctx.addIssue({ input: data, code: "custom", message: `${data} is not one of our allowed strings`, }); return data; }, z.string()); const result = await schema.safeParseAsync("asdf"); expect(result.error!.issues).toHaveLength(1); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "code": "custom", "message": "asdf is not one of our allowed strings", "path": [] } ]], "success": false, } `); }); test("z.NEVER in preprocess", () => { const foo = z.preprocess((val, ctx) => { if (!val) { ctx.addIssue({ input: val, code: "custom", message: "bad" }); return z.NEVER; } return val; }, z.number()); type foo = z.infer<typeof foo>; expectTypeOf<foo>().toEqualTypeOf<number>(); const result = foo.safeParse(undefined); expect(result.error!.issues).toHaveLength(1); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "code": "custom", "message": "bad", "path": [] } ]], "success": false, } `); }); test("preprocess as the second property of object", () => { const schema = z.object({ nonEmptyStr: z.string().min(1), positiveNum: z.preprocess((v) => Number(v), z.number().positive()), }); const result = schema.safeParse({ nonEmptyStr: "", positiveNum: "", }); expect(result.error!.issues).toHaveLength(2); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "origin": "string", "code": "too_small", "minimum": 1, "inclusive": true, "path": [ "nonEmptyStr" ], "message": "Too small: expected string to have >=1 characters" }, { "origin": "number", "code": "too_small", "minimum": 0, "inclusive": false, "path": [ "positiveNum" ], "message": "Too small: expected number to be >0" } ]], "success": false, } `); }); test("preprocess validates with sibling errors", () => { const schema = z.object({ missing: z.string().refine(() => false), preprocess: z.preprocess((data: any) => data?.trim(), z.string().regex(/ asdf/)), }); const result = schema.safeParse({ preprocess: " asdf" }); expect(result.error!.issues).toHaveLength(2); expect(result).toMatchInlineSnapshot(` { "error": [ZodError: [ { "expected": "string", "code": "invalid_type", "path": [ "missing" ], "message": "Invalid input: expected string, received undefined" }, { "origin": "string", "code": "invalid_format", "format": "regex", "pattern": "/ asdf/", "path": [ "preprocess" ], "message": "Invalid string: must match pattern / asdf/" } ]], "success": false, } `); }); test("perform transform with non-fatal issues", () => { const A = z .string() .refine((_) => false) .min(4) .transform((val) => val.length) .pipe(z.number()) .refine((_) => false); expect(A.safeParse("asdfasdf").error!.issues).toHaveLength(1); expect(A.safeParse("asdfasdf").error).toMatchInlineSnapshot(` [ZodError: [ { "code": "custom", "path": [], "message": "Invalid input" } ]] `); }); test("preprocess accepts absent object keys (4.3 parity)", () => { const schema = z.object({ a: z.preprocess((v) => v ?? "X", z.string()) }); expect(schema.parse({})).toEqual({ a: "X" }); expect(schema.parse({ a: "hi" })).toEqual({ a: "hi" }); expect(schema.parse({ a: undefined })).toEqual({ a: "X" }); // Outer optional clobbers preprocess output on undefined input expect( z .preprocess((v) => v ?? "X", z.string()) .optional() .parse(undefined) ).toBeUndefined(); expect( z .preprocess((v) => v ?? "X", z.string()) .optional() .parse("hi") ).toBe("hi"); expect(z.object({ a: z.preprocess((v) => v ?? "X", z.string()).optional() }).parse({})).toEqual({}); // Top-level direct call unchanged expect(z.preprocess((v) => v ?? "X", z.string()).parse(undefined)).toBe("X"); }); // https://github.com/colinhacks/zod/issues/5917 test("optional propagates through preprocess inside object", () => { const outer = z.object({ x: z.preprocess((v) => v, z.number()).optional() }); const inner = z.object({ x: z.preprocess((v) => v, z.number().optional()) }); expect(outer.safeParse({}).success).toBe(true); expect(inner.safeParse({}).success).toBe(true); expect(outer.safeParse({ x: 1 })).toEqual({ success: true, data: { x: 1 } }); expect(inner.safeParse({ x: 1 })).toEqual({ success: true, data: { x: 1 } }); expect(inner._zod.def.shape.x._zod.optin).toBe("optional"); expect(inner._zod.def.shape.x._zod.optout).toBe("optional"); }); test("preprocess is a structural subtype of ZodPipe", () => { const schema = z.preprocess((v) => v, z.string()); expect(schema).toBeInstanceOf(z.ZodPipe); expect(schema).toBeInstanceOf(z.ZodPreprocess); expect(schema._zod.def.type).toBe("pipe"); }); test("preprocess does not propagate values/propValues from inner schema", () => { const inner = z.preprocess((v) => v, z.literal("test")); expect(inner._zod.values).toBeUndefined(); expect(inner._zod.propValues).toBeUndefined(); }); test("preprocess as discriminator throws at construction (no propValues to inherit)", () => { const schema = z.discriminatedUnion("kind", [ z.object({ kind: z.preprocess((v: any) => String(v).toUpperCase(), z.literal("A")), a: z.string() }), z.object({ kind: z.preprocess((v: any) => String(v).toUpperCase(), z.literal("B")), b: z.number() }), ]); expect(() => schema.parse({ kind: "a", a: "x" })).toThrow(/Invalid discriminated union option/); }); test("preprocess as record key does not restrict accepted keys", () => { const schema = z.record( z.preprocess((v: any) => String(v).toLowerCase(), z.enum(["a", "b"])), z.string() ); expect(schema.safeParse({ A: "x", B: "y" })).toEqual({ success: true, data: { a: "x", b: "y" } }); });