Xray-core/common/signal/pubsub/pubsub_test.go
2020-12-04 09:36:16 +08:00

35 lines
460 B
Go

package pubsub_test
import (
"testing"
. "github.com/xtls/xray-core/common/signal/pubsub"
)
func TestPubsub(t *testing.T) {
service := NewService()
sub := service.Subscribe("a")
service.Publish("a", 1)
select {
case v := <-sub.Wait():
if v != 1 {
t.Error("expected subscribed value 1, but got ", v)
}
default:
t.Fail()
}
sub.Close()
service.Publish("a", 2)
select {
case <-sub.Wait():
t.Fail()
default:
}
service.Cleanup()
}