NestJs 对 typeorm 分页进行单元测试时出现错误 queryBuilder.take(...).skip 不是函数

Question

我正在尝试模拟 createQueryBuilder 对分页部分进行单元测试，但出现以下错误

queryBuilder.take(...).skip 不是函数

我的 createQueryBuilder 模拟

    createQueryBuilder: jest.fn(() => ({
      delete: jest.fn().mockReturnThis(),
      innerJoinAndSelect: jest.fn().mockReturnThis(),
      innerJoin: jest.fn().mockReturnThis(),
      from: jest.fn().mockReturnThis(),
      where: jest.fn().mockReturnThis(),
      execute: jest.fn().mockReturnThis(),
      getOne: jest.fn().mockReturnThis(),
      orderBy : jest.fn().mockReturnThis(),
      take : skip => ({
       skip: 5
      }), 
  }))

查找全部的单元测试

it("Find All", async () => {
    const pageData: PaginationDto = {
      page: 1,
      limit: 10,
      sortBy: "id",
      sortOrder: -1,
      relatinalData: relatinalData.no

    };
    expect(await service.getGroups(pageData)).toEqual({
      docs: [{
        groupName: "group123",
        description: "Group Description",
        roleId: 1
      },
      {
        groupName: "group123",
        description: "Group Description",
        roleId: 1
      }]
    });
});

服务

import { paginate } from 'nestjs-typeorm-paginate';

async getGroups(paginationDto: PaginationDto) {
        const pageOptions = await this.databaseService.preparePageData(paginationDto, 'groups');
        const queryBuilder = this.groupModel.createQueryBuilder('groups');
        const flag = paginationDto.relatinalData;
        if (relatinalData.yes === flag) {
            queryBuilder.leftJoinAndSelect("groups.role", "role");
        }
        queryBuilder.orderBy(pageOptions.order);
        pageOptions['route'] = environment.hostname + "groups";
        return await paginate(queryBuilder,pageOptions);
}

分页页面选项准备功能

async preparePageData(paginationDto, modelName) {
    const sort = {};
    if (paginationDto.sortBy && paginationDto.sortOrder) {
        const sortValue = paginationDto.sortBy
        const sortOrder = paginationDto.sortOrder;
        if (paginationDto.relatinalData && relatinalData.yes === paginationDto.relatinalData) {
            sort[modelName + '.' + sortValue] = sortOrder;
        } else {
            sort[sortValue] = sortOrder;
        }
    }
    const options = {
        page: paginationDto.page ? Number(paginationDto.page) : constant.pageLimit,
        limit: paginationDto.limit ? Number(paginationDto.limit) : constant.limit,
        order: sort ? sort : constant.sort
    };
    return options;
}

任何人都可以帮我模拟Nestjs-typeorm-paginate

Answer 1

我能够模拟Nestjs-typeorm-paginate包的 createQueryBuilder。

createQueryBuilder 模拟

createQueryBuilder: jest.fn(() => ({
    leftJoinAndSelect: jest.fn().mockReturnThis(),
    orderBy: jest.fn().mockReturnThis(),
    take: jest.fn().mockReturnThis(),
    skip: jest.fn().mockReturnThis(),
    getManyAndCount: jest.fn().mockResolvedValue([groupPageResponse]),
  }))