diff options
Diffstat (limited to 'pkg/p2p/network_test.go')
| -rw-r--r-- | pkg/p2p/network_test.go | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/pkg/p2p/network_test.go b/pkg/p2p/network_test.go index faad9d4..fd5dde0 100644 --- a/pkg/p2p/network_test.go +++ b/pkg/p2p/network_test.go @@ -13,7 +13,7 @@ func TestPeerToPeerCommunication(t *testing.T) { // Create a mock of the first peer (peer-1) peer1Opts := TCPNetworkOpts{ ListenAddr: ":9001", - HandshakeFn: func() error { return nil }, + HandshakeFn: DefaultHandshake, RetryDelay: time.Second * 2, Logger: zap.L(), } @@ -26,7 +26,7 @@ func TestPeerToPeerCommunication(t *testing.T) { // Create a mock of the second peer (peer-2) peer2Opts := TCPNetworkOpts{ ListenAddr: ":9002", - HandshakeFn: func() error { return nil }, + HandshakeFn: DefaultHandshake, RetryDelay: time.Second * 2, Logger: zap.L(), OnReceiveFn: func(msg Message) { |