import { expect, expectTypeOf, test } from "vitest"; import * as z from "zod/v4"; function nest(schema: TData) { return z.object({ nested: schema, }); } test("generics", () => { const a = nest(z.object({ a: z.string() })); type a = z.infer; expectTypeOf().toEqualTypeOf<{ nested: { a: string } }>(); const b = nest(z.object({ a: z.string().optional() })); type b = z.infer; expectTypeOf().toEqualTypeOf<{ nested: { a?: string | undefined } }>(); }); test("generics with optional", () => { async function stripOuter(schema: TData, data: unknown) { return z .object({ nested: schema.optional(), }) .transform((data) => { return data.nested; }) .parse({ nested: data }); } const result = stripOuter(z.object({ a: z.string() }), { a: "asdf" }); expectTypeOf().toEqualTypeOf>(); }); // test("assignability", () => { // const createSchemaAndParse = (key: K, valueSchema: VS, data: unknown) => { // const schema = z.object({ // [key]: valueSchema, // }); // // return { [key]: valueSchema }; // const parsed = schema.parse(data); // return parsed; // // const inferred: z.infer> = parsed; // // return inferred; // }; // const parsed = createSchemaAndParse("foo", z.string(), { foo: "" }); // expectTypeOf().toEqualTypeOf<{ foo: string }>(); // }); test("nested no undefined", () => { const inner = z.string().or(z.array(z.string())); const outer = z.object({ inner }); type outerSchema = z.infer; expectTypeOf().toEqualTypeOf<{ inner: string | string[] }>(); expect(outer.safeParse({ inner: undefined }).success).toEqual(false); }); test("generic on output type", () => { const createV4Schema = (opts: { schema: z.ZodType; }) => { return opts.schema; }; createV4Schema({ schema: z.object({ name: z.string(), }), })?._zod?.output?.name; });