mirror of
https://gitlab.torproject.org/tpo/anti-censorship/pluggable-transports/snowflake.git
synced 2025-10-13 11:11:30 -04:00
Add Event Bus Test
This commit is contained in:
parent
5f03f88d73
commit
b5ef18803f
1 changed files with 32 additions and 0 deletions
32
common/event/bus_test.go
Normal file
32
common/event/bus_test.go
Normal file
|
@ -0,0 +1,32 @@
|
|||
package event
|
||||
|
||||
import (
|
||||
"github.com/stretchr/testify/assert"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type stubReceiver struct {
|
||||
counter int
|
||||
}
|
||||
|
||||
func (s *stubReceiver) OnNewSnowflakeEvent(event SnowflakeEvent) {
|
||||
s.counter++
|
||||
}
|
||||
|
||||
func TestBusDispatch(t *testing.T) {
|
||||
EventBus := NewSnowflakeEventDispatcher()
|
||||
StubReceiverA := &stubReceiver{}
|
||||
StubReceiverB := &stubReceiver{}
|
||||
EventBus.AddSnowflakeEventListener(StubReceiverA)
|
||||
EventBus.AddSnowflakeEventListener(StubReceiverB)
|
||||
assert.Equal(t, 0, StubReceiverA.counter)
|
||||
assert.Equal(t, 0, StubReceiverB.counter)
|
||||
EventBus.OnNewSnowflakeEvent(EventOnSnowflakeConnected{})
|
||||
assert.Equal(t, 1, StubReceiverA.counter)
|
||||
assert.Equal(t, 1, StubReceiverB.counter)
|
||||
EventBus.RemoveSnowflakeEventListener(StubReceiverB)
|
||||
EventBus.OnNewSnowflakeEvent(EventOnSnowflakeConnected{})
|
||||
assert.Equal(t, 2, StubReceiverA.counter)
|
||||
assert.Equal(t, 1, StubReceiverB.counter)
|
||||
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue