mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-11-26 19:42:23 +00:00
101 lines
2.3 KiB
Go
101 lines
2.3 KiB
Go
package network
|
|
|
|
import (
|
|
"bytes"
|
|
"reflect"
|
|
"testing"
|
|
|
|
"github.com/anthdm/neo-go/pkg/network/payload"
|
|
)
|
|
|
|
// func TestNewMessage(t *testing.T) {
|
|
// payload := []byte{}
|
|
// m := newMessage(ModeTestNet, cmdVersion, payload)
|
|
|
|
// if have, want := m.Length, uint32(0); want != have {
|
|
// t.Errorf("want %d have %d", want, have)
|
|
// }
|
|
// if have, want := len(m.Command), 12; want != have {
|
|
// t.Errorf("want %d have %d", want, have)
|
|
// }
|
|
|
|
// sum := sumSHA256(sumSHA256(payload))[:4]
|
|
// sumuint32 := binary.LittleEndian.Uint32(sum)
|
|
// if have, want := m.Checksum, sumuint32; want != have {
|
|
// t.Errorf("want %d have %d", want, have)
|
|
// }
|
|
// }
|
|
func TestMessageEncodeDecode(t *testing.T) {
|
|
m := newMessage(ModeTestNet, cmdVersion, nil)
|
|
|
|
buf := &bytes.Buffer{}
|
|
if err := m.encode(buf); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
if n := len(buf.Bytes()); n < minMessageSize {
|
|
t.Fatalf("message should be at least %d bytes got %d", minMessageSize, n)
|
|
}
|
|
if n := len(buf.Bytes()); n > minMessageSize {
|
|
t.Fatalf("message without a payload should be exact %d bytes got %d", minMessageSize, n)
|
|
}
|
|
|
|
md := &Message{}
|
|
if err := md.decode(buf); err != nil {
|
|
t.Error(err)
|
|
}
|
|
if !reflect.DeepEqual(m, md) {
|
|
t.Errorf("both messages should be equal: %v != %v", m, md)
|
|
}
|
|
}
|
|
|
|
func TestMessageEncodeDecodeWithVersion(t *testing.T) {
|
|
p := payload.NewVersion(2000, "/neo/", 0, true)
|
|
m := newMessage(ModeTestNet, cmdVersion, p)
|
|
|
|
buf := &bytes.Buffer{}
|
|
if err := m.encode(buf); err != nil {
|
|
t.Error(err)
|
|
}
|
|
t.Log(buf.Len())
|
|
|
|
m1 := &Message{}
|
|
if err := m1.decode(buf); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
p1 := m1.Payload.(*payload.Version)
|
|
|
|
t.Log(p1)
|
|
}
|
|
|
|
func TestMessageInvalidChecksum(t *testing.T) {
|
|
m := newMessage(ModeTestNet, cmdVersion, nil)
|
|
m.Checksum = 1337
|
|
|
|
buf := &bytes.Buffer{}
|
|
if err := m.encode(buf); err != nil {
|
|
t.Error(err)
|
|
}
|
|
|
|
md := &Message{}
|
|
if err := md.decode(buf); err == nil {
|
|
t.Error("decode should failed with checkum mismatch error")
|
|
}
|
|
}
|
|
|
|
// func TestNewVersionPayload(t *testing.T) {
|
|
// ua := "/neo/0.0.1/"
|
|
// p := newVersionPayload(3000, ua, 0, true)
|
|
// b, err := p.encode()
|
|
// if err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
|
|
// pd := &Version{}
|
|
// if err := pd.decode(b); err != nil {
|
|
// t.Fatal(err)
|
|
// }
|
|
// if !reflect.DeepEqual(p, pd) {
|
|
// t.Errorf("both payloads should be equal: %v != %v", p, pd)
|
|
// }
|
|
// }
|