diff options
Diffstat (limited to 'eventer/eventer_test.go')
-rw-r--r-- | eventer/eventer_test.go | 57 |
1 files changed, 52 insertions, 5 deletions
diff --git a/eventer/eventer_test.go b/eventer/eventer_test.go index b35267af6..a5db6d901 100644 --- a/eventer/eventer_test.go +++ b/eventer/eventer_test.go @@ -1,9 +1,13 @@ package eventer -import "testing" +import ( + "math/rand" + "testing" + "time" +) func TestChannel(t *testing.T) { - eventer := New(nil) + eventer := New() c := make(Channel, 1) eventer.RegisterChannel("test", c) @@ -17,7 +21,7 @@ func TestChannel(t *testing.T) { } func TestFunction(t *testing.T) { - eventer := New(nil) + eventer := New() var data string eventer.RegisterFunc("test", func(ev Event) { @@ -31,7 +35,7 @@ func TestFunction(t *testing.T) { } func TestRegister(t *testing.T) { - eventer := New(nil) + eventer := New() c := eventer.Register("test") eventer.Post("test", "hello world") @@ -44,7 +48,7 @@ func TestRegister(t *testing.T) { } func TestOn(t *testing.T) { - eventer := New(nil) + eventer := New() c := make(Channel, 1) eventer.On("test", c) @@ -64,3 +68,46 @@ func TestOn(t *testing.T) { t.Error("Expected function event with data 'hello world'. Got", data) } } + +func TestConcurrentUsage(t *testing.T) { + rand.Seed(time.Now().Unix()) + eventer := New() + stop := make(chan struct{}) + recv := make(chan int) + poster := func() { + for { + select { + case <-stop: + return + default: + eventer.Post("test", "hi") + } + } + } + listener := func(i int) { + time.Sleep(time.Duration(rand.Intn(99)) * time.Millisecond) + c := eventer.Register("test") + // wait for the first event + <-c + recv <- i + // keep receiving to prevent deadlock + for { + select { + case <-stop: + return + case <-c: + } + } + } + + nlisteners := 200 + go poster() + for i := 0; i < nlisteners; i++ { + go listener(i) + } + // wait until everyone has been served + for i := 0; i < nlisteners; i++ { + <-recv + } + close(stop) +} |