From 7c5a6176b27b6b0c0c3ef8a4aedbdec871391a80 Mon Sep 17 00:00:00 2001 From: Santo Cariotti Date: Fri, 18 Apr 2025 16:09:32 +0200 Subject: Add type on messages between peers --- pkg/p2p/network_test.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'pkg/p2p/network_test.go') diff --git a/pkg/p2p/network_test.go b/pkg/p2p/network_test.go index fd5dde0..15d0d8e 100644 --- a/pkg/p2p/network_test.go +++ b/pkg/p2p/network_test.go @@ -44,10 +44,10 @@ func TestPeerToPeerCommunication(t *testing.T) { time.Sleep(5 * time.Second) // Send a message from peer-1 to peer-2 - err := peer1.Send("peer-2", []byte("Hey from peer-1!")) + err := peer1.Send("peer-2", []byte("simple-msg"), []byte("Hey from peer-1!")) assert.NoError(t, err) - err = peer2.Send("peer-1", []byte("Hey from peer-2!")) + err = peer2.Send("peer-1", []byte("simple-msg"), []byte("Hey from peer-2!")) assert.NoError(t, err) // Allow some time for the message to be received and handled @@ -74,6 +74,6 @@ func TestSendFailure(t *testing.T) { _ = NewTCPNetwork("peer-2", peer2Opts) // Attempt to send a message without establishing a connection first - err := peer1.Send("peer-2", []byte("Message without connection")) + err := peer1.Send("peer-2", []byte("msg"), []byte("Message without connection")) assert.Error(t, err, "Expected error when sending to a non-connected peer") } -- cgit v1.2.3-18-g5258