UNPKG

zod

Version:

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

185 lines (170 loc) • 8.95 kB
import { expect, expectTypeOf, test } from "vitest"; import * as z from "zod/v4"; enum testEnum { A = 0, B = 1, } test("flat inference", () => { const readonlyString = z.string().readonly(); const readonlyNumber = z.number().readonly(); const readonlyNaN = z.nan().readonly(); const readonlyBigInt = z.bigint().readonly(); const readonlyBoolean = z.boolean().readonly(); const readonlyDate = z.date().readonly(); const readonlyUndefined = z.undefined().readonly(); const readonlyNull = z.null().readonly(); const readonlyAny = z.any().readonly(); const readonlyUnknown = z.unknown().readonly(); const readonlyVoid = z.void().readonly(); const readonlyStringArray = z.array(z.string()).readonly(); const readonlyTuple = z.tuple([z.string(), z.number()]).readonly(); const readonlyMap = z.map(z.string(), z.date()).readonly(); const readonlySet = z.set(z.string()).readonly(); const readonlyStringRecord = z.record(z.string(), z.string()).readonly(); const readonlyNumberRecord = z.record(z.string(), z.number()).readonly(); const readonlyObject = z.object({ a: z.string(), 1: z.number() }).readonly(); const readonlyEnum = z.nativeEnum(testEnum).readonly(); const readonlyPromise = z.promise(z.string()).readonly(); expectTypeOf<typeof readonlyString._output>().toEqualTypeOf<string>(); expectTypeOf<typeof readonlyNumber._output>().toEqualTypeOf<number>(); expectTypeOf<typeof readonlyNaN._output>().toEqualTypeOf<number>(); expectTypeOf<typeof readonlyBigInt._output>().toEqualTypeOf<bigint>(); expectTypeOf<typeof readonlyBoolean._output>().toEqualTypeOf<boolean>(); expectTypeOf<typeof readonlyDate._output>().toEqualTypeOf<Date>(); expectTypeOf<typeof readonlyUndefined._output>().toEqualTypeOf<undefined>(); expectTypeOf<typeof readonlyNull._output>().toEqualTypeOf<null>(); expectTypeOf<typeof readonlyAny._output>().toEqualTypeOf<any>(); expectTypeOf<typeof readonlyUnknown._output>().toEqualTypeOf<Readonly<unknown>>(); expectTypeOf<typeof readonlyVoid._output>().toEqualTypeOf<void>(); expectTypeOf<typeof readonlyStringArray._output>().toEqualTypeOf<readonly string[]>(); expectTypeOf<typeof readonlyTuple._output>().toEqualTypeOf<readonly [string, number]>(); expectTypeOf<typeof readonlyMap._output>().toEqualTypeOf<ReadonlyMap<string, Date>>(); expectTypeOf<typeof readonlySet._output>().toEqualTypeOf<ReadonlySet<string>>(); expectTypeOf<typeof readonlyStringRecord._output>().toEqualTypeOf<Readonly<Record<string, string>>>(); expectTypeOf<typeof readonlyNumberRecord._output>().toEqualTypeOf<Readonly<Record<string, number>>>(); expectTypeOf<typeof readonlyObject._output>().toEqualTypeOf<{ readonly a: string; readonly 1: number }>(); expectTypeOf<typeof readonlyEnum._output>().toEqualTypeOf<Readonly<testEnum>>(); expectTypeOf<typeof readonlyPromise._output>().toEqualTypeOf<Promise<string>>(); }); test("object freezing", async () => { expect(Object.isFrozen(z.array(z.string()).readonly().parse(["a"]))).toBe(true); expect(Object.isFrozen(z.tuple([z.string(), z.number()]).readonly().parse(["a", 1]))).toBe(true); expect( Object.isFrozen( z .map(z.string(), z.date()) .readonly() .parse(new Map([["a", new Date()]])) ) ).toBe(true); expect(Object.isFrozen(z.record(z.string(), z.string()).readonly().parse({ a: "b" }))).toBe(true); expect(Object.isFrozen(z.record(z.string(), z.number()).readonly().parse({ a: 1 }))).toBe(true); expect(Object.isFrozen(z.object({ a: z.string(), 1: z.number() }).readonly().parse({ a: "b", 1: 2 }))).toBe(true); expect( Object.isFrozen( await z .set(z.promise(z.string())) .readonly() .parseAsync(new Set([Promise.resolve("a")])) ) ).toBe(true); expect(Object.isFrozen(await z.promise(z.string()).readonly().parseAsync(Promise.resolve("a")))).toBe(true); }); test("async object freezing", async () => { expect(Object.isFrozen(await z.array(z.string()).readonly().parseAsync(["a"]))).toBe(true); expect(Object.isFrozen(await z.tuple([z.string(), z.number()]).readonly().parseAsync(["a", 1]))).toBe(true); expect( Object.isFrozen( await z .map(z.string(), z.date()) .readonly() .parseAsync(new Map([["a", new Date()]])) ) ).toBe(true); expect( Object.isFrozen( await z .set(z.promise(z.string())) .readonly() .parseAsync(new Set([Promise.resolve("a")])) ) ).toBe(true); expect(Object.isFrozen(await z.record(z.string(), z.string()).readonly().parseAsync({ a: "b" }))).toBe(true); expect(Object.isFrozen(await z.record(z.string(), z.number()).readonly().parseAsync({ a: 1 }))).toBe(true); expect( Object.isFrozen(await z.object({ a: z.string(), 1: z.number() }).readonly().parseAsync({ a: "b", 1: 2 })) ).toBe(true); expect(Object.isFrozen(await z.promise(z.string()).readonly().parseAsync(Promise.resolve("a")))).toBe(true); }); test("readonly inference", () => { const readonlyStringArray = z.string().array().readonly(); const readonlyStringTuple = z.tuple([z.string()]).readonly(); const deepReadonly = z.object({ a: z.string() }).readonly(); type readonlyStringArray = z.infer<typeof readonlyStringArray>; type readonlyStringTuple = z.infer<typeof readonlyStringTuple>; type deepReadonly = z.infer<typeof deepReadonly>; expectTypeOf<readonlyStringArray>().toEqualTypeOf<readonly string[]>(); expectTypeOf<readonlyStringTuple>().toEqualTypeOf<readonly [string]>(); expectTypeOf<deepReadonly>().toEqualTypeOf<{ readonly a: string }>(); }); test("readonly parse", () => { const schema = z.array(z.string()).readonly(); const readonlyArray = ["a", "b", "c"] as const; const mutableArray = ["a", "b", "c"]; const result1 = schema.parse(readonlyArray); const result2 = schema.parse(mutableArray); expect(result1).toEqual(readonlyArray); expect(result2).toEqual(mutableArray); }); test("readonly parse with tuples", () => { const schema = z.tuple([z.string(), z.number()]).readonly(); schema.parse(["a", 1]); }); test("readonly and the get method", () => { const readonlyString = z.string().readonly(); const readonlyNumber1 = z.number().readonly(); const readonlyNumber2 = z.number().readonly(); const readonlyBigInt = z.bigint().readonly(); const readonlyBoolean = z.boolean().readonly(); const readonlyDate = z.date().readonly(); const readonlyUndefined = z.undefined().readonly(); const readonlyNull = z.null().readonly(); const readonlyAny = z.any().readonly(); const readonlyUnknown = z.unknown().readonly(); const readonlyVoid = z.void().readonly(); // const readonlyFunction = z.function(z.tuple([z.string(), z.number()]), z.unknown()).readonly(); const readonlyStringArray = z.string().array().readonly(); const readonlyTuple = z.tuple([z.string(), z.number()]).readonly(); expectTypeOf<z.infer<typeof readonlyString>>().toEqualTypeOf<string>(); expectTypeOf<z.infer<typeof readonlyNumber1>>().toEqualTypeOf<number>(); expectTypeOf<z.infer<typeof readonlyNumber2>>().toEqualTypeOf<number>(); expectTypeOf<z.infer<typeof readonlyBigInt>>().toEqualTypeOf<bigint>(); expectTypeOf<z.infer<typeof readonlyBoolean>>().toEqualTypeOf<boolean>(); expectTypeOf<z.infer<typeof readonlyDate>>().toEqualTypeOf<Date>(); expectTypeOf<z.infer<typeof readonlyUndefined>>().toEqualTypeOf<undefined>(); expectTypeOf<z.infer<typeof readonlyNull>>().toEqualTypeOf<null>(); expectTypeOf<z.infer<typeof readonlyAny>>().toEqualTypeOf<any>(); expectTypeOf<z.infer<typeof readonlyUnknown>>().toEqualTypeOf<Readonly<unknown>>(); expectTypeOf<z.infer<typeof readonlyVoid>>().toEqualTypeOf<void>(); // expectTypeOf<z.infer<typeof readonlyFunction>>().toEqualTypeOf< // (args_0: string, args_1: number, ...args_2: unknown[]) => unknown // >(); expectTypeOf<z.infer<typeof readonlyStringArray>>().toEqualTypeOf<readonly string[]>(); expectTypeOf<z.infer<typeof readonlyTuple>>().toEqualTypeOf<readonly [string, number]>(); expect(readonlyString.parse("asdf")).toEqual("asdf"); expect(readonlyNumber1.parse(1234)).toEqual(1234); expect(readonlyNumber2.parse(1234)).toEqual(1234); const bigIntVal = BigInt(1); expect(readonlyBigInt.parse(bigIntVal)).toEqual(bigIntVal); expect(readonlyBoolean.parse(true)).toEqual(true); const dateVal = new Date(); expect(readonlyDate.parse(dateVal)).toEqual(dateVal); expect(readonlyUndefined.parse(undefined)).toEqual(undefined); expect(readonlyNull.parse(null)).toEqual(null); expect(readonlyAny.parse("whatever")).toEqual("whatever"); expect(readonlyUnknown.parse("whatever")).toEqual("whatever"); expect(readonlyVoid.parse(undefined)).toEqual(undefined); // expect(readonlyFunction.parse(() => void 0)).toEqual(() => void 0); expect(readonlyStringArray.parse(["asdf"])).toEqual(["asdf"]); expect(readonlyTuple.parse(["asdf", 1234])).toEqual(["asdf", 1234]); });