123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199
import { expect } from "https://deno.land/x/expect@v0.2.6/mod.ts";const test = Deno.test;
import * as z from "../index.ts";
test("valid", () => {  expect(    z      .discriminatedUnion("type", [        z.object({ type: z.literal("a"), a: z.string() }),        z.object({ type: z.literal("b"), b: z.string() }),      ])      .parse({ type: "a", a: "abc" })  ).toEqual({ type: "a", a: "abc" });});
test("valid - discriminator value of various primitive types", () => {  const schema = z.discriminatedUnion("type", [    z.object({ type: z.literal("1"), val: z.literal(1) }),    z.object({ type: z.literal(1), val: z.literal(2) }),    z.object({ type: z.literal(BigInt(1)), val: z.literal(3) }),    z.object({ type: z.literal("true"), val: z.literal(4) }),    z.object({ type: z.literal(true), val: z.literal(5) }),    z.object({ type: z.literal("null"), val: z.literal(6) }),    z.object({ type: z.literal(null), val: z.literal(7) }),    z.object({ type: z.literal("undefined"), val: z.literal(8) }),    z.object({ type: z.literal(undefined), val: z.literal(9) }),  ]);
  expect(schema.parse({ type: "1", val: 1 })).toEqual({ type: "1", val: 1 });  expect(schema.parse({ type: 1, val: 2 })).toEqual({ type: 1, val: 2 });  expect(schema.parse({ type: BigInt(1), val: 3 })).toEqual({    type: BigInt(1),    val: 3,  });  expect(schema.parse({ type: "true", val: 4 })).toEqual({    type: "true",    val: 4,  });  expect(schema.parse({ type: true, val: 5 })).toEqual({    type: true,    val: 5,  });  expect(schema.parse({ type: "null", val: 6 })).toEqual({    type: "null",    val: 6,  });  expect(schema.parse({ type: null, val: 7 })).toEqual({    type: null,    val: 7,  });  expect(schema.parse({ type: "undefined", val: 8 })).toEqual({    type: "undefined",    val: 8,  });  expect(schema.parse({ type: undefined, val: 9 })).toEqual({    type: undefined,    val: 9,  });});
test("invalid - null", () => {  try {    z.discriminatedUnion("type", [      z.object({ type: z.literal("a"), a: z.string() }),      z.object({ type: z.literal("b"), b: z.string() }),    ]).parse(null);    throw new Error();  } catch (e: any) {    expect(JSON.parse(e.message)).toEqual([      {        code: z.ZodIssueCode.invalid_type,        expected: z.ZodParsedType.object,        message: "Expected object, received null",        received: z.ZodParsedType.null,        path: [],      },    ]);  }});
test("invalid discriminator value", () => {  try {    z.discriminatedUnion("type", [      z.object({ type: z.literal("a"), a: z.string() }),      z.object({ type: z.literal("b"), b: z.string() }),    ]).parse({ type: "x", a: "abc" });    throw new Error();  } catch (e: any) {    expect(JSON.parse(e.message)).toEqual([      {        code: z.ZodIssueCode.invalid_union_discriminator,        options: ["a", "b"],        message: "Invalid discriminator value. Expected 'a' | 'b'",        path: ["type"],      },    ]);  }});
test("valid discriminator value, invalid data", () => {  try {    z.discriminatedUnion("type", [      z.object({ type: z.literal("a"), a: z.string() }),      z.object({ type: z.literal("b"), b: z.string() }),    ]).parse({ type: "a", b: "abc" });    throw new Error();  } catch (e: any) {    expect(JSON.parse(e.message)).toEqual([      {        code: z.ZodIssueCode.invalid_type,        expected: z.ZodParsedType.string,        message: "Required",        path: ["a"],        received: z.ZodParsedType.undefined,      },    ]);  }});
test("wrong schema - missing discriminator", () => {  try {    z.discriminatedUnion("type", [      z.object({ type: z.literal("a"), a: z.string() }),      z.object({ b: z.string() }) as any,    ]);    throw new Error();  } catch (e: any) {    expect(e.message).toEqual(      "The discriminator value could not be extracted from all the provided schemas"    );  }});
test("wrong schema - duplicate discriminator values", () => {  try {    z.discriminatedUnion("type", [      z.object({ type: z.literal("a"), a: z.string() }),      z.object({ type: z.literal("a"), b: z.string() }),    ]);    throw new Error();  } catch (e: any) {    expect(e.message).toEqual(      "Some of the discriminator values are not unique"    );  }});
test("async - valid", async () => {  expect(    await z      .discriminatedUnion("type", [        z.object({          type: z.literal("a"),          a: z            .string()            .refine(async () => true)            .transform(async (val) => Number(val)),        }),        z.object({          type: z.literal("b"),          b: z.string(),        }),      ])      .parseAsync({ type: "a", a: "1" })  ).toEqual({ type: "a", a: 1 });});
test("async - invalid", async () => {  try {    await z      .discriminatedUnion("type", [        z.object({          type: z.literal("a"),          a: z            .string()            .refine(async () => true)            .transform(async (val) => val),        }),        z.object({          type: z.literal("b"),          b: z.string(),        }),      ])      .parseAsync({ type: "a", a: 1 });    throw new Error();  } catch (e: any) {    expect(JSON.parse(e.message)).toEqual([      {        code: "invalid_type",        expected: "string",        received: "number",        path: ["a"],        message: "Expected string, received number",      },    ]);  }});