Fix firstparty tests

colinhacks · Mar 14, 2024 · 0575b65 · 0575b65
1 parent 0f3811f
commit 0575b65
Show file tree

Hide file tree

Showing 5 changed files with 35 additions and 4 deletions.
diff --git a/deno/lib/__tests__/firstparty.test.ts b/deno/lib/__tests__/firstparty.test.ts
@@ -78,6 +78,10 @@ test("first party switch", () => {
       break;
     case z.ZodFirstPartyTypeKind.ZodPipeline:
       break;
+    case z.ZodFirstPartyTypeKind.ZodSymbol:
+      break;
+    case z.ZodFirstPartyTypeKind.ZodReadonly:
+      break;
     default:
       util.assertNever(def);
   }

diff --git a/deno/lib/__tests__/firstpartyschematypes.test.ts b/deno/lib/__tests__/firstpartyschematypes.test.ts
@@ -0,0 +1,21 @@
+// @ts-ignore TS6133
+import { expect } from "https://deno.land/x/expect@v0.2.6/mod.ts";
+const test = Deno.test;
+
+import { 
+  ZodFirstPartySchemaTypes,
+  ZodFirstPartyTypeKind,
+} from "../index.ts";
+import { util } from "../helpers/util.ts";
+
+test("Identify missing [ZodFirstPartySchemaTypes]", () => {
+  type ZodFirstPartySchemaForType<T extends ZodFirstPartyTypeKind> = ZodFirstPartySchemaTypes extends infer Schema
+    ? Schema extends { _def: { typeName: T } }
+      ? Schema
+      : never
+    : never;
+  type ZodMappedTypes = { [key in ZodFirstPartyTypeKind]: ZodFirstPartySchemaForType<key> };
+  type ZodFirstPartySchemaTypesMissingFromUnion = keyof { [key in keyof ZodMappedTypes as ZodMappedTypes[key] extends { _def: never } ? key : never]: unknown };
+
+  util.assertEqual<ZodFirstPartySchemaTypesMissingFromUnion, never>(true);
+});
diff --git a/deno/lib/types.ts b/deno/lib/types.ts
@@ -4978,7 +4978,9 @@ export type ZodFirstPartySchemaTypes =
   | ZodCatch<any>
   | ZodPromise<any>
   | ZodBranded<any, any>
-  | ZodPipeline<any, any>;
+  | ZodPipeline<any, any>
+  | ZodReadonly<any>
+  | ZodSymbol;
 
 // requires TS 4.4+
 abstract class Class {

diff --git a/src/__tests__/firstparty.test.ts b/src/__tests__/firstparty.test.ts
@@ -77,6 +77,10 @@ test("first party switch", () => {
       break;
     case z.ZodFirstPartyTypeKind.ZodPipeline:
       break;
+    case z.ZodFirstPartyTypeKind.ZodSymbol:
+      break;
+    case z.ZodFirstPartyTypeKind.ZodReadonly:
+      break;
     default:
       util.assertNever(def);
   }

diff --git a/src/__tests__/firstpartyschematypes.test.ts b/src/__tests__/firstpartyschematypes.test.ts
@@ -1,11 +1,11 @@
 // @ts-ignore TS6133
 import { test } from "@jest/globals";
-import { util } from "../helpers/util";
 
 import { 
   ZodFirstPartySchemaTypes,
   ZodFirstPartyTypeKind,
 } from "..";
+import { util } from "../helpers/util";
 
 test("Identify missing [ZodFirstPartySchemaTypes]", () => {
   type ZodFirstPartySchemaForType<T extends ZodFirstPartyTypeKind> = ZodFirstPartySchemaTypes extends infer Schema
@@ -16,5 +16,5 @@ test("Identify missing [ZodFirstPartySchemaTypes]", () => {
   type ZodMappedTypes = { [key in ZodFirstPartyTypeKind]: ZodFirstPartySchemaForType<key> };
   type ZodFirstPartySchemaTypesMissingFromUnion = keyof { [key in keyof ZodMappedTypes as ZodMappedTypes[key] extends { _def: never } ? key : never]: unknown };
 
-  util.assertNever({} as ZodFirstPartySchemaTypesMissingFromUnion);
-});
+  util.assertEqual<ZodFirstPartySchemaTypesMissingFromUnion, never>(true);
+});