javascript - 对具有回调函数的函数进行单元测试(请求模块)

Question

我正在我的 NodeJS 项目中进行测试，并且我想要对以下函数进行单元测试:

function myRequest(targetUrl, reqBody) {
  return new Promise((resolve, reject) => {
    request.post(targetUrl, { json: reqBody }, (error, response, body) => {
      if (!error && response.statusCode === 200) {
        resolve(body.transferId);
      } else {
        reject(error || body.description || body);
      }
    });
  });
}

我正在使用 Mocha 和 sinon。如何测试这个功能？

首先，我使用自己的模拟请求模块测试了成功场景。现在，我想做错误场景，这可能是 post 函数出错。在不更改或创建新的请求模拟(以返回错误)的情况下，我该如何做到这一点？有可能吗？

Answer 1

这是单元测试解决方案，您应该使用sinon.stub:

例如index.ts:

import request from 'request';

export function myRequest(targetUrl, reqBody) {
  return new Promise((resolve, reject) => {
    request.post(targetUrl, { json: reqBody }, (error, response, body) => {
      if (!error && response.statusCode === 200) {
        resolve(body.transferId);
      } else {
        reject(error || body.description || body);
      }
    });
  });
}

index.spec.ts:

import { myRequest } from '.';
import chai from 'chai';
import sinon from 'sinon';
import chaiAsPromised from 'chai-as-promised';
import request from 'request';
chai.use(chaiAsPromised);

const { expect } = chai;

describe('myRequest', () => {
  it('should request success', async done => {
    // @ts-ignore
    const stub = sinon.stub(request, 'post').callsFake((uri, options, callback) => {
      const mResponse = { statusCode: 200 };
      const mBody = { transferId: 1 };
      callback(null, mResponse, mBody);
      done();
    });
    const actualValue = await myRequest('url', {});
    // @ts-ignore
    stub.calledOnceWith('url', { json: {} });
    expect(actualValue).to.eq(1);
    stub.restore();
  });

  it('should throw error use request error', async done => {
    const mError = new Error('Internal server error');
    const mResponse = { statusCode: 500 };
    // @ts-ignore
    const stub = sinon.stub(request, 'post').callsFake((uri, options, callback) => {
      callback(mError, mResponse, null);
      done();
    });
    await expect(myRequest('url', {})).to.be.rejectedWith(mError);
    // @ts-ignore
    stub.calledOnceWith('url', { json: {} });
    stub.restore();
  });

  it('should throw error use body.description as error message', async done => {
    const mResponse = { statusCode: 500 };
    const mBody = { description: 'some error' };
    // @ts-ignore
    const stub = sinon.stub(request, 'post').callsFake((uri, options, callback) => {
      callback(null, mResponse, mBody);
      done();
    });
    await expect(myRequest('url', {})).to.be.rejectedWith(mBody.description);
    // @ts-ignore
    stub.calledOnceWith('url', { json: {} });
    stub.restore();
  });

  it('should throw error use body as error message', async done => {
    const mResponse = { statusCode: 500 };
    const mBody = 'some error';
    // @ts-ignore
    const stub = sinon.stub(request, 'post').callsFake((uri, options, callback) => {
      callback(null, mResponse, mBody);
      done();
    });
    await expect(myRequest('url', {})).to.be.rejectedWith(mBody);
    // @ts-ignore
    stub.calledOnceWith('url', { json: {} });
    stub.restore();
  });
});

100%覆盖率的单元测试结果:

  myRequest
    ✓ should request success
    ✓ should throw error use request error
    ✓ should throw error use body.description as error message
    ✓ should throw error use body as error message


  4 passing (13ms)

---------------|----------|----------|----------|----------|-------------------|
File           |  % Stmts | % Branch |  % Funcs |  % Lines | Uncovered Line #s |
---------------|----------|----------|----------|----------|-------------------|
All files      |      100 |      100 |      100 |      100 |                   |
 index.spec.ts |      100 |      100 |      100 |      100 |                   |
 index.ts      |      100 |      100 |      100 |      100 |                   |
---------------|----------|----------|----------|----------|-------------------|

源代码:https://github.com/mrdulin/mocha-chai-sinon-codelab/tree/master/src/stackoverflow/58822996