mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-25 12:57:18 +00:00
8d234014a4
* add two types of encoding/decoding ssz * fix tests * lint * lint
63 lines
1.5 KiB
Go
63 lines
1.5 KiB
Go
package encoder_test
|
|
|
|
import (
|
|
"bytes"
|
|
"testing"
|
|
|
|
"github.com/gogo/protobuf/proto"
|
|
"github.com/prysmaticlabs/prysm/beacon-chain/p2p/encoder"
|
|
testpb "github.com/prysmaticlabs/prysm/proto/testing"
|
|
)
|
|
|
|
func TestSszNetworkEncoder_RoundTrip(t *testing.T) {
|
|
e := &encoder.SszNetworkEncoder{UseSnappyCompression: false}
|
|
testRoundTrip(t, e)
|
|
testRoundTripWithLength(t, e)
|
|
}
|
|
|
|
func TestSszNetworkEncoder_RoundTrip_Snappy(t *testing.T) {
|
|
e := &encoder.SszNetworkEncoder{UseSnappyCompression: true}
|
|
testRoundTrip(t, e)
|
|
testRoundTripWithLength(t, e)
|
|
}
|
|
|
|
func testRoundTrip(t *testing.T, e *encoder.SszNetworkEncoder) {
|
|
buf := new(bytes.Buffer)
|
|
msg := &testpb.TestSimpleMessage{
|
|
Foo: []byte("fooooo"),
|
|
Bar: 9001,
|
|
}
|
|
_, err := e.Encode(buf, msg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
decoded := &testpb.TestSimpleMessage{}
|
|
if err := e.Decode(buf.Bytes(), decoded); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !proto.Equal(decoded, msg) {
|
|
t.Logf("decoded=%+v\n", decoded)
|
|
t.Error("Decoded message is not the same as original")
|
|
}
|
|
}
|
|
|
|
func testRoundTripWithLength(t *testing.T, e *encoder.SszNetworkEncoder) {
|
|
buf := new(bytes.Buffer)
|
|
msg := &testpb.TestSimpleMessage{
|
|
Foo: []byte("fooooo"),
|
|
Bar: 9001,
|
|
}
|
|
_, err := e.EncodeWithLength(buf, msg)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
decoded := &testpb.TestSimpleMessage{}
|
|
if err := e.DecodeWithLength(buf, decoded); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !proto.Equal(decoded, msg) {
|
|
t.Logf("decoded=%+v\n", decoded)
|
|
t.Error("Decoded message is not the same as original")
|
|
}
|
|
}
|