prysm-pulse/beacon-chain/p2p/encoder/varint_test.go
terencechain d17996f8b0
Update to V4 🚀 (#12134)
* Update V3 from V4

* Fix build v3 -> v4

* Update ssz

* Update beacon_chain.pb.go

* Fix formatter import

* Update update-mockgen.sh comment to v4

* Fix conflicts. Pass build and tests

* Fix test
2023-03-17 18:52:56 +00:00

39 lines
1.1 KiB
Go

package encoder
import (
"bytes"
"testing"
"github.com/gogo/protobuf/proto"
"github.com/prysmaticlabs/prysm/v4/testing/assert"
"github.com/prysmaticlabs/prysm/v4/testing/require"
)
func TestReadVarint(t *testing.T) {
data := []byte("foobar data")
prefixedData := append(proto.EncodeVarint(uint64(len(data))), data...)
vi, err := readVarint(bytes.NewBuffer(prefixedData))
require.NoError(t, err)
assert.Equal(t, uint64(len(data)), vi, "Received wrong varint")
}
func TestReadVarint_ExceedsMaxLength(t *testing.T) {
fByte := byte(1 << 7)
// Terminating byte.
tByte := byte(1 << 6)
var header []byte
for i := 0; i < 9; i++ {
header = append(header, fByte)
}
header = append(header, tByte)
_, err := readVarint(bytes.NewBuffer(header))
require.NoError(t, err, "Expected no error from reading valid header")
length := len(header)
// Add an additional byte to make header invalid.
header = append(header[:length-1], []byte{fByte, tByte}...)
_, err = readVarint(bytes.NewBuffer(header))
assert.ErrorContains(t, errExcessMaxLength.Error(), err, "Expected error from reading invalid header")
}