首先声明:我不了解 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();
});
});