如何在Vitest中移除unknown类型转换创建Repository部分Mock？

以下几种方法可以帮你在不实现Repository所有方法、不用<Repository><unknown>类型转换的前提下，完成Vitest测试中的Mock：

1. 用 Partial<Repository> 分层断言 #

Partial<T> 会将目标类型的所有属性转为可选，这样你只需要定义测试用到的方法，再分层断言为Repository即可：

const mockRepo = {
  repositoryMethod: vi.fn().mockReturnValue('测试返回值')
} as Partial<Repository> as Repository;

const factoryResult = myFactory(mockRepo);

这种方式比转unknown更直观，TypeScript会检查你定义的方法是否符合Repository的类型规范。

2. 利用 Vitest 自动生成Mock（适用于模块导出的Repository） #

如果Repository是从某个模块导出的类或对象，直接用vi.mock生成自动Mock，只覆盖需要的方法：

// 假设Repository来自 './repository' 模块
vi.mock('./repository', () => ({
  Repository: vi.fn().mockImplementation(() => ({
    repositoryMethod: vi.fn().mockReturnValue('测试返回值')
  }))
}));

// 测试用例中使用
const repoInstance = new Repository();
const factoryResult = myFactory(repoInstance);

Vitest会自动为未定义的方法生成空Mock（默认返回undefined），完全不用手动实现所有方法。

3. 用 Pick<Repository, 'repositoryMethod'> 提取必要类型 #

通过Pick直接提取Repository中需要用到的方法类型，再断言为完整的Repository类型：

type RequiredRepoMethods = Pick<Repository, 'repositoryMethod'>;

const mockRepo = {
  repositoryMethod: vi.fn().mockReturnValue('测试返回值')
} as RequiredRepoMethods as Repository;

这种方式明确指定了测试依赖的方法，可读性更强。

4. 直接断言并校验方法签名 #

如果需要确保Mock方法的参数、返回值严格匹配Repository的定义，可以直接断言并编写符合签名的方法：

const mockRepo = {
  repositoryMethod: vi.fn((input: Parameters<Repository['repositoryMethod']>[0]) => {
    // 按测试需求实现逻辑
    return '测试返回值';
  })
} as Repository;

TypeScript会自动校验你定义的repositoryMethod是否与原类型匹配，未定义的方法在测试中只要不被调用就不会引发问题。

内容的提问来源于stack exchange，提问作者Sławomir Ossowski