TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests 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) { func TestNewMessage(t *testing.T) {
payload := []byte{} payload := []byte{}
m := newMessage(ModeTestNet, "version", payload) m := newMessage(ModeTestNet, cmdVersion, payload)
if have, want := m.Length, uint32(0); want != have { if have, want := m.Length, uint32(0); want != have {
t.Errorf("want %d have %d", want, have) t.Errorf("want %d have %d", want, have)
@ -25,13 +25,20 @@ func TestNewMessage(t *testing.T) {
} }
} }
func TestMessageEncodeDecode(t *testing.T) { func TestMessageEncodeDecode(t *testing.T) {
m := newMessage(ModeTestNet, "version", []byte{}) m := newMessage(ModeTestNet, cmdVersion, []byte{})
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
if err := m.encode(buf); err != nil { if err := m.encode(buf); err != nil {
t.Error(err) 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{} md := &Message{}
if err := md.decode(buf); err != nil { if err := md.decode(buf); err != nil {
t.Error(err) t.Error(err)
@ -42,7 +49,7 @@ func TestMessageEncodeDecode(t *testing.T) {
} }
func TestMessageInvalidChecksum(t *testing.T) { func TestMessageInvalidChecksum(t *testing.T) {
m := newMessage(ModeTestNet, "version", []byte{}) m := newMessage(ModeTestNet, cmdVersion, []byte{})
m.Checksum = 1337 m.Checksum = 1337
buf := &bytes.Buffer{} buf := &bytes.Buffer{}
@ -57,7 +64,8 @@ func TestMessageInvalidChecksum(t *testing.T) {
} }
func TestNewVersionPayload(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() b, err := p.encode()
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)