fix: Allow mocking procedures with no return value (#36)

* fix: Allow mocking procedures with no return value * fix: Improve types
maloguertin · Apr 22, 2024 · 6a9ebd8 · 6a9ebd8
1 parent b0812ae
commit 6a9ebd8
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/src/types.ts b/src/types.ts
@@ -16,7 +16,9 @@ export type ExtractInput<T extends ProcedureParams> = T extends ProcedureParams<
   : never
 
 export type ExtractOutput<T> = T extends Procedure<ProcedureType, infer ProcedureParams>
-  ? ProcedureParams['_output_out'] extends DefaultBodyType
+  ? ProcedureParams['_output_out'] extends void
+    ? void
+    : ProcedureParams['_output_out'] extends DefaultBodyType
     ? ProcedureParams['_output_out']
     : never
   : never

diff --git a/test/createTRPCMsw.test.ts b/test/createTRPCMsw.test.ts
@@ -16,6 +16,15 @@ describe('proxy returned by createMswTrpc', () => {
     >()
   })
 
+  it('should interpret procedure without return as void', () => {
+    mswTrpc.noReturn.mutation(input => {
+      return
+    })
+    expectTypeOf(mswTrpc.noReturn.mutation).toEqualTypeOf<
+      (handler: (input: void) => PromiseOrValue<void>) => RequestHandler
+    >
+  })
+
   describe('with merged routers', () => {
     it('should expose property query on properties that match TRPC query procedures', () => {
       expectTypeOf(nestedMswTrpc.users.userById.query).toEqualTypeOf<

diff --git a/test/setup.ts b/test/setup.ts
@@ -112,6 +112,7 @@ const appRouter = t.router({
 
       return input
     }),
+  noReturn: t.procedure.mutation(() => {}),
 })
 
 const appRouterWithSuperJson = tWithSuperJson.router({