neo-go/_pkg.dev/wire/payload/mversion_test.go

package payload

import (
	"bytes"
	"math/rand"
	"net"
	"testing"
	"time"

	"github.com/CityOfZion/neo-go/pkg/wire/protocol"
	"github.com/stretchr/testify/assert"
)

func TestValidNewVersionMessage(t *testing.T) {

	expectedIP := "127.0.0.1"
	expectedPort := 8333
	tcpAddrMe := &net.TCPAddr{IP: net.ParseIP(expectedIP), Port: expectedPort}
	nonce := randRange(12949672, 42949672)
	message, err := NewVersionMessage(tcpAddrMe, 0, true, protocol.DefaultVersion, protocol.UserAgent, nonce, protocol.NodePeerService)

	assert.Equal(t, nil, err)
	assert.Equal(t, expectedIP, message.IP.String())
	assert.Equal(t, uint16(expectedPort), message.Port)
	assert.Equal(t, protocol.DefaultVersion, message.Version)
}
func TestEncode(t *testing.T) {

	expectedIP := "127.0.0.1"
	expectedPort := 8333
	tcpAddrMe := &net.TCPAddr{IP: net.ParseIP(expectedIP), Port: expectedPort}
	nonce := randRange(12949672, 42949672)
	message, err := NewVersionMessage(tcpAddrMe, 0, true, protocol.DefaultVersion, protocol.UserAgent, nonce, protocol.NodePeerService)

	buf := new(bytes.Buffer)
	err = message.EncodePayload(buf)

	assert.Equal(t, nil, err)
	assert.Equal(t, len(message.UserAgent)+minMsgVersionSize, int(buf.Len()))
}
func TestLenIsCorrect(t *testing.T) {

	expectedIP := "127.0.0.1"
	expectedPort := 8333
	tcpAddrMe := &net.TCPAddr{IP: net.ParseIP(expectedIP), Port: expectedPort}
	nonce := randRange(12949672, 42949672)
	message, err := NewVersionMessage(tcpAddrMe, 0, true, protocol.DefaultVersion, protocol.UserAgent, nonce, protocol.NodePeerService)

	buf := new(bytes.Buffer)
	err = message.EncodePayload(buf)
	assert.Equal(t, nil, err)

	assert.Equal(t, len(message.UserAgent)+minMsgVersionSize, len(buf.Bytes()))
}

func randRange(min, max int) uint32 {
	rand.Seed(time.Now().Unix() + int64(rand.Uint64()))
	return uint32(rand.Intn(max-min) + min)
}