feat(vitest): add return type and promisable mockFactory

packages/vitest/src/types/mocker.ts

@@ -1,6 +1,8 @@
+type Promisable<T> = T | Promise<T>
+
 export type MockFactoryWithHelper<M = unknown> = (
-  importOriginal: <T extends M>() => Promise<T>
-) => any
+  importOriginal: <T extends M = M>() => Promise<T>
+) => Promisable<Partial<M>>
 export type MockFactory = () => any
 
 export type MockMap = Map<string, Record<string, string | null | MockFactory>>

test/config/test/workers-option.test.ts

@@ -4,6 +4,7 @@ import { getWorkersCountByPercentage } from 'vitest/src/utils/workers.js'
 import * as testUtils from '../../test-utils'
 
 vi.mock(import('node:os'), async importOriginal => ({
+  ...(await importOriginal()),
   default: {
     ...(await importOriginal()).default,
     availableParallelism: () => 10,

test/core/test/mocking/factory.test.ts

@@ -54,7 +54,7 @@ vi.mock('../../src/mocks/log.ts', async () => {
     },
   }
 })
-
+// @ts-expect-error null is not allowed to mocked implementation
 vi.mock('../../src/mocks/default.ts', () => null)
 
 describe('mocking with factory', () => {