124 lines
3.3 KiB
Go
124 lines
3.3 KiB
Go
// nolint
|
|
package auth
|
|
|
|
import (
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/crypto"
|
|
"github.com/tendermint/tendermint/crypto/ed25519"
|
|
dbm "github.com/tendermint/tendermint/libs/db"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/store"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/params"
|
|
)
|
|
|
|
type testInput struct {
|
|
cdc *codec.Codec
|
|
ctx sdk.Context
|
|
ak AccountKeeper
|
|
fck FeeCollectionKeeper
|
|
}
|
|
|
|
func setupTestInput() testInput {
|
|
db := dbm.NewMemDB()
|
|
|
|
cdc := codec.New()
|
|
RegisterBaseAccount(cdc)
|
|
|
|
authCapKey := sdk.NewKVStoreKey("authCapKey")
|
|
fckCapKey := sdk.NewKVStoreKey("fckCapKey")
|
|
keyParams := sdk.NewKVStoreKey("params")
|
|
tkeyParams := sdk.NewTransientStoreKey("transient_params")
|
|
|
|
ms := store.NewCommitMultiStore(db)
|
|
ms.MountStoreWithDB(authCapKey, sdk.StoreTypeIAVL, db)
|
|
ms.MountStoreWithDB(fckCapKey, sdk.StoreTypeIAVL, db)
|
|
ms.MountStoreWithDB(keyParams, sdk.StoreTypeIAVL, db)
|
|
ms.MountStoreWithDB(tkeyParams, sdk.StoreTypeTransient, db)
|
|
ms.LoadLatestVersion()
|
|
|
|
pk := params.NewKeeper(cdc, keyParams, tkeyParams)
|
|
ak := NewAccountKeeper(cdc, authCapKey, pk.Subspace(DefaultParamspace), ProtoBaseAccount)
|
|
fck := NewFeeCollectionKeeper(cdc, fckCapKey)
|
|
ctx := sdk.NewContext(ms, abci.Header{ChainID: "test-chain-id"}, false, log.NewNopLogger())
|
|
|
|
ak.SetParams(ctx, DefaultParams())
|
|
|
|
return testInput{cdc: cdc, ctx: ctx, ak: ak, fck: fck}
|
|
}
|
|
|
|
func newTestMsg(addrs ...sdk.AccAddress) *sdk.TestMsg {
|
|
return sdk.NewTestMsg(addrs...)
|
|
}
|
|
|
|
func newStdFee() StdFee {
|
|
return NewStdFee(50000,
|
|
sdk.Coins{sdk.NewInt64Coin("atom", 150)},
|
|
)
|
|
}
|
|
|
|
// coins to more than cover the fee
|
|
func newCoins() sdk.Coins {
|
|
return sdk.Coins{
|
|
sdk.NewInt64Coin("atom", 10000000),
|
|
}
|
|
}
|
|
|
|
func keyPubAddr() (crypto.PrivKey, crypto.PubKey, sdk.AccAddress) {
|
|
key := ed25519.GenPrivKey()
|
|
pub := key.PubKey()
|
|
addr := sdk.AccAddress(pub.Address())
|
|
return key, pub, addr
|
|
}
|
|
|
|
func newTestTx(ctx sdk.Context, msgs []sdk.Msg, privs []crypto.PrivKey, accNums []uint64, seqs []uint64, fee StdFee) sdk.Tx {
|
|
sigs := make([]StdSignature, len(privs))
|
|
for i, priv := range privs {
|
|
signBytes := StdSignBytes(ctx.ChainID(), accNums[i], seqs[i], fee, msgs, "")
|
|
|
|
sig, err := priv.Sign(signBytes)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sigs[i] = StdSignature{PubKey: priv.PubKey(), Signature: sig}
|
|
}
|
|
|
|
tx := NewStdTx(msgs, fee, sigs, "")
|
|
return tx
|
|
}
|
|
|
|
func newTestTxWithMemo(ctx sdk.Context, msgs []sdk.Msg, privs []crypto.PrivKey, accNums []uint64, seqs []uint64, fee StdFee, memo string) sdk.Tx {
|
|
sigs := make([]StdSignature, len(privs))
|
|
for i, priv := range privs {
|
|
signBytes := StdSignBytes(ctx.ChainID(), accNums[i], seqs[i], fee, msgs, memo)
|
|
|
|
sig, err := priv.Sign(signBytes)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sigs[i] = StdSignature{PubKey: priv.PubKey(), Signature: sig}
|
|
}
|
|
|
|
tx := NewStdTx(msgs, fee, sigs, memo)
|
|
return tx
|
|
}
|
|
|
|
func newTestTxWithSignBytes(msgs []sdk.Msg, privs []crypto.PrivKey, accNums []uint64, seqs []uint64, fee StdFee, signBytes []byte, memo string) sdk.Tx {
|
|
sigs := make([]StdSignature, len(privs))
|
|
for i, priv := range privs {
|
|
sig, err := priv.Sign(signBytes)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
sigs[i] = StdSignature{PubKey: priv.PubKey(), Signature: sig}
|
|
}
|
|
|
|
tx := NewStdTx(msgs, fee, sigs, memo)
|
|
return tx
|
|
}
|