如何存根EntityManager和连接在TypeORM与开玩笑

Question

我在NestJS上用Typescript开发了一个应用程序，使用TypeORM和用Jest编写的单元测试。我有一个使用如下事务的函数：

async createMany(users: User[]) {
  await this.connection.transaction(async manager => {
    await manager.save(users[0]);
    await manager.save(users[1]);
  });
}

这是NestJS文档中的一个示例。我通过this以大致相同的方式完成。联系事务，但业务逻辑不同。

问题是我想做一个单元测试来测试这个服务功能。所以我需要以某种方式模拟这两个。连接及其管理器。或者至少是经理。我不知道如何用笑话来表达。我无法创建没有连接的管理器。我无法在没有管理器返回的情况下创建模拟连接。

在NestJS中，同时使用TypeORM和Jest是标准的。必须有一种使用事务编写单元测试的方法。但我不知道怎么做。

注意，我询问的是单元测试模拟ORM。不是直接使用测试db实例的集成测试。

Answer 1

我相信你也可以：

import { getConnectionToken } from '@nestjs/typeorm';

{
  provide: getConnectionToken(),
  useClass: Connection,
},

Answer 2

您需要使用模拟的连接创建一个测试模块。像这样的事情：

import { Test } from "@nestjs/testing";

describe("UsersService", () => {
    let usersService;
    let connection;
    const mockConnection = () => ({
        transaction: jest.fn()
    });

    beforeEach(async () => {
        const module = await Test.createTestingModule({
            providers: [
                UsersService,
                {
                    provide: Connection,
                    useFactory: mockConnection
                }
            ],
        }).compile();

        usersService = await module.get<UsersService>(UsersService); 
        connection = await modle.get<Connection>(Connection);
    });

    describe("some tests", () => {
        it("should test something", async () => {
            const someMockedUsers = [/* some users */];
            const mockedManager = {
                save: jest.fn()
            }
            connection.transaction.mockImplementation((cb) => {
                cb(mockedManager);
            });

            await userService.createMany(someMockedUsers);

            expect(connection.transaction).toHaveBeenCalled();
            expect(mockedManager.save).toHaveBeenCalledTimes(2);
            // ...

        });
    });


});