25 lines
497 B
Go
25 lines
497 B
Go
|
package payload
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/CityOfZion/neo-go/pkg/wire/util/Checksum"
|
||
|
|
|
||
|
|
"github.com/CityOfZion/neo-go/pkg/wire/command"
|
||
|
|
"github.com/stretchr/testify/assert"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestNewGetAddr(t *testing.T) {
|
||
|
|
|
||
|
|
getAddrMessage, err := NewGetAddrMessage()
|
||
|
|
assert.Equal(t, nil, err)
|
||
|
|
|
||
|
|
assert.Equal(t, command.GetAddr, getAddrMessage.Command())
|
||
|
|
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
|
||
|
|
assert.Equal(t, int(3806393949), int(checksum.FromBuf(buf)))
|
||
|
|
assert.Equal(t, int(0), len(buf.Bytes()))
|
||
|
|
}
|