Schema.union: Better tests + Test for .optional()

numfin · Apr 29, 2023 · 37c3687 · 37c3687
1 parent a79d75c
commit 37c3687
Showing 1 changed file with 15 additions and 4 deletions.
diff --git a/src/schema/api/union.test.ts b/src/schema/api/union.test.ts
@@ -114,11 +114,22 @@ test("matchSome union", (t) => {
 
 test("Union inside dict", (t) => {
   const schema = SchemaDict({
-    a: SchemaUnion({
-      v: SchemaStr<string>(),
-      z: SchemaNum<number>(),
+    field: SchemaUnion({
+      str: SchemaStr<string>(),
+      num: SchemaNum<number>(),
     }),
   });
 
-  t.true(schema.parse({ a: "" }).unwrap().a.is("v"));
+  t.true(schema.parse({ field: "" }).unwrap().field.is("str"));
+});
+
+test("Optional union", (t) => {
+  const schema = SchemaDict({
+    inner: SchemaUnion({
+      str: SchemaStr<string>(),
+      num: SchemaNum<number>(),
+    }).optional(),
+  });
+
+  t.true(schema.parse({ inner: 3 }).unwrap().inner.unwrap().is("num"));
 });