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]) }