61 lines
1.7 KiB
Go
61 lines
1.7 KiB
Go
package subscription
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"badat.dev/maeqtt/v2/mqtt/packets"
|
|
)
|
|
|
|
func assertMatches(topic packets.Topic, topicName string, shouldMatch bool, t *testing.T) {
|
|
tree := newSubscriptionTreeNode()
|
|
channel := make(SubscriptionChannel)
|
|
topicFilter := packets.TopicFilter{
|
|
Topic: topic,
|
|
MaxQoS: 1,
|
|
}
|
|
tree.Subscribe(topicFilter, channel)
|
|
subs := tree.GetSubscriptions(topicName)
|
|
|
|
if (len(subs) != 1) && shouldMatch {
|
|
t.Errorf("Topic %v did not match %v", topic, topicName)
|
|
}
|
|
if (len(subs) == 1) && !shouldMatch {
|
|
t.Errorf("Topic %v matched %v (it wasn't supposed to)", topic, topicName )
|
|
}
|
|
}
|
|
|
|
func TestSubscribe(t *testing.T) {
|
|
t.Parallel()
|
|
topic, _ := packets.ParseTopic("a/b/c")
|
|
go assertMatches(topic, "a/b/c", true, t)
|
|
go assertMatches(topic, "a/c/c", false, t)
|
|
go assertMatches(topic, "b/b/c", false, t)
|
|
go assertMatches(topic, "aaa/c/a", false, t)
|
|
}
|
|
|
|
func TestSingleLevelWildcard(t *testing.T) {
|
|
t.Parallel()
|
|
topic, _ := packets.ParseTopic("a/+/c")
|
|
go assertMatches(topic, "a/b/c", true, t)
|
|
go assertMatches(topic, "a/c/c", true, t)
|
|
go assertMatches(topic, "a/b/d", false, t)
|
|
go assertMatches(topic, "aaa/c/a", false, t)
|
|
|
|
topic, _ = packets.ParseTopic("+/+/+")
|
|
go assertMatches(topic, "a/b/c", true, t)
|
|
go assertMatches(topic, "a/c/c", true, t)
|
|
go assertMatches(topic, "a/b/d/e", false, t)
|
|
go assertMatches(topic, "c/a", true, t)
|
|
}
|
|
|
|
func TestMultiLevelWildcard(t *testing.T) {
|
|
t.Parallel()
|
|
topic, _ := packets.ParseTopic("a/b/c/#")
|
|
go assertMatches(topic, "a/b/c", true, t)
|
|
go assertMatches(topic, "a/b/c/a", true, t)
|
|
go assertMatches(topic, "a/b/c/d", true, t)
|
|
go assertMatches(topic, "a/b/c/f", true, t)
|
|
|
|
go assertMatches(topic, "a/b/d/a", false, t)
|
|
}
|