Race test for pkg/pubsub package

Signed-off-by: Alexander Morozov <lk4d4@docker.com> Upstream-commit: 8aa7ba731a954ffb626c2a07f524c8116ab3f212 Component: engine
2015-07-13 09:05:47 -07:00
parent 71018bb017
commit 9a60e47a81
1 changed files with 20 additions and 0 deletions
@@ -96,6 +96,26 @@ func newTestSubscriber(p *Publisher) *testSubscriber {
 	return ts
 }

+// for testing with -race
+func TestPubSubRace(t *testing.T) {
+	p := NewPublisher(0, 1024)
+	var subs [](*testSubscriber)
+	for j := 0; j < 50; j++ {
+		subs = append(subs, newTestSubscriber(p))
+	}
+	for j := 0; j < 1000; j++ {
+		p.Publish(sampleText)
+	}
+	time.AfterFunc(1*time.Second, func() {
+		for _, s := range subs {
+			p.Evict(s.dataCh)
+		}
+	})
+	for _, s := range subs {
+		s.Wait()
+	}
+}
+
 func BenchmarkPubSub(b *testing.B) {
 	for i := 0; i < b.N; i++ {
 		b.StopTimer()