453 lines
11 KiB
Go
453 lines
11 KiB
Go
// Copyright (c) Mainflux
|
|
// SPDX-License-Identifier: Apache-2.0
|
|
|
|
package mqtt_test
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"log"
|
|
"testing"
|
|
|
|
"github.com/mainflux/mainflux"
|
|
authmocks "github.com/mainflux/mainflux/auth/mocks"
|
|
"github.com/mainflux/mainflux/internal/testsutil"
|
|
"github.com/mainflux/mainflux/logger"
|
|
"github.com/mainflux/mainflux/mqtt"
|
|
"github.com/mainflux/mainflux/mqtt/mocks"
|
|
"github.com/mainflux/mainflux/pkg/errors"
|
|
"github.com/mainflux/mproxy/pkg/session"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
const (
|
|
thingID = "513d02d2-16c1-4f23-98be-9e12f8fee898"
|
|
thingID1 = "513d02d2-16c1-4f23-98be-9e12f8fee899"
|
|
password = "password"
|
|
password1 = "password1"
|
|
chanID = "123e4567-e89b-12d3-a456-000000000001"
|
|
invalidID = authmocks.InvalidValue
|
|
clientID = "clientID"
|
|
clientID1 = "clientID1"
|
|
subtopic = "testSubtopic"
|
|
invalidChannelIDTopic = "channels/**/messages"
|
|
)
|
|
|
|
var (
|
|
topicMsg = "channels/%s/messages"
|
|
topic = fmt.Sprintf(topicMsg, chanID)
|
|
invalidTopic = authmocks.InvalidValue
|
|
payload = []byte("[{'n':'test-name', 'v': 1.2}]")
|
|
topics = []string{topic}
|
|
invalidTopics = []string{authmocks.InvalidValue}
|
|
invalidChanIDTopics = []string{fmt.Sprintf(topicMsg, authmocks.InvalidValue)}
|
|
// Test log messages for cases the handler does not provide a return value.
|
|
logBuffer = bytes.Buffer{}
|
|
sessionClient = session.Session{
|
|
ID: clientID,
|
|
Username: thingID,
|
|
Password: []byte(password),
|
|
}
|
|
sessionClientSub = session.Session{
|
|
ID: clientID1,
|
|
Username: thingID1,
|
|
Password: []byte(password1),
|
|
}
|
|
invalidThingSessionClient = session.Session{
|
|
ID: clientID,
|
|
Username: invalidID,
|
|
Password: []byte(password),
|
|
}
|
|
)
|
|
|
|
func TestAuthConnect(t *testing.T) {
|
|
handler, _ := newHandler()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
err error
|
|
session *session.Session
|
|
}{
|
|
{
|
|
desc: "connect without active session",
|
|
err: mqtt.ErrClientNotInitialized,
|
|
session: nil,
|
|
},
|
|
{
|
|
desc: "connect without clientID",
|
|
err: mqtt.ErrMissingClientID,
|
|
session: &session.Session{
|
|
ID: "",
|
|
Username: thingID,
|
|
Password: []byte(password),
|
|
},
|
|
},
|
|
{
|
|
desc: "connect with invalid password",
|
|
err: nil,
|
|
session: &session.Session{
|
|
ID: clientID,
|
|
Username: thingID,
|
|
Password: []byte(""),
|
|
},
|
|
},
|
|
{
|
|
desc: "connect with valid password and invalid username",
|
|
err: nil,
|
|
session: &invalidThingSessionClient,
|
|
},
|
|
{
|
|
desc: "connect with valid username and password",
|
|
err: nil,
|
|
session: &sessionClient,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.AuthConnect(ctx)
|
|
assert.True(t, errors.Contains(err, tc.err), fmt.Sprintf("%s: expected %s got %s\n", tc.desc, tc.err, err))
|
|
}
|
|
}
|
|
|
|
func TestAuthPublish(t *testing.T) {
|
|
handler, auth := newHandler()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
err error
|
|
topic *string
|
|
payload []byte
|
|
}{
|
|
{
|
|
desc: "publish with an inactive client",
|
|
session: nil,
|
|
err: mqtt.ErrClientNotInitialized,
|
|
topic: &topic,
|
|
payload: payload,
|
|
},
|
|
{
|
|
desc: "publish without topic",
|
|
session: &sessionClient,
|
|
err: mqtt.ErrMissingTopicPub,
|
|
topic: nil,
|
|
payload: payload,
|
|
},
|
|
{
|
|
desc: "publish with malformed topic",
|
|
session: &sessionClient,
|
|
err: mqtt.ErrMalformedTopic,
|
|
topic: &invalidTopic,
|
|
payload: payload,
|
|
},
|
|
{
|
|
desc: "publish successfully",
|
|
session: &sessionClient,
|
|
err: nil,
|
|
topic: &topic,
|
|
payload: payload,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
repocall := auth.On("Authorize", mock.Anything, mock.Anything).Return(&mainflux.AuthorizeRes{Authorized: true, Id: testsutil.GenerateUUID(t)}, nil)
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.AuthPublish(ctx, tc.topic, &tc.payload)
|
|
assert.True(t, errors.Contains(err, tc.err), fmt.Sprintf("%s: expected %s got %s\n", tc.desc, tc.err, err))
|
|
repocall.Unset()
|
|
}
|
|
}
|
|
|
|
func TestAuthSubscribe(t *testing.T) {
|
|
handler, auth := newHandler()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
err error
|
|
topic *[]string
|
|
}{
|
|
{
|
|
desc: "subscribe without active session",
|
|
session: nil,
|
|
err: mqtt.ErrClientNotInitialized,
|
|
topic: &topics,
|
|
},
|
|
{
|
|
desc: "subscribe without topics",
|
|
session: &sessionClient,
|
|
err: mqtt.ErrMissingTopicSub,
|
|
topic: nil,
|
|
},
|
|
{
|
|
desc: "subscribe with invalid topics",
|
|
session: &sessionClient,
|
|
err: mqtt.ErrMalformedTopic,
|
|
topic: &invalidTopics,
|
|
},
|
|
{
|
|
desc: "subscribe with invalid channel ID",
|
|
session: &sessionClient,
|
|
err: errors.ErrAuthorization,
|
|
topic: &invalidChanIDTopics,
|
|
},
|
|
{
|
|
desc: "subscribe successfully",
|
|
session: &sessionClientSub,
|
|
err: nil,
|
|
topic: &topics,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
repocall := auth.On("Authorize", mock.Anything, mock.Anything).Return(&mainflux.AuthorizeRes{Authorized: true, Id: testsutil.GenerateUUID(t)}, nil)
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.AuthSubscribe(ctx, tc.topic)
|
|
assert.True(t, errors.Contains(err, tc.err), fmt.Sprintf("%s: expected %s got %s\n", tc.desc, tc.err, err))
|
|
repocall.Unset()
|
|
}
|
|
}
|
|
|
|
func TestConnect(t *testing.T) {
|
|
handler, _ := newHandler()
|
|
logBuffer.Reset()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
err error
|
|
logMsg string
|
|
}{
|
|
{
|
|
desc: "connect without active session",
|
|
session: nil,
|
|
err: errors.Wrap(mqtt.ErrFailedConnect, mqtt.ErrClientNotInitialized),
|
|
},
|
|
{
|
|
desc: "connect with active session",
|
|
session: &sessionClient,
|
|
logMsg: fmt.Sprintf(mqtt.LogInfoConnected, clientID),
|
|
err: nil,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.Connect(ctx)
|
|
assert.Contains(t, logBuffer.String(), tc.logMsg)
|
|
assert.Equal(t, tc.err, err)
|
|
}
|
|
}
|
|
|
|
func TestPublish(t *testing.T) {
|
|
handler, _ := newHandler()
|
|
logBuffer.Reset()
|
|
|
|
malformedSubtopics := topic + "/" + subtopic + "%"
|
|
wrongCharSubtopics := topic + "/" + subtopic + ">"
|
|
validSubtopic := topic + "/" + subtopic
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
topic string
|
|
payload []byte
|
|
logMsg string
|
|
err error
|
|
}{
|
|
{
|
|
desc: "publish without active session",
|
|
session: nil,
|
|
topic: topic,
|
|
payload: payload,
|
|
err: errors.Wrap(mqtt.ErrFailedPublish, mqtt.ErrClientNotInitialized),
|
|
},
|
|
{
|
|
desc: "publish with invalid topic",
|
|
session: &sessionClient,
|
|
topic: invalidTopic,
|
|
payload: payload,
|
|
logMsg: fmt.Sprintf(mqtt.LogInfoPublished, clientID, invalidTopic),
|
|
err: errors.Wrap(mqtt.ErrFailedPublish, mqtt.ErrMalformedTopic),
|
|
},
|
|
{
|
|
desc: "publish with invalid channel ID",
|
|
session: &sessionClient,
|
|
topic: invalidChannelIDTopic,
|
|
payload: payload,
|
|
err: errors.Wrap(mqtt.ErrFailedPublish, mqtt.ErrMalformedTopic),
|
|
},
|
|
{
|
|
desc: "publish with malformed subtopic",
|
|
session: &sessionClient,
|
|
topic: malformedSubtopics,
|
|
payload: payload,
|
|
err: errors.Wrap(mqtt.ErrFailedParseSubtopic, mqtt.ErrMalformedSubtopic),
|
|
},
|
|
{
|
|
desc: "publish with subtopic containing wrong character",
|
|
session: &sessionClient,
|
|
topic: wrongCharSubtopics,
|
|
payload: payload,
|
|
err: errors.Wrap(mqtt.ErrFailedParseSubtopic, mqtt.ErrMalformedSubtopic),
|
|
},
|
|
{
|
|
desc: "publish with subtopic",
|
|
session: &sessionClient,
|
|
topic: validSubtopic,
|
|
payload: payload,
|
|
logMsg: subtopic,
|
|
},
|
|
{
|
|
desc: "publish without subtopic",
|
|
session: &sessionClient,
|
|
topic: topic,
|
|
payload: payload,
|
|
logMsg: "",
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.Publish(ctx, &tc.topic, &tc.payload)
|
|
assert.Contains(t, logBuffer.String(), tc.logMsg)
|
|
assert.Equal(t, tc.err, err)
|
|
}
|
|
}
|
|
|
|
func TestSubscribe(t *testing.T) {
|
|
handler, _ := newHandler()
|
|
logBuffer.Reset()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
topic []string
|
|
logMsg string
|
|
err error
|
|
}{
|
|
{
|
|
desc: "subscribe without active session",
|
|
session: nil,
|
|
topic: topics,
|
|
err: errors.Wrap(mqtt.ErrFailedSubscribe, mqtt.ErrClientNotInitialized),
|
|
},
|
|
{
|
|
desc: "subscribe with valid session and topics",
|
|
session: &sessionClient,
|
|
topic: topics,
|
|
logMsg: fmt.Sprintf(mqtt.LogInfoSubscribed, clientID, topics[0]),
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.Subscribe(ctx, &tc.topic)
|
|
assert.Contains(t, logBuffer.String(), tc.logMsg)
|
|
assert.Equal(t, tc.err, err)
|
|
}
|
|
}
|
|
|
|
func TestUnsubscribe(t *testing.T) {
|
|
handler, _ := newHandler()
|
|
logBuffer.Reset()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
topic []string
|
|
logMsg string
|
|
err error
|
|
}{
|
|
{
|
|
desc: "unsubscribe without active session",
|
|
session: nil,
|
|
topic: topics,
|
|
err: errors.Wrap(mqtt.ErrFailedUnsubscribe, mqtt.ErrClientNotInitialized),
|
|
},
|
|
{
|
|
desc: "unsubscribe with valid session and topics",
|
|
session: &sessionClient,
|
|
topic: topics,
|
|
logMsg: fmt.Sprintf(mqtt.LogInfoUnsubscribed, clientID, topics[0]),
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.Unsubscribe(ctx, &tc.topic)
|
|
assert.Contains(t, logBuffer.String(), tc.logMsg)
|
|
assert.Equal(t, tc.err, err)
|
|
}
|
|
}
|
|
|
|
func TestDisconnect(t *testing.T) {
|
|
handler, _ := newHandler()
|
|
logBuffer.Reset()
|
|
|
|
cases := []struct {
|
|
desc string
|
|
session *session.Session
|
|
topic []string
|
|
logMsg string
|
|
err error
|
|
}{
|
|
{
|
|
desc: "disconnect without active session",
|
|
session: nil,
|
|
topic: topics,
|
|
err: errors.Wrap(mqtt.ErrFailedDisconnect, mqtt.ErrClientNotInitialized),
|
|
},
|
|
{
|
|
desc: "disconnect with valid session",
|
|
session: &sessionClient,
|
|
topic: topics,
|
|
err: nil,
|
|
},
|
|
}
|
|
|
|
for _, tc := range cases {
|
|
ctx := context.TODO()
|
|
if tc.session != nil {
|
|
ctx = session.NewContext(ctx, tc.session)
|
|
}
|
|
err := handler.Disconnect(ctx)
|
|
assert.Contains(t, logBuffer.String(), tc.logMsg)
|
|
assert.Equal(t, tc.err, err)
|
|
}
|
|
}
|
|
|
|
func newHandler() (session.Handler, *authmocks.Service) {
|
|
logger, err := logger.New(&logBuffer, "debug")
|
|
if err != nil {
|
|
log.Fatalf("failed to create logger: %s", err)
|
|
}
|
|
auth := new(authmocks.Service)
|
|
eventStore := mocks.NewEventStore()
|
|
return mqtt.NewHandler(mocks.NewPublisher(), eventStore, logger, auth), auth
|
|
}
|