From 42f60e727bd913a8009e487e142dd8936a173a88 Mon Sep 17 00:00:00 2001
From: "alex.sharov" <AskAlexSharov@gmail.com>
Date: Sat, 3 Jul 2021 13:00:57 +0700
Subject: [PATCH] fix test

---
 p2p/peer_test.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/p2p/peer_test.go b/p2p/peer_test.go
index 76840c4edc..8ba4dda874 100644
--- a/p2p/peer_test.go
+++ b/p2p/peer_test.go
@@ -188,8 +188,8 @@ func TestPeerDisconnect(t *testing.T) {
 	}
 	select {
 	case reason := <-disc:
-		if reason != DiscQuitting {
-			t.Errorf("run returned wrong reason: got %v, want %v", reason, DiscQuitting)
+		if !strings.Contains(reason.Error(), DiscQuitting.Error()) {
+			t.Errorf("run returned wrong reason: got %v, want %v", reason.Error(), DiscQuitting.Error())
 		}
 	case <-time.After(500 * time.Millisecond):
 		t.Error("peer did not return")
-- 
GitLab