90 lines
1.6 KiB
Go
90 lines
1.6 KiB
Go
package mqtt
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/eclipse/paho.mqtt.golang"
|
|
"github.com/micro/go-micro/broker"
|
|
)
|
|
|
|
func TestMQTTMock(t *testing.T) {
|
|
c := newMockClient()
|
|
|
|
if tk := c.Connect(); tk == nil {
|
|
t.Fatal("got nil token")
|
|
}
|
|
|
|
if tk := c.Subscribe("mock", 0, func(cm mqtt.Client, m mqtt.Message) {
|
|
t.Logf("Received payload %+v", string(m.Payload()))
|
|
}); tk == nil {
|
|
t.Fatal("got nil token")
|
|
}
|
|
|
|
if tk := c.Publish("mock", 0, false, []byte(`hello world`)); tk == nil {
|
|
t.Fatal("got nil token")
|
|
}
|
|
|
|
if tk := c.Unsubscribe("mock"); tk == nil {
|
|
t.Fatal("got nil token")
|
|
}
|
|
|
|
c.Disconnect(0)
|
|
}
|
|
|
|
func TestMQTTHandler(t *testing.T) {
|
|
p := &mqttPub{
|
|
topic: "mock",
|
|
msg: &broker.Message{Body: []byte(`hello`)},
|
|
}
|
|
|
|
if p.Topic() != "mock" {
|
|
t.Fatal("Expected topic mock got", p.Topic())
|
|
}
|
|
|
|
if string(p.Message().Body) != "hello" {
|
|
t.Fatal("Expected `hello` message got %s", string(p.Message().Body))
|
|
}
|
|
|
|
s := &mqttSub{
|
|
topic: "mock",
|
|
client: newMockClient(),
|
|
}
|
|
|
|
s.client.Connect()
|
|
|
|
if s.Topic() != "mock" {
|
|
t.Fatal("Expected topic mock got", s.Topic())
|
|
}
|
|
|
|
if err := s.Unsubscribe(); err != nil {
|
|
t.Fatal("Error unsubscribing", err)
|
|
}
|
|
|
|
s.client.Disconnect(0)
|
|
}
|
|
|
|
func TestMQTT(t *testing.T) {
|
|
b := NewBroker()
|
|
|
|
if err := b.Init(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
// use mock client
|
|
b.(*mqttBroker).client = newMockClient()
|
|
|
|
if tk := b.(*mqttBroker).client.Connect(); tk == nil {
|
|
t.Fatal("got nil token")
|
|
}
|
|
|
|
if err := b.Publish("mock", &broker.Message{Body: []byte(`hello`)}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
if err := b.Disconnect(); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
b.(*mqttBroker).client.Disconnect(0)
|
|
}
|