156 lines
3.3 KiB
Go
156 lines
3.3 KiB
Go
package mqtt_consumer
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/influxdata/telegraf/plugins/parsers"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
|
|
"github.com/eclipse/paho.mqtt.golang"
|
|
)
|
|
|
|
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 TestRunParser(t *testing.T) {
|
|
n := newTestMQTTConsumer()
|
|
acc := testutil.Accumulator{}
|
|
n.acc = &acc
|
|
n.parser, _ = parsers.NewInfluxParser()
|
|
|
|
n.recvMessage(nil, mqttMsg(testMsg))
|
|
|
|
if a := acc.NFields(); a != 1 {
|
|
t.Errorf("got %v, expected %v", a, 1)
|
|
}
|
|
}
|
|
|
|
// Test that the parser ignores invalid messages
|
|
func TestRunParserInvalidMsg(t *testing.T) {
|
|
n := newTestMQTTConsumer()
|
|
acc := testutil.Accumulator{}
|
|
n.acc = &acc
|
|
n.parser, _ = parsers.NewInfluxParser()
|
|
|
|
n.recvMessage(nil, mqttMsg(invalidMsg))
|
|
|
|
if a := acc.NFields(); a != 0 {
|
|
t.Errorf("got %v, expected %v", a, 0)
|
|
}
|
|
assert.Len(t, acc.Errors, 1)
|
|
}
|
|
|
|
// Test that the parser parses line format messages into metrics
|
|
func TestRunParserAndGather(t *testing.T) {
|
|
n := newTestMQTTConsumer()
|
|
acc := testutil.Accumulator{}
|
|
n.acc = &acc
|
|
n.parser, _ = parsers.NewInfluxParser()
|
|
|
|
n.recvMessage(nil, mqttMsg(testMsg))
|
|
|
|
acc.AssertContainsFields(t, "cpu_load_short",
|
|
map[string]interface{}{"value": float64(23422)})
|
|
}
|
|
|
|
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) 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
|
|
}
|