v2ray-core/app/pubsub/internal/pubsub_test.go

package internal_test

import (
	"testing"
	"time"

	"github.com/v2ray/v2ray-core/app/pubsub"
	. "github.com/v2ray/v2ray-core/app/pubsub/internal"
	apptesting "github.com/v2ray/v2ray-core/app/testing"
	v2testing "github.com/v2ray/v2ray-core/testing"
	"github.com/v2ray/v2ray-core/testing/assert"
)

func TestPubsub(t *testing.T) {
	v2testing.Current(t)

	messages := make(map[string]pubsub.PubsubMessage)

	ps := New()
	ps.Subscribe(&apptesting.Context{}, "t1", func(message pubsub.PubsubMessage) {
		messages["t1"] = message
	})

	ps.Subscribe(&apptesting.Context{}, "t2", func(message pubsub.PubsubMessage) {
		messages["t2"] = message
	})

	message := pubsub.PubsubMessage([]byte("This is a pubsub message."))
	ps.Publish(&apptesting.Context{}, "t2", message)
	<-time.Tick(time.Second)

	_, found := messages["t1"]
	assert.Bool(found).IsFalse()

	actualMessage, found := messages["t2"]
	assert.Bool(found).IsTrue()
	assert.StringLiteral(string(actualMessage)).Equals(string(message))
}