Jay
76 lines
1.6 KiB
Go
76 lines
1.6 KiB
Go
package prism
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"git.wisehodl.dev/jay/go-roots-ws"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNoticeHandler(t *testing.T) {
|
|
t.Run("notice delivered", func(t *testing.T) {
|
|
p, envoy := newMockEnvoy(t)
|
|
h := NewNoticeHandler(envoy)
|
|
t.Cleanup(h.Close)
|
|
|
|
p.receive([]byte(envelope.EncloseNotice("hello")))
|
|
|
|
var got Notice
|
|
Eventually(t, func() bool {
|
|
select {
|
|
case got = <-h.Notices():
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}, "notice not delivered")
|
|
|
|
assert.Equal(t, "hello", got.Message)
|
|
assert.Equal(t, p.url, got.PeerID)
|
|
assert.False(t, got.Timestamp.IsZero())
|
|
})
|
|
|
|
t.Run("malformed ignored", func(t *testing.T) {
|
|
p, envoy := newMockEnvoy(t)
|
|
h := NewNoticeHandler(envoy)
|
|
t.Cleanup(h.Close)
|
|
|
|
p.receive([]byte("not json"))
|
|
|
|
Never(t, func() bool {
|
|
select {
|
|
case <-h.Notices():
|
|
return true
|
|
default:
|
|
return false
|
|
}
|
|
}, "notices channel should remain empty")
|
|
})
|
|
|
|
t.Run("observer notified", func(t *testing.T) {
|
|
obs := &mockObserver{}
|
|
p, envoy := newMockEnvoy(t, WithEmbassyObserver(obs))
|
|
h := NewNoticeHandler(envoy)
|
|
t.Cleanup(h.Close)
|
|
|
|
p.receive([]byte(envelope.EncloseNotice("hello")))
|
|
|
|
Eventually(t, func() bool {
|
|
events := EventsOf[NoticeReceived](obs)
|
|
return len(events) == 1 &&
|
|
events[0].Message == "hello" &&
|
|
!events[0].At.IsZero()
|
|
}, "NoticeReceived observable not recorded")
|
|
})
|
|
|
|
t.Run("close cleans up", func(t *testing.T) {
|
|
_, envoy := newMockEnvoy(t)
|
|
h := NewNoticeHandler(envoy)
|
|
|
|
h.Close()
|
|
|
|
_, ok := <-h.Notices()
|
|
assert.False(t, ok, "notices channel should be closed")
|
|
})
|
|
}
|