mirror of
https://github.com/luxfi/node.git
synced 2026-10-30 15:24:05 +00:00
zeekayandHanzo Dev
37e2940ead
Two packages were named for the same word and meant different things. network/ dialed peers, exchanged keys, throttled and tracked them; nets/ tracked which chains share a validator set and when they have all bootstrapped. One letter apart, and neither name said which was which. The peer layer is mesh — the word this estate already uses for it, in lux-rs/node's own mesh.rs. That frees network for the thing that is actually a network: a set of chains sharing a validator set. A network holds chains; a chain belongs to one network. github.com/luxfi/p2p stays what it is, the gossip library the VMs use. Three concerns, three names, none of them borrowed from another. Co-authored-by: Hanzo Dev <[email protected]>
102 lines
2.2 KiB
Go
102 lines
2.2 KiB
Go
// Copyright (C) 2019-2025, Lux Industries Inc. All rights reserved.
|
|
// See the file LICENSE for licensing terms.
|
|
|
|
package peer
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"github.com/luxfi/constants"
|
|
)
|
|
|
|
func TestWriteMsgLen(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
tt := []struct {
|
|
msgLen uint32
|
|
msgLimit uint32
|
|
expectedErr error
|
|
}{
|
|
{
|
|
msgLen: constants.DefaultMaxMessageSize,
|
|
msgLimit: 1,
|
|
expectedErr: errMaxMessageLengthExceeded,
|
|
},
|
|
{
|
|
msgLen: constants.DefaultMaxMessageSize,
|
|
msgLimit: constants.DefaultMaxMessageSize,
|
|
expectedErr: nil,
|
|
},
|
|
{
|
|
msgLen: 1,
|
|
msgLimit: constants.DefaultMaxMessageSize,
|
|
expectedErr: nil,
|
|
},
|
|
}
|
|
for _, tv := range tt {
|
|
msgLenBytes, err := writeMsgLen(tv.msgLen, tv.msgLimit)
|
|
require.ErrorIs(err, tv.expectedErr)
|
|
if tv.expectedErr != nil {
|
|
continue
|
|
}
|
|
|
|
msgLen, err := readMsgLen(msgLenBytes[:], tv.msgLimit)
|
|
require.NoError(err)
|
|
require.Equal(tv.msgLen, msgLen)
|
|
}
|
|
}
|
|
|
|
func TestReadMsgLen(t *testing.T) {
|
|
require := require.New(t)
|
|
|
|
tt := []struct {
|
|
msgLenBytes []byte
|
|
msgLimit uint32
|
|
expectedErr error
|
|
expectedMsgLen uint32
|
|
}{
|
|
{
|
|
msgLenBytes: []byte{0b11111111, 0xFF},
|
|
msgLimit: math.MaxInt32,
|
|
expectedErr: errInvalidMessageLength,
|
|
expectedMsgLen: 0,
|
|
},
|
|
{
|
|
msgLenBytes: []byte{0xFF, 0xFF, 0xFF, 0xFF},
|
|
msgLimit: constants.DefaultMaxMessageSize,
|
|
expectedErr: errMaxMessageLengthExceeded,
|
|
expectedMsgLen: 0,
|
|
},
|
|
{
|
|
msgLenBytes: []byte{0xFF, 0xFF, 0xFF, 0xFF},
|
|
msgLimit: math.MaxUint32,
|
|
expectedErr: nil,
|
|
expectedMsgLen: math.MaxUint32,
|
|
},
|
|
{
|
|
msgLenBytes: []byte{0x00, 0x00, 0x00, 0x01},
|
|
msgLimit: 10,
|
|
expectedErr: nil,
|
|
expectedMsgLen: 1,
|
|
},
|
|
}
|
|
for _, tv := range tt {
|
|
msgLen, err := readMsgLen(tv.msgLenBytes, tv.msgLimit)
|
|
require.ErrorIs(err, tv.expectedErr)
|
|
if tv.expectedErr != nil {
|
|
continue
|
|
}
|
|
require.Equal(tv.expectedMsgLen, msgLen)
|
|
|
|
msgLenBytes, err := writeMsgLen(msgLen, tv.msgLimit)
|
|
require.NoError(err)
|
|
|
|
msgLenAfterWrite, err := readMsgLen(msgLenBytes[:], tv.msgLimit)
|
|
require.NoError(err)
|
|
require.Equal(tv.expectedMsgLen, msgLenAfterWrite)
|
|
}
|
|
}
|