128 lines
4.3 KiB
TypeScript
128 lines
4.3 KiB
TypeScript
import { expect, expectTypeOf, test } from "vitest";
|
|
import * as z from "zod/v4";
|
|
|
|
const fish = z.object({
|
|
name: z.string(),
|
|
age: z.number(),
|
|
nested: z.object({}),
|
|
});
|
|
|
|
test("pick type inference", () => {
|
|
const nameonlyFish = fish.pick({ name: true });
|
|
type nameonlyFish = z.infer<typeof nameonlyFish>;
|
|
expectTypeOf<nameonlyFish>().toEqualTypeOf<{ name: string }>();
|
|
});
|
|
|
|
test("pick parse - success", () => {
|
|
const nameonlyFish = fish.pick({ name: true });
|
|
nameonlyFish.parse({ name: "bob" });
|
|
|
|
// @ts-expect-error checking runtime picks `name` only.
|
|
const anotherNameonlyFish = fish.pick({ name: true, age: false });
|
|
anotherNameonlyFish.parse({ name: "bob" });
|
|
});
|
|
|
|
test("pick parse - fail", () => {
|
|
fish.pick({ name: true }).parse({ name: "12" } as any);
|
|
fish.pick({ name: true }).parse({ name: "bob", age: 12 } as any);
|
|
fish.pick({ age: true }).parse({ age: 12 } as any);
|
|
|
|
const nameonlyFish = fish.pick({ name: true }).strict();
|
|
const bad1 = () => nameonlyFish.parse({ name: 12 } as any);
|
|
const bad2 = () => nameonlyFish.parse({ name: "bob", age: 12 } as any);
|
|
const bad3 = () => nameonlyFish.parse({ age: 12 } as any);
|
|
|
|
// @ts-expect-error checking runtime picks `name` only.
|
|
const anotherNameonlyFish = fish.pick({ name: true, age: false }).strict();
|
|
const bad4 = () => anotherNameonlyFish.parse({ name: "bob", age: 12 } as any);
|
|
|
|
expect(bad1).toThrow();
|
|
expect(bad2).toThrow();
|
|
expect(bad3).toThrow();
|
|
expect(bad4).toThrow();
|
|
});
|
|
|
|
test("pick - remove optional", () => {
|
|
const schema = z.object({ a: z.string(), b: z.string().optional() });
|
|
expect("a" in schema._zod.def.shape).toEqual(true);
|
|
expect("b" in schema._zod.def.shape!).toEqual(true);
|
|
const picked = schema.pick({ a: true });
|
|
expect("a" in picked._zod.def.shape).toEqual(true);
|
|
expect("b" in picked._zod.def.shape!).toEqual(false);
|
|
});
|
|
|
|
test("omit type inference", () => {
|
|
const nonameFish = fish.omit({ name: true });
|
|
type nonameFish = z.infer<typeof nonameFish>;
|
|
|
|
expectTypeOf<nonameFish>().toEqualTypeOf<{ age: number; nested: Record<string, never> }>();
|
|
});
|
|
|
|
test("omit parse - success", () => {
|
|
const nonameFish = fish.omit({ name: true });
|
|
nonameFish.parse({ age: 12, nested: {} });
|
|
|
|
// @ts-expect-error checking runtime omits `name` only.
|
|
const anotherNonameFish = fish.omit({ name: true, age: false });
|
|
anotherNonameFish.parse({ age: 12, nested: {} });
|
|
});
|
|
|
|
test("omit parse - fail", () => {
|
|
const nonameFish = fish.omit({ name: true });
|
|
const bad1 = () => nonameFish.parse({ name: 12 } as any);
|
|
const bad2 = () => nonameFish.parse({ age: 12 } as any);
|
|
const bad3 = () => nonameFish.parse({} as any);
|
|
|
|
// @ts-expect-error checking runtime omits `name` only.
|
|
const anotherNonameFish = fish.omit({ name: true, age: false });
|
|
const bad4 = () => anotherNonameFish.parse({ nested: {} } as any);
|
|
|
|
expect(bad1).toThrow();
|
|
expect(bad2).toThrow();
|
|
expect(bad3).toThrow();
|
|
expect(bad4).toThrow();
|
|
});
|
|
|
|
test("omit - remove optional", () => {
|
|
const schema = z.object({ a: z.string(), b: z.string().optional() });
|
|
expect("a" in schema._zod.def.shape).toEqual(true);
|
|
const omitted = schema.omit({ a: true });
|
|
expect("a" in omitted._zod.def.shape).toEqual(false);
|
|
});
|
|
|
|
test("nonstrict inference", () => {
|
|
const laxfish = fish.pick({ name: true }).catchall(z.any());
|
|
type laxfish = z.infer<typeof laxfish>;
|
|
expectTypeOf<laxfish>().toEqualTypeOf<{ name: string; [k: string]: any }>();
|
|
});
|
|
|
|
test("nonstrict parsing - pass", () => {
|
|
const laxfish = fish.passthrough().pick({ name: true });
|
|
laxfish.parse({ name: "asdf", whatever: "asdf" });
|
|
laxfish.parse({ name: "asdf", age: 12, nested: {} });
|
|
});
|
|
|
|
test("nonstrict parsing - fail", () => {
|
|
const laxfish = fish.passthrough().pick({ name: true });
|
|
const bad = () => laxfish.parse({ whatever: "asdf" } as any);
|
|
expect(bad).toThrow();
|
|
});
|
|
|
|
test("pick/omit/required/partial - do not allow unknown keys", () => {
|
|
const schema = z.object({
|
|
name: z.string(),
|
|
age: z.number(),
|
|
});
|
|
|
|
expect(() => schema.pick({ name: true, asdf: true })).toThrow();
|
|
|
|
// @ts-expect-error
|
|
expect(() => schema.pick({ $unknown: true })).toThrow();
|
|
// @ts-expect-error
|
|
expect(() => schema.omit({ $unknown: true })).toThrow();
|
|
// @ts-expect-error
|
|
expect(() => schema.required({ $unknown: true })).toThrow();
|
|
// @ts-expect-error
|
|
expect(() => schema.partial({ $unknown: true })).toThrow();
|
|
});
|