Question

我有以下代码。

  private client: any;

  this.client = mqtt.connect(url, mqttOptions);

  this.client.on('message', (topic: string, message: string) => {
   console.log('came inside onMessage');
    let ksiotMessage: KSIOTMessage;
    let receivedTopic: KSIOTTopic;
    receivedTopic = getValue(topic);
    ksiotMessage = {
      ksiotTopic: receivedTopic,
      payload: message
    };
    messageReceivedCallBack.onMessageReceived(ksiotMessage);
  });

在这里，有一个匿名函数在mqtt客户端发出消息事件时被触发。我无法控制这个，这意味着它不是我，它将这个绑定到一个事件发射器。此外，我无法添加计时器并等待此事件触发，因为我使用sinon模拟了客户端的连接。所以没有真正的联系。那么如何手动发出此消息事件并查看是否已调用messageReceivedCallBack的onMessageReceived？我是Javascript单元测试的新手。目前我正在使用mocha和sinon来满足我的单元测试要求。

请建议。

Answer 1

首先是免责声明：我不了解TypeScript，也不知道应用程序的内部，因此以下内容可能无法正常使用。但是，它可能会为您提供有关如何实施这些测试的一些想法。

我抽象了你的客户端代码（需要测试的代码）：

// mqtt-module.js
const mqtt = require('mqtt');
const messageReceivedCallBack = require('./callbacks');

// This mimics what your code does. Instead of a class I just make it a
// simple function.
module.exports = {
 setup() {
    let client = mqtt.connect();
    client.on('message', (topic, message) => {
      // We want to test this.
      messageReceivedCallBack.onMessageReceived(message);
    });
  }
}

在您的代码中，不清楚messageReceivedCallBack的来源。因为你必须能够从Sinon访问它，它应该在一个可导入的模块中（但是，这确实依赖于导入被缓存的事实，就像require()一样，我不确定如果TS这样做了。）

这是我使用的简单模拟：

// callbacks.js
module.exports = {
  onMessageReceived(message) {}
};

最后，测试本身。它相当复杂，因为它需要做各种各样的事情：

创建一个EventEmitter子类，用于将原始MqttClient替换为
存根各种功能和回调
设置测试环境

代码：

// test.js
const mqtt = require('mqtt');

// The modules mentioned above.
const mqttModule              = require('./mqtt-module');
const messageReceivedCallBack = require('./callbacks');

// Set up Sinon and Chai
const sinon = require('sinon');
const chai  = require('chai');
let expect  = chai.expect;
chai.use(require('sinon-chai'));

// Create a fake client for testing purposes.
const EventEmitter = require('events').EventEmitter;
class Client extends EventEmitter {}

// The test case.
describe('my test case', () => {
  var mockClient;

  beforeEach(() => {
    mockClient = new Client();
    // mqtt.connect() returns a fake client instance, with
    // just enough logic to allow events to be emitted and
    // received.
    sinon.stub(mqtt, 'connect').returns(mockClient);

    // Call the setup of our MQTT class (this will likely be
    // entirely different in your case, but the idea is that
    // it gets called _after_ `mqtt.connect()` gets stubbed.
    mqttModule.setup();
  });

  afterEach(() => {
    // Restore the original.
    mqtt.connect.restore();
  });

  it('should call messageReceivedCallBack.onMessageReceived', () => {
    // The message that we're going to pass.
    let message = 'this is a test message';

    // We want to stub messageReceivedCallBack.onMessageReceived()
    sinon.stub(messageReceivedCallBack, 'onMessageReceived');

    // Emit a `message` event on our mock client, which will trigger
    // the `client.on('message', ...)` in your MQTT class.
    mockClient.emit('message', 'topic', message);

    // Test if the stub was called with the proper argument.
    expect(messageReceivedCallBack.onMessageReceived).to.be.calledWith(message);

    // Restore to the original function.
    messageReceivedCallBack.onMessageReceived.restore();
  });

});

Answer 2

export class MqttClientMock extends events.EventEmitter {
  subscribe = sinon.stub()
  publish = sinon.stub()
  unsubscribe = sinon.stub()
}

sandbox.stub(mqtt, 'connect').returns(new MqttClientMock() as undefined as MqttClient)

单元测试不受事件发射器约束的事件

2 个答案: