From 14510c4eb00f4b83758cc2eb0ea3e66a9820a772 Mon Sep 17 00:00:00 2001 From: William Banfield Date: Wed, 19 Oct 2022 17:21:50 -0400 Subject: [PATCH] fix evidence tests --- evidence/reactor_test.go | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/evidence/reactor_test.go b/evidence/reactor_test.go index a2d82bf71..0d7d1110d 100644 --- a/evidence/reactor_test.go +++ b/evidence/reactor_test.go @@ -208,7 +208,10 @@ func TestReactorBroadcastEvidenceMemoryLeak(t *testing.T) { // i.e. broadcastEvidenceRoutine finishes when peer is stopped defer leaktest.CheckTimeout(t, 10*time.Second)() - p.On("Send", evidence.EvidenceChannel, mock.AnythingOfType("[]uint8")).Return(false) + p.On("Send", mock.MatchedBy(func(i interface{}) bool { + e, ok := i.(p2p.Envelope) + return ok && e.ChannelID == evidence.EvidenceChannel + })).Return(false) quitChan := make(<-chan struct{}) p.On("Quit").Return(quitChan) ps := peerState{2}