如何在 TypeORM 中模拟 EntityManager？

我有以下代码从我的 NestJS 服务中模拟实体管理器，它在某种程度上有效，但问题在于指定什么findOne应该返回不同的实体。

    {
      provide: getEntityManagerToken(),
      useFactory: (connection: Connection) => {
        const entityManager = {
          getRepository: jest.fn(() => {
            return {
              findOne: jest.fn().mockResolvedValue({}),
              save: jest.fn().mockResolvedValue(carData),
              createQueryBuilder: jest.fn(() => ({
                select: jest.fn().mockReturnThis(),
                from: jest.fn().mockReturnThis(),
                leftJoinAndSelect: jest.fn().mockReturnThis(),
                where: jest.fn().mockReturnThis(),
                addGroupBy: jest.fn().mockReturnThis(),
                groupBy: jest.fn().mockReturnThis(),
                getOne: jest.fn().mockResolvedValue(''),
              })),
            };
          }),
        };
        return entityManager;
      },
    },

我的服务如下代码所示：

export class CarService {
  constructor(
    @InjectEntityManager()
    private readonly entityManager: EntityManager,
  ) {
    this.carRepository = entityManager.getRepository(Car);
    this.modelRepository = entityManager.getRepository(Model);
   }
    
      const car = await this.carRepository.findOne(data.id);

      const model = await this.modelRepository.findOne(data.modelId);
 

我如何模拟 findOne 结果this.carRepository与 findOne 结果不同this.modelRepository考虑到它们都将在同一个服务中被调用。

我们有可能使用实体管理器API https://github.com/typeorm/typeorm/blob/master/docs/entity-manager-api.md创建具有多个存储库的事务，如果这些操作中的任何一个失败，这些存储库可以自动执行回滚，并且在测试中，我们可以使用单例模式来定义此实体的模拟，该实体允许我们返回这些存储库的相同实例来测试所有这些读写操作均已按预期执行。例如

import { createMock } from '@golevelup/ts-jest';
import { Test, TestingModule } from '@nestjs/testing';
import { EntityManager } from 'typeorm';

describe('MyService', () => {
  let service: MyService;
  let entityManager: EntityManager;
  beforeEach(async () => {
    const module: TestingModule = await Test.createTestingModule({
      providers: [
        MyService,
        { provide: EntityManager, useValue: createMock<EntityManager>() },
      ],
    }).compile();

    service = module.get(CatalogsService);
    entityManager = module.get(EntityManager);
  });
  
  it('should call a transaction correctly', async () => {
    const mockEntityManager = {
      save: jest.fn(),
      getCustomRepository: jest.fn((fn) => mockEntityManager[fn] || (mockEntityManager[fn] = createMock<typeof fn>())),
    };
    const spyTransaction = (entityManager.transaction as jest.Mock).mockImplementation((cb) => cb(mockEntityManager));
    const firstRepo: FirstRepository = mockEntityManager.getCustomRepository(SubCatalogRepository);
    const secondRepo: SecondRepository = mockEntityManager.getCustomRepository(SecondRepository);
    await service.saveSomething('MOCK DATA');

    expect(spyTransaction).toHaveBeenCalled();
    expect(firstRepo.save).toHaveBeenCalled();
    expect(secondRepo.save).toHaveBeenCalled();
    expect(mockEntityManager.save).toHaveBeenCalled();
  });
});

正如我们所看到的，使用以下方法来模拟这些外部服务（例如 TypeORM EntityManager 等）非常容易创建模拟 https://github.com/golevelup/nestjs/tree/master/packages/testing#motivation函数，它自动注入 Jest 模拟函数来替换我们的代码所依赖的外部实现，这在单元测试中应该不重要。

快乐编码！