prysm-pulse/beacon-chain/p2p/encoder/varint_test.go
Preston Van Loon 7cc32c4dda
Various code inspection resolutions (#7438)
* remove unused code

* remove defer use in loop

* Remove unused methods and constants

* gofmt and gaz

* nilness check

* remove unused args

* Add TODO for refactoring subscribeWithBase to remove unused arg. It seems too involved to include in this sweeping PR. https://github.com/prysmaticlabs/prysm/issues/7437

* replace empty slice declaration

* Remove unnecessary type conversions

* remove redundant type declaration

* rename receivers to be consistent

* Remove bootnode query tool. It is now obsolete by discv5

* Remove relay node. It is no longer used or supported

* Revert "Remove relay node. It is no longer used or supported"

This reverts commit 4bd7717334dad85ef4766ed9bc4da711fb5fa810.

* Delete unused test directory

* Delete unsupported gcp startup script

* Delete old k8s script

* build fixes

* fix build

* go mod tidy

* revert slasher/db/kv/block_header.go

* fix build

* remove redundant nil check

* combine func args

Co-authored-by: prylabs-bulldozer[bot] <58059840+prylabs-bulldozer[bot]@users.noreply.github.com>
Co-authored-by: Victor Farazdagi <simple.square@gmail.com>
2020-10-12 08:11:05 +00:00

39 lines
1.1 KiB
Go

package encoder
import (
"bytes"
"testing"
"github.com/gogo/protobuf/proto"
"github.com/prysmaticlabs/prysm/shared/testutil/assert"
"github.com/prysmaticlabs/prysm/shared/testutil/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")
}