neo-go/pkg/network/message_test.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)
// 	}
// }