116 lines
2.3 KiB
Go
116 lines
2.3 KiB
Go
package mqtt_consumer
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/eclipse/paho.mqtt.golang"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
const (
|
|
testMsg = "cpu_load_short,host=server01 value=23422.0 1422568543702900257\n"
|
|
invalidMsg = "cpu_load_short,host=server01 1422568543702900257\n"
|
|
)
|
|
|
|
func newTestMQTTConsumer() *MQTTConsumer {
|
|
n := &MQTTConsumer{
|
|
Topics: []string{"telegraf"},
|
|
Servers: []string{"localhost:1883"},
|
|
}
|
|
|
|
return n
|
|
}
|
|
|
|
// Test that default client has random ID
|
|
func TestRandomClientID(t *testing.T) {
|
|
m1 := &MQTTConsumer{
|
|
Servers: []string{"localhost:1883"}}
|
|
opts, err := m1.createOpts()
|
|
assert.NoError(t, err)
|
|
|
|
m2 := &MQTTConsumer{
|
|
Servers: []string{"localhost:1883"}}
|
|
opts2, err2 := m2.createOpts()
|
|
assert.NoError(t, err2)
|
|
|
|
assert.NotEqual(t, opts.ClientID, opts2.ClientID)
|
|
}
|
|
|
|
// Test that default client has random ID
|
|
func TestClientID(t *testing.T) {
|
|
m1 := &MQTTConsumer{
|
|
Servers: []string{"localhost:1883"},
|
|
ClientID: "telegraf-test",
|
|
}
|
|
opts, err := m1.createOpts()
|
|
assert.NoError(t, err)
|
|
|
|
m2 := &MQTTConsumer{
|
|
Servers: []string{"localhost:1883"},
|
|
ClientID: "telegraf-test",
|
|
}
|
|
opts2, err2 := m2.createOpts()
|
|
assert.NoError(t, err2)
|
|
|
|
assert.Equal(t, "telegraf-test", opts2.ClientID)
|
|
assert.Equal(t, "telegraf-test", opts.ClientID)
|
|
}
|
|
|
|
// Test that Start() fails if client ID is not set but persistent is
|
|
func TestPersistentClientIDFail(t *testing.T) {
|
|
m1 := &MQTTConsumer{
|
|
Servers: []string{"localhost:1883"},
|
|
PersistentSession: true,
|
|
}
|
|
acc := testutil.Accumulator{}
|
|
err := m1.Start(&acc)
|
|
assert.Error(t, err)
|
|
}
|
|
|
|
func mqttMsg(val string) mqtt.Message {
|
|
return &message{
|
|
topic: "telegraf/unit_test",
|
|
payload: []byte(val),
|
|
}
|
|
}
|
|
|
|
// Take the message struct from the paho mqtt client library for returning
|
|
// a test message interface.
|
|
type message struct {
|
|
duplicate bool
|
|
qos byte
|
|
retained bool
|
|
topic string
|
|
messageID uint16
|
|
payload []byte
|
|
}
|
|
|
|
func (m *message) Duplicate() bool {
|
|
return m.duplicate
|
|
}
|
|
|
|
func (m *message) Ack() {
|
|
return
|
|
}
|
|
|
|
func (m *message) Qos() byte {
|
|
return m.qos
|
|
}
|
|
|
|
func (m *message) Retained() bool {
|
|
return m.retained
|
|
}
|
|
|
|
func (m *message) Topic() string {
|
|
return m.topic
|
|
}
|
|
|
|
func (m *message) MessageID() uint16 {
|
|
return m.messageID
|
|
}
|
|
|
|
func (m *message) Payload() []byte {
|
|
return m.payload
|
|
}
|