neoneo-go/pkg/network/message_test.go

82 lines
1.8 KiB
Go
Raw Normal View History

2018-01-26 18:04:13 +00:00
package network
import (
"bytes"
"encoding/binary"
"reflect"
"testing"
)
func TestNewMessage(t *testing.T) {
payload := []byte{}
2018-01-26 20:43:07 +00:00
m := newMessage(ModeTestNet, cmdVersion, payload)
2018-01-26 18:04:13 +00:00
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) {
2018-01-26 20:43:07 +00:00
m := newMessage(ModeTestNet, cmdVersion, []byte{})
2018-01-26 18:04:13 +00:00
buf := &bytes.Buffer{}
if err := m.encode(buf); err != nil {
t.Error(err)
}
2018-01-26 20:43:07 +00:00
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)
}
2018-01-26 18:04:13 +00:00
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 TestMessageInvalidChecksum(t *testing.T) {
2018-01-26 20:43:07 +00:00
m := newMessage(ModeTestNet, cmdVersion, []byte{})
2018-01-26 18:04:13 +00:00
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) {
2018-01-26 20:43:07 +00:00
ua := "/neo/0.0.1/"
p := newVersionPayload(3000, ua, 0, true)
2018-01-26 18:04:13 +00:00
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)
}
}