Attempt to fix flakey announce test (#112)

uber · Mar 11, 2019 · 51def02 · 51def02
1 parent af2ac4d
commit 51def02
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/lib/torrent/scheduler/announcer/announcer_test.go b/lib/torrent/scheduler/announcer/announcer_test.go
@@ -15,6 +15,7 @@ package announcer
 
 import (
 	"errors"
+	"runtime"
 	"testing"
 	"time"
 
@@ -39,6 +40,7 @@ func newMockEvents() *mockEvents {
 func (e *mockEvents) AnnounceTick() { e.tick <- struct{}{} }
 
 func (e *mockEvents) expectTick(t *testing.T) {
+	runtime.Gosched()
 	select {
 	case <-e.tick:
 	case <-time.After(500 * time.Millisecond):
@@ -47,6 +49,7 @@ func (e *mockEvents) expectTick(t *testing.T) {
 }
 
 func (e *mockEvents) expectNoTick(t *testing.T) {
+	runtime.Gosched()
 	select {
 	case <-e.tick:
 		require.FailNow(t, "Unexpected tick")