mirror of
https://github.com/nspcc-dev/neo-go.git
synced 2024-10-17 09:46:24 +00:00
5bf00db2c9
The logic here is that we'll have all binary encoding/decoding done via our io package, which simplifies error handling. This functionality doesn't belong to util, so it's moved. This also expands BufBinWriter with Reset() method to fit the needs of core package.
35 lines
915 B
Go
35 lines
915 B
Go
package payload
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/CityOfZion/neo-go/pkg/io"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestVersionEncodeDecode(t *testing.T) {
|
|
var port uint16 = 3000
|
|
var id uint32 = 13337
|
|
useragent := "/NEO:0.0.1/"
|
|
var height uint32 = 100500
|
|
var relay = true
|
|
|
|
version := NewVersion(id, port, useragent, height, relay)
|
|
|
|
buf := io.NewBufBinWriter()
|
|
err := version.EncodeBinary(buf.BinWriter)
|
|
assert.Nil(t, err)
|
|
b := buf.Bytes()
|
|
assert.Equal(t, int(version.Size()), len(b))
|
|
|
|
r := io.NewBinReaderFromBuf(b)
|
|
versionDecoded := &Version{}
|
|
err = versionDecoded.DecodeBinary(r)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, versionDecoded.Nonce, id)
|
|
assert.Equal(t, versionDecoded.Port, port)
|
|
assert.Equal(t, versionDecoded.UserAgent, []byte(useragent))
|
|
assert.Equal(t, versionDecoded.StartHeight, height)
|
|
assert.Equal(t, versionDecoded.Relay, relay)
|
|
assert.Equal(t, version, versionDecoded)
|
|
}
|