TrueCloudLab/neoneo-go
7
0
Fork 2
Code Issues Pull requests Projects Releases Packages Wiki Activity

wip refact 2

Browse source
This commit is contained in:
anthdm 2018-01-27 16:00:28 +01:00
parent ccaaf07dad
commit 754a473488
9 changed files with 362 additions and 177 deletions
Download patch file Download diff file Expand all files Collapse all files

96
pkg/network/message_test.go
View file

@ -2,30 +2,31 @@ package network
import (
"bytes"
"encoding/binary"
"reflect"
"testing"
"github.com/anthdm/neo-go/pkg/network/payload"
)
func TestNewMessage(t *testing.T) {
payload := []byte{}
m := newMessage(ModeTestNet, cmdVersion, 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)
}
// 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)
}
}
// 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, []byte{})
m := newMessage(ModeTestNet, cmdVersion, nil)
buf := &bytes.Buffer{}
if err := m.encode(buf); err != nil {
@ -48,34 +49,53 @@ func TestMessageEncodeDecode(t *testing.T) {
}
}
func TestMessageInvalidChecksum(t *testing.T) {
m := newMessage(ModeTestNet, cmdVersion, []byte{})
m.Checksum = 1337
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())
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 {
m1 := &Message{}
if err := m1.decode(buf); err != nil {
t.Fatal(err)
}
p1 := m1.Payload.(*payload.Version)
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)
}
t.Log(p1)
}
// func TestMessageInvalidChecksum(t *testing.T) {
// m := newMessage(ModeTestNet, cmdVersion, []byte{})
// 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)
// }
// }