From 0c68b891fe4824d79739aa7550af89c01b3d152d Mon Sep 17 00:00:00 2001 From: Waldemar Quevedo Date: Tue, 5 Sep 2023 21:17:22 -0700 Subject: [PATCH] flakes: Fix TestAccountSubjectMapping When it fails it hangs the build. Signed-off-by: Waldemar Quevedo --- test/accounts_cycles_test.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/test/accounts_cycles_test.go b/test/accounts_cycles_test.go index 4c5018a7..71a4ed93 100644 --- a/test/accounts_cycles_test.go +++ b/test/accounts_cycles_test.go @@ -367,6 +367,7 @@ func TestAccountSubjectMapping(t *testing.T) { t.Fatalf("Unexpected error: %v", err) } sub1.AutoUnsubscribe(numMessages * 2) + nc1.Flush() nc2 := clientConnectToServer(t, s) defer nc2.Close() @@ -385,7 +386,12 @@ func TestAccountSubjectMapping(t *testing.T) { partitionsReceived := make([]int, numMessages) for i := 0; i < numMessages; i++ { - subject := <-subjectsReceived + var subject string + select { + case subject = <-subjectsReceived: + case <-time.After(5 * time.Second): + t.Fatal("Timed out waiting for messages") + } sTokens := strings.Split(subject, ".") if err != nil { t.Fatalf("Unexpected error: %v", err)