mirror of
https://gitlab.com/pulsechaincom/go-pulse.git
synced 2024-12-25 12:57:17 +00:00
eventer: add test for concurrent Post/Register
This test reports the race condition when run using "go test -race".
This commit is contained in:
parent
7c9508ed71
commit
d451269977
@ -1,6 +1,10 @@
|
||||
package eventer
|
||||
|
||||
import "testing"
|
||||
import (
|
||||
"math/rand"
|
||||
"testing"
|
||||
"time"
|
||||
)
|
||||
|
||||
func TestChannel(t *testing.T) {
|
||||
eventer := New()
|
||||
@ -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)
|
||||
}
|
||||
|
Loading…
Reference in New Issue
Block a user