在使用Jest时,模拟被测试者调用的服务方法

时间:2017-05-01 22:15:54

标签: javascript jestjs

我正在尝试模拟一种方法的服务,我从我的测试中导出为模块。 这是我使用" sinon"的东西,但我想尽可能多地使用jest。

这是一个经典测试,我有一个"身份验证"服务和邮件"服务。

"身份验证"服务可以注册新用户,并在每次新注册后,要求邮件服务向新用户发送欢迎电子邮件"。

因此测试我的身份验证服务的注册方法,我想断言(并模拟)"发送"邮件服务的方法。

怎么做?这是我尝试过的,但它调用了原始的mailer.send方法:

// authentication.js

const mailer = require('./mailer');

class authentication {
  register() { // The method i am trying to test
    // ...

    mailer.send();
  }
}

const authentication = new Authentication();

module.exports = authentication;


// mailer.js

class Mailer {
  send() { // The method i am trying to mock
    // ...
  }
}

const mailer = new Mailer();

module.exports = mailer;


// authentication.test.js

const authentication = require('../../services/authentication');

describe('Service Authentication', () => {
  describe('register', () => {
    test('should send a welcome email', done => {
      co(function* () {
        try {
          jest.mock('../../services/mailer');
          const mailer = require('../../services/mailer');
          mailer.send = jest.fn( () => { // I would like this mock to be called in authentication.register()
            console.log('SEND MOCK CALLED !');
            return Promise.resolve();
          });

          yield authentication.register(knownUser);

          // expect();

          done();
        } catch(e) {
          done(e);
        }
      });
    });
  });
});

1 个答案:

答案 0 :(得分:3)

首先,您必须使用间谍模拟mailer模块,以便稍后设置。你可以让jest知道在测试中使用promise,看看docs有两种方法可以做到这一点。

const authentication = require('../../services/authentication');
const mailer = require('../../services/mailer');
jest.mock('../../services/mailer', () => ({send: jest.fn()})); 

describe('Service Authentication', () => {
  describe('register', () => {
    test('should send a welcome email', async() => {
          const p = Promise.resolve()
          mailer.send.mockImplementation(() =>  p) 
          authentication.register(knownUser);
          await p
          expect(mailer.send).toHaveBeenCalled;
        }
      });
    });
  });
});