fix: broken test

pactus-project · Aug 30, 2023 · 7833c7b · 7833c7b
1 parent 06ae7f6
commit 7833c7b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sync/handler_hello_test.go b/sync/handler_hello_test.go
@@ -106,7 +106,7 @@ func TestParsingHelloMessages(t *testing.T) {
 func TestSendingHelloMessage(t *testing.T) {
 	td := setup(t, nil)
 
-	to := td.RandomPeerID()
+	to := td.RandPeerID()
 	td.sync.sayHello(true, to)
 
 	bdl := td.shouldPublishMessageWithThisType(t, td.network, message.TypeHello)

diff --git a/sync/peerset/peer_set_test.go b/sync/peerset/peer_set_test.go
@@ -63,11 +63,11 @@ func TestPeerSet(t *testing.T) {
 	})
 
 	t.Run("Testing GetPeer", func(t *testing.T) {
-		p := peerSet.GetPeer(peer.ID("connected"))
+		p := peerSet.GetPeer(pid2)
 		assert.Equal(t, pid2, p.PeerID)
 		assert.Equal(t, StatusCodeKnown, p.Status)
 
-		p = peerSet.GetPeer(peer.ID("unknown"))
+		p = peerSet.GetPeer(ts.RandPeerID())
 		assert.Equal(t, peer.ID(""), p.PeerID)
 		assert.Equal(t, StatusCodeUnknown, p.Status)
 	})