51 lines
1.1 KiB
Go
51 lines
1.1 KiB
Go
package auth
|
|
|
|
import (
|
|
"testing"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/assert"
|
|
crypto "github.com/tendermint/go-crypto"
|
|
wire "github.com/tendermint/go-wire"
|
|
)
|
|
|
|
func TestBaseAccount(t *testing.T) {
|
|
key := crypto.GenPrivKeyEd25519()
|
|
pub := key.PubKey()
|
|
addr := pub.Address()
|
|
someCoins := sdk.Coins{{"atom", 123}, {"eth", 246}}
|
|
seq := int64(7)
|
|
|
|
acc := NewBaseAccountWithAddress(addr)
|
|
|
|
// need a codec for marshaling
|
|
codec := wire.NewCodec()
|
|
crypto.RegisterWire(codec)
|
|
|
|
err := acc.SetPubKey(pub)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, pub, acc.GetPubKey())
|
|
|
|
assert.Equal(t, addr, acc.GetAddress())
|
|
|
|
err = acc.SetCoins(someCoins)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, someCoins, acc.GetCoins())
|
|
|
|
err = acc.SetSequence(seq)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, seq, acc.GetSequence())
|
|
|
|
b, err := codec.MarshalBinary(acc)
|
|
assert.Nil(t, err)
|
|
|
|
var acc2 BaseAccount
|
|
err = codec.UnmarshalBinary(b, &acc2)
|
|
assert.Nil(t, err)
|
|
assert.Equal(t, acc, acc2)
|
|
|
|
acc2 = BaseAccount{}
|
|
err = codec.UnmarshalBinary(b[:len(b)/2], &acc2)
|
|
assert.NotNil(t, err)
|
|
}
|