From 2ab624a91786054608cf32985c851d1be5d03a1f Mon Sep 17 00:00:00 2001 From: Sandro Jijavadze Date: Tue, 21 Jun 2022 19:36:20 +0400 Subject: [PATCH] fix flaky test gosched hack Signed-off-by: Sandro Jijavadze --- pkg/relay/relay_test.go | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pkg/relay/relay_test.go b/pkg/relay/relay_test.go index 97883d9..7a2eb6b 100644 --- a/pkg/relay/relay_test.go +++ b/pkg/relay/relay_test.go @@ -15,6 +15,7 @@ package relay import ( "fmt" + "runtime" "testing" "time" @@ -68,6 +69,14 @@ func TestRelay_RelayLine(t *testing.T) { for _, line := range tt.args.lines { r.RelayLine(line) } + + for goSchedTimes := 0; goSchedTimes < 1000; goSchedTimes++ { + if len(r.bufferChannel) == 0 { + break + } + runtime.Gosched() + } + // Tick time forward to trigger a packet send. clock.ClockInstance.Instant = time.Unix(1, 10) clock.ClockInstance.TickerCh <- time.Unix(0, 0)