Test against a bit bigger router

otahontas · Aug 27, 2023 · 09c91b1 · 09c91b1
1 parent 9f84d93
commit 09c91b1
Show file tree

Hide file tree

Showing 4 changed files with 64 additions and 23 deletions.
diff --git a/.changeset/odd-hornets-hear.md b/.changeset/odd-hornets-hear.md
@@ -0,0 +1,5 @@
+---
+"trpc-rtk-query": patch
+---
+
+Add tests for bigger router and queries after adding endpointoptions
diff --git a/test/create-endpoint-definitions.test-d.ts b/test/create-endpoint-definitions.test-d.ts
@@ -16,9 +16,7 @@ describe("create endpoint definitions", () => {
         "api",
         never
       >;
-      expectTypeOf<QueryKeys<Definitions>>().toEqualTypeOf<
-        "getUserById" | "listUsers" | "nested_Deep_GetVeryNestedMessage"
-      >();
+      expectTypeOf<QueryKeys<Definitions>>().not.toEqualTypeOf<never>();
     });
   });
 });
diff --git a/test/create-trpc-api.test-d.ts b/test/create-trpc-api.test-d.ts
@@ -8,6 +8,20 @@ import { describe, expectTypeOf, it } from "vitest";
 import { createApi, enhanceApi } from "../src";
 import { type AppRouter, testClientOptions } from "./fixtures";
 
+type QueryRoutes =
+  | "extraProcedure1"
+  | "extraProcedure2"
+  | "extraProcedure3"
+  | "extraProcedure4"
+  | "extraProcedure5"
+  | "extraProcedure6"
+  | "extraProcedure7"
+  | "extraProcedure8"
+  | "extraProcedure9"
+  | "extraProcedure10"
+  | "getUserById"
+  | "listUsers"
+  | "nested_Deep_GetVeryNestedMessage";
 // Tests each scenery with one query and one mutation
 // This can't be really parametrized since these are statically checked, so we need
 // to copypaste each test case.
@@ -71,11 +85,7 @@ describe("create-trpc-api", () => {
     // use prefetch
 
     expectTypeOf(usePrefetch).toBeFunction();
-    expectTypeOf(usePrefetch)
-      .parameter(0)
-      .toMatchTypeOf<
-        "getUserById" | "listUsers" | "nested_Deep_GetVeryNestedMessage"
-      >();
+    expectTypeOf(usePrefetch).parameter(0).toMatchTypeOf<QueryRoutes>();
   });
 
   it("allows creating api with get client while infering types from getClient func", () => {
@@ -137,11 +147,7 @@ describe("create-trpc-api", () => {
 
     // use prefetch
     expectTypeOf(usePrefetch).toBeFunction();
-    expectTypeOf(usePrefetch)
-      .parameter(0)
-      .toMatchTypeOf<
-        "getUserById" | "listUsers" | "nested_Deep_GetVeryNestedMessage"
-      >();
+    expectTypeOf(usePrefetch).parameter(0).toMatchTypeOf<QueryRoutes>();
   });
 
   it("allows injecting trpc api to existing api while infering types from client and api", () => {
@@ -218,11 +224,7 @@ describe("create-trpc-api", () => {
 
     // use prefetch should have types from both previous and trpc endpoints
     expectTypeOf(usePrefetch).toBeFunction();
-    expectTypeOf(usePrefetch)
-      .parameter(0)
-      .toMatchTypeOf<
-        "getResponse" | "getUserById" | "listUsers" | "nested_Deep_GetVeryNestedMessage"
-      >();
+    expectTypeOf(usePrefetch).parameter(0).toMatchTypeOf<"getResponse" | QueryRoutes>();
 
     // existing query through api & endpoints.getResponse
     const {
@@ -314,11 +316,7 @@ describe("create-trpc-api", () => {
 
     // use prefetch should have types from both previous and trpc endpoints
     expectTypeOf(usePrefetch).toBeFunction();
-    expectTypeOf(usePrefetch)
-      .parameter(0)
-      .toMatchTypeOf<
-        "getResponse" | "getUserById" | "listUsers" | "nested_Deep_GetVeryNestedMessage"
-      >();
+    expectTypeOf(usePrefetch).parameter(0).toMatchTypeOf<"getResponse" | QueryRoutes>();
 
     // existing query through api & endpoints.getResponse
     const {

diff --git a/test/fixtures.ts b/test/fixtures.ts
@@ -39,6 +39,15 @@ const veryDeepRouter = router({
   }),
 });
 
+const extraProcedureInputSchema = z.object({
+  bar: z.number(),
+  foo: z.string(),
+  nested: z.object({
+    bar: z.number(),
+    foo: z.string(),
+  }),
+});
+
 export const appRouter = router({
   createUser: procedure.input(z.string()).mutation(async (options) => {
     const { input } = options;
@@ -48,6 +57,37 @@ export const appRouter = router({
       name: input,
     };
   }),
+  // Some extra procedures to see if stuff fails with bigger routers
+  extraProcedure1: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure2: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure3: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure4: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure5: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure6: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure7: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure8: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure9: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
+  extraProcedure10: procedure
+    .input(extraProcedureInputSchema)
+    .query(async ({ input }) => input),
   getUserById: procedure.input(z.number()).query(async (options) => {
     const { input } = options;
     // Retrieve the user with the given ID