fix(types): fix PartialMock with async TReturns (#3462)

vitest-dev · May 29, 2023 · b664d42 · b664d42
1 parent cd5d58b
commit b664d42
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/packages/spy/src/index.ts b/packages/spy/src/index.ts
@@ -62,7 +62,7 @@ export interface Mock<TArgs extends any[] = any, TReturns = any> extends SpyInst
   new (...args: TArgs): TReturns
   (...args: TArgs): TReturns
 }
-export interface PartialMock<TArgs extends any[] = any, TReturns = any> extends SpyInstance<TArgs, Partial<TReturns>> {
+export interface PartialMock<TArgs extends any[] = any, TReturns = any> extends SpyInstance<TArgs, TReturns extends Promise<Awaited<TReturns>> ? Promise<Partial<Awaited<TReturns>>> : Partial<TReturns>> {
   new (...args: TArgs): TReturns
   (...args: TArgs): TReturns
 }

diff --git a/test/core/test/vi.spec.ts b/test/core/test/vi.spec.ts
@@ -65,6 +65,22 @@ describe('testing vi utils', () => {
     vi.mocked(mockFactory, { partial: true, deep: true }).mockReturnValue({
       baz: 'baz',
     })
+
+    type FooBarAsyncFactory = () => Promise<FooBar>
+
+    const mockFactoryAsync: FooBarAsyncFactory = vi.fn()
+
+    vi.mocked(mockFactoryAsync, { partial: true }).mockResolvedValue({
+      foo: vi.fn(),
+    })
+
+    vi.mocked(mockFactoryAsync, { partial: true, deep: false }).mockResolvedValue({
+      bar: vi.fn(),
+    })
+
+    vi.mocked(mockFactoryAsync, { partial: true, deep: true }).mockResolvedValue({
+      baz: 'baz',
+    })
   })
 
   test('can change config', () => {