From f883eb88caa258b3a2b6ccf4fd67de7a62024941 Mon Sep 17 00:00:00 2001 From: Tsachi Herman Date: Thu, 24 Feb 2022 10:16:14 -0500 Subject: [PATCH] fix test --- agreement/pseudonode_test.go | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/agreement/pseudonode_test.go b/agreement/pseudonode_test.go index b4209ffdee..dfb9f2ce1f 100644 --- a/agreement/pseudonode_test.go +++ b/agreement/pseudonode_test.go @@ -520,11 +520,12 @@ func TestPseudonodeFailedEnqueuedTasks(t *testing.T) { for i := 0; i < pseudonodeVerificationBacklog*2; i++ { ch, err = pb.MakeProposals(context.Background(), startRound, period(i)) if err != nil { - require.Subset(t, []int{pseudonodeVerificationBacklog, pseudonodeVerificationBacklog + 1}, []int{i}) + require.ErrorAs(t, errPseudonodeBacklogFull, &err) break } channels = append(channels, ch) } + enqueuedProposals := len(channels) require.Error(t, err, "MakeProposals did not returned an error when being overflowed with requests") require.True(t, errors.Is(err, errPseudonodeBacklogFull)) @@ -533,17 +534,17 @@ func TestPseudonodeFailedEnqueuedTasks(t *testing.T) { for i := 0; i < pseudonodeVerificationBacklog*2; i++ { ch, err = pb.MakeVotes(context.Background(), startRound, period(i), step(i%5), makeProposalValue(period(i), accounts[0].Address()), persist) if err != nil { - require.Subset(t, []int{pseudonodeVerificationBacklog, pseudonodeVerificationBacklog + 1}, []int{i}) + require.ErrorAs(t, errPseudonodeBacklogFull, &err) break } channels = append(channels, ch) } require.Error(t, err, "MakeVotes did not returned an error when being overflowed with requests") - + enqueuedVotes := len(channels) - enqueuedProposals // drain output channels. for _, ch := range channels { drainChannel(ch) } - require.Equal(t, 330, subStrLogger.instancesFound[0]) - require.Equal(t, 330, subStrLogger.instancesFound[1]) + require.Equal(t, enqueuedVotes*len(accounts), subStrLogger.instancesFound[0]) + require.Equal(t, enqueuedProposals*len(accounts), subStrLogger.instancesFound[1]) }