TrueCloudLab/neo-go
6
0
Fork 0
mirror of https://github.com/nspcc-dev/neo-go.git synced 2024-10-16 05:43:24 +00:00
Code Issues Projects Releases Packages Wiki Activity

extended message test.

Browse source
This commit is contained in:
anthdm 2018-01-26 21:43:07 +01:00
parent 058459c65d
commit 98a6831d6a
1 changed files with 12 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
pkg/network/message_test.go
View file

@ -9,7 +9,7 @@ import (
func TestNewMessage(t *testing.T) {
payload := []byte{}
m := newMessage(ModeTestNet, "version", payload)
m := newMessage(ModeTestNet, cmdVersion, payload)
if have, want := m.Length, uint32(0); want != have {
t.Errorf("want %d have %d", want, have)
@ -25,13 +25,20 @@ func TestNewMessage(t *testing.T) {
}
}
func TestMessageEncodeDecode(t *testing.T) {
m := newMessage(ModeTestNet, "version", []byte{})
m := newMessage(ModeTestNet, cmdVersion, []byte{})
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)
@ -42,7 +49,7 @@ func TestMessageEncodeDecode(t *testing.T) {
}
func TestMessageInvalidChecksum(t *testing.T) {
m := newMessage(ModeTestNet, "version", []byte{})
m := newMessage(ModeTestNet, cmdVersion, []byte{})
m.Checksum = 1337
buf := &bytes.Buffer{}
@ -57,7 +64,8 @@ func TestMessageInvalidChecksum(t *testing.T) {
}
func TestNewVersionPayload(t *testing.T) {
p := newVersionPayload(3000, "/neo/", 0, true)
ua := "/neo/0.0.1/"
p := newVersionPayload(3000, ua, 0, true)
b, err := p.encode()
if err != nil {
t.Fatal(err)