tests: other zod validators

colinhacks · Nov 23, 2022 · 0503ebf · 0503ebf
1 parent dd7162f
commit 0503ebf
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/deno/lib/__tests__/discriminatedUnions.test.ts b/deno/lib/__tests__/discriminatedUnions.test.ts
@@ -64,6 +64,27 @@ test("valid - discriminator value of various primitive types", () => {
   });
 });
 
+test("valid - various zod validator discriminators", () => {
+  const schema = z.discriminatedUnion("type", [
+    z.object({ type: z.undefined(), val: z.literal(1) }),
+    z.object({ type: z.null(), val: z.literal(2) }),
+    z.object({ type: z.enum(["a", "b", "c"]), val: z.literal(3) }),
+  ]);
+
+  expect(schema.parse({ type: undefined, val: 1 })).toEqual({
+    type: undefined,
+    val: 1,
+  });
+  expect(schema.parse({ type: null, val: 2 })).toEqual({
+    type: null,
+    val: 2,
+  });
+  expect(schema.parse({ type: "c", val: 3 })).toEqual({
+    type: "c",
+    val: 3,
+  });
+});
+
 test("valid - wrapped optional discriminator value ", () => {
   const schema = z.discriminatedUnion("type", [
     z.object({ type: z.literal("1").optional(), val: z.literal(1) }),

diff --git a/src/__tests__/discriminatedUnions.test.ts b/src/__tests__/discriminatedUnions.test.ts
@@ -63,6 +63,27 @@ test("valid - discriminator value of various primitive types", () => {
   });
 });
 
+test("valid - various zod validator discriminators", () => {
+  const schema = z.discriminatedUnion("type", [
+    z.object({ type: z.undefined(), val: z.literal(1) }),
+    z.object({ type: z.null(), val: z.literal(2) }),
+    z.object({ type: z.enum(["a", "b", "c"]), val: z.literal(3) }),
+  ]);
+
+  expect(schema.parse({ type: undefined, val: 1 })).toEqual({
+    type: undefined,
+    val: 1,
+  });
+  expect(schema.parse({ type: null, val: 2 })).toEqual({
+    type: null,
+    val: 2,
+  });
+  expect(schema.parse({ type: "c", val: 3 })).toEqual({
+    type: "c",
+    val: 3,
+  });
+});
+
 test("valid - wrapped optional discriminator value ", () => {
   const schema = z.discriminatedUnion("type", [
     z.object({ type: z.literal("1").optional(), val: z.literal(1) }),