package subscription

import (
	"testing"

	"badat.dev/maeqtt/v2/mqtt/packets"
)

func TestSubscribe(t *testing.T) {
	tree := newSubscriptionTreeNode()
	topic, _ := packets.ParseTopic("a/b/c")
	channel := make(SubscriptionChannel)
	topicFilter := packets.TopicFilter{
		Topic:  topic,
		MaxQoS: 1,
	}
	tree.Subscribe(topicFilter, channel)
	subs, lock := tree.GetSubscriptions("a/b/c")
	defer lock.Unlock()

	if len(subs) != 1 {
		t.Errorf("Error storing subscriptions, expected to len(subs) to be 1, got: %v \n", len(subs))
	}
	if subs[0].MaxQoS != topicFilter.MaxQoS || subs[0].SubscriptionChannel != channel {
		t.Error("Error with data stored in a subscription")
	}
}