cosmos-sdk/modules/coin/helper.go

package coin

import (
	crypto "github.com/tendermint/go-crypto"
	"github.com/tendermint/go-wire/data"

	"github.com/tendermint/basecoin"
	"github.com/tendermint/basecoin/modules/auth"
)

// AccountWithKey is a helper for tests, that includes and account
// along with the private key to access it.
type AccountWithKey struct {
	Key      crypto.PrivKey
	Sequence uint32
	Account
}

// NewAccountWithKey creates an account with the given balance
// and a random private key
func NewAccountWithKey(coins Coins) *AccountWithKey {
	return &AccountWithKey{
		Key:     crypto.GenPrivKeyEd25519().Wrap(),
		Account: Account{Coins: coins},
	}
}

// Address returns the public key address for this account
func (a *AccountWithKey) Address() []byte {
	return a.Key.PubKey().Address()
}

// Actor returns the basecoin actor associated with this account
func (a *AccountWithKey) Actor() basecoin.Actor {
	return auth.SigPerm(a.Key.PubKey().Address())
}

// NextSequence returns the next sequence to sign with
func (a *AccountWithKey) NextSequence() uint32 {
	a.Sequence++
	return a.Sequence
}

// MakeOption returns a string to use with SetOption to initialize this account
//
// This is intended for use in test cases
func (a *AccountWithKey) MakeOption() string {
	info := GenesisAccount{
		Address: a.Address(),
		Balance: a.Coins,
	}
	js, err := data.ToJSON(info)
	if err != nil {
		panic(err)
	}
	return string(js)
}