如何为箭头函数内部的逻辑正确创建单元测试

问题描述

我正在使用 HapiJS 开发现有的 NodeJS Web 服务，Hapi Lab 与 Sinon 一起进行测试。该服务使用 massjs 连接到 Postgres 数据库。有一种方法是由其他人实现的，它没有单元测试。现在我正在重用这个方法，我想为它实现一些单元测试。 该方法在其中执行一个 massjs 事务，持久化到多个表中。

async createSomething(payload) {

    const { code,member } = payload;

    const foundCompany = await this.rawDb.ethnics.tenants.findOne({ code });

    if (foundCompany && foundCompany.companyId) {

        const { companyId } = foundCompany;
        const { foreignId } = member;

        return await this.rawDb.withTransaction(async (tx) => {

            const foundMember = await tx.ethnics.members.findOne({ foreign_id: foreignId,company_id: companyId });

            if (!foundMember) {

                //some business logic...
                const newMember = await tx.ethnics.members.insert(member);
                //more business logic persisting to other tables...
                return newMember;
            }
        });
    }
}

问题是，我不知道如何只在箭头函数内部存根，而不存根整个箭头函数。我只想存根 tx 的调用。我也不想使用数据库，而是存根 rawDb 属性。从单元测试的角度来看，这可行吗？

解决方法

是的，这是可行的。有两种选择：

1. 直接存根 MassiveJS 方法。

存根大量方法的示例findOne：

const massive = require('massive');
const sinon = require('sinon');

// Stub massive.Readable class method findOne.
// You need to find where is the real method findOne and stub it.
const stub = sinon.stub(massive.Readable,'findOne');
// You can resolve it.
stub.resolves();
// Or you can throw it.
stub.throws(new Error('xxx'));

2. 在内存中使用 pg 进行测试。

仅出于测试目的，您可以使用以下模块：test-pg-pool 或 pg-mem。测试前启动测试pg，测试结束后销毁。