首次提交

This commit is contained in:
2026-07-13 12:06:16 +08:00
commit e3c810b1a6
1690 changed files with 228324 additions and 0 deletions
+327
View File
@@ -0,0 +1,327 @@
'use strict'
const { test } = require('tap')
const { setup, connect, subscribe, noError } = require('./helper')
const aedes = require('../')
test('count connected clients', function (t) {
t.plan(4)
const broker = aedes()
t.teardown(broker.close.bind(broker))
t.equal(broker.connectedClients, 0, 'no connected clients')
connect(setup(broker), {
}, function () {
t.equal(broker.connectedClients, 1, 'one connected clients')
const last = connect(setup(broker), {
}, function () {
t.equal(broker.connectedClients, 2, 'two connected clients')
last.conn.destroy()
// needed because destroy() will do the trick before
// the next tick
setImmediate(function () {
t.equal(broker.connectedClients, 1, 'one connected clients')
})
})
})
})
test('call published method', function (t) {
t.plan(4)
const broker = aedes()
t.teardown(broker.close.bind(broker))
broker.published = function (packet, client, done) {
t.equal(packet.topic, 'hello', 'topic matches')
t.equal(packet.payload.toString(), 'world', 'payload matches')
t.equal(client, null, 'no client')
done()
}
broker.publish({
topic: 'hello',
payload: Buffer.from('world')
}, function (err) {
t.error(err, 'no error')
})
})
test('call published method with client', function (t) {
t.plan(4)
const broker = aedes()
t.teardown(broker.close.bind(broker))
broker.published = function (packet, client, done) {
// for internal messages, client will be null
if (client) {
t.equal(packet.topic, 'hello', 'topic matches')
t.equal(packet.payload.toString(), 'world', 'payload matches')
t.equal(packet.qos, 1)
t.equal(packet.messageId, 42)
done()
}
}
const s = connect(setup(broker))
s.inStream.write({
cmd: 'publish',
topic: 'hello',
payload: Buffer.from('world'),
qos: 1,
messageId: 42
})
})
test('emit publish event with client - QoS 0', function (t) {
t.plan(3)
const broker = aedes()
t.teardown(broker.close.bind(broker))
broker.on('publish', function (packet, client) {
// for internal messages, client will be null
if (client) {
t.equal(packet.qos, 0)
t.equal(packet.topic, 'hello', 'topic matches')
t.equal(packet.payload.toString(), 'world', 'payload matches')
}
})
const s = connect(setup(broker))
s.inStream.write({
cmd: 'publish',
topic: 'hello',
payload: Buffer.from('world'),
qos: 0
})
})
test('emit publish event with client - QoS 1', function (t) {
t.plan(4)
const broker = aedes()
t.teardown(broker.close.bind(broker))
broker.on('publish', function (packet, client) {
// for internal messages, client will be null
if (client) {
t.equal(packet.qos, 1)
t.equal(packet.messageId, 42)
t.equal(packet.topic, 'hello', 'topic matches')
t.equal(packet.payload.toString(), 'world', 'payload matches')
}
})
const s = connect(setup(broker))
s.inStream.write({
cmd: 'publish',
topic: 'hello',
payload: Buffer.from('world'),
qos: 1,
messageId: 42
})
})
test('emit subscribe event', function (t) {
t.plan(6)
const broker = aedes()
t.teardown(broker.close.bind(broker))
const s = connect(setup(broker), { clientId: 'abcde' })
broker.on('subscribe', function (subscriptions, client) {
t.same(subscriptions, [{
topic: 'hello',
qos: 0
}], 'topic matches')
t.equal(client.id, 'abcde', 'client matches')
})
subscribe(t, s, 'hello', 0, function () {
t.pass('subscribe completed')
})
})
test('emit subscribe event if unrecognized params in subscribe packet structure', function (t) {
t.plan(3)
const broker = aedes()
t.teardown(broker.close.bind(broker))
const s = noError(connect(setup(broker)))
const subs = [{ topic: 'hello', qos: 0 }]
broker.on('subscribe', function (subscriptions, client) {
t.equal(subscriptions, subs)
t.same(client, s.client)
})
s.client.subscribe({
subscriptions: subs,
restore: true
}, function (err) {
t.error(err)
})
})
test('emit unsubscribe event', function (t) {
t.plan(6)
const broker = aedes()
t.teardown(broker.close.bind(broker))
const s = connect(setup(broker), { clean: true, clientId: 'abcde' })
broker.on('unsubscribe', function (unsubscriptions, client) {
t.same(unsubscriptions, [
'hello'
], 'unsubscription matches')
t.equal(client.id, 'abcde', 'client matches')
})
subscribe(t, s, 'hello', 0, function () {
s.inStream.write({
cmd: 'unsubscribe',
messageId: 43,
unsubscriptions: ['hello']
})
s.outStream.once('data', function (packet) {
t.pass('subscribe completed')
})
})
})
test('emit unsubscribe event if unrecognized params in unsubscribe packet structure', function (t) {
t.plan(3)
const broker = aedes()
t.teardown(broker.close.bind(broker))
const s = noError(connect(setup(broker)))
const unsubs = [{ topic: 'hello', qos: 0 }]
broker.on('unsubscribe', function (unsubscriptions, client) {
t.equal(unsubscriptions, unsubs)
t.same(client, s.client)
})
s.client.unsubscribe({
unsubscriptions: unsubs,
close: true
}, function (err) {
t.error(err)
})
})
test('dont emit unsubscribe event on client close', function (t) {
t.plan(3)
const broker = aedes()
t.teardown(broker.close.bind(broker))
const s = noError(connect(setup(broker), { clientId: 'abcde' }), t)
broker.on('unsubscribe', function (unsubscriptions, client) {
t.error('unsubscribe should not be emitted')
})
subscribe(t, s, 'hello', 0, function () {
s.inStream.end({
cmd: 'disconnect'
})
s.outStream.once('data', function (packet) {
t.pass('unsubscribe completed')
})
})
})
test('emit clientDisconnect event', function (t) {
t.plan(1)
const broker = aedes()
t.teardown(broker.close.bind(broker))
broker.on('clientDisconnect', function (client) {
t.equal(client.id, 'abcde', 'client matches')
})
const s = noError(connect(setup(broker), { clientId: 'abcde' }), t)
s.inStream.end({
cmd: 'disconnect'
})
s.outStream.resume()
})
test('emits client', function (t) {
t.plan(1)
const broker = aedes()
t.teardown(broker.close.bind(broker))
broker.on('client', function (client) {
t.equal(client.id, 'abcde', 'clientId matches')
})
connect(setup(broker), {
clientId: 'abcde'
})
})
test('get aedes version', function (t) {
t.plan(1)
const broker = aedes()
t.teardown(broker.close.bind(broker))
t.equal(broker.version, require('../package.json').version)
})
test('connect and connackSent event', { timeout: 50 }, function (t) {
t.plan(3)
const s = setup()
t.teardown(s.broker.close.bind(s.broker))
const clientId = 'my-client'
s.broker.on('connackSent', function (packet, client) {
t.equal(packet.returnCode, 0)
t.equal(client.id, clientId, 'connackSent event and clientId matches')
})
s.inStream.write({
cmd: 'connect',
protocolId: 'MQTT',
protocolVersion: 4,
clean: true,
clientId,
keepalive: 0
})
s.outStream.on('data', function (packet) {
t.same(packet, {
cmd: 'connack',
returnCode: 0,
length: 2,
qos: 0,
retain: false,
dup: false,
topic: null,
payload: null,
sessionPresent: false
}, 'successful connack')
})
})