test: migrate remaining e2e tests to integration tests (#22364)

Co-authored-by: Julien Robert <julien@rbrt.fr>
Co-authored-by: Marko <marko@baricevic.me>
This commit is contained in:
Akhil Kumar P
2024-10-27 14:15:02 +00:00
committed by GitHub
co-authored by Julien Robert Marko
parent fc5536a641
commit e37d71ad24
19 changed files with 61 additions and 101 deletions
@@ -0,0 +1,97 @@
//go:build app_v1
package accounts
import (
"math/rand"
"testing"
gogoproto "github.com/cosmos/gogoproto/proto"
gogoany "github.com/cosmos/gogoproto/types/any"
"cosmossdk.io/simapp"
baseaccountv1 "cosmossdk.io/x/accounts/defaults/base/v1"
"cosmossdk.io/x/bank/testutil"
banktypes "cosmossdk.io/x/bank/types"
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
cryptotypes "github.com/cosmos/cosmos-sdk/crypto/types"
"github.com/cosmos/cosmos-sdk/testutil/sims"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/stretchr/testify/require"
)
func TestBaseAccount(t *testing.T) {
app := setupApp(t)
ak := app.AccountsKeeper
ctx := sdk.NewContext(app.CommitMultiStore(), false, app.Logger())
_, baseAccountAddr, err := ak.Init(ctx, "base", accCreator, &baseaccountv1.MsgInit{
PubKey: toAnyPb(t, privKey.PubKey()),
}, nil)
require.NoError(t, err)
// fund base account! this will also cause an auth base account to be created
// by the bank module.
// TODO: fixed by letting x/auth rely on x/accounts for acc existence checks.
fundAccount(t, app, ctx, baseAccountAddr, "1000000stake")
// now we make the account send a tx, public key not present.
// so we know it will default to x/accounts calling.
msg := &banktypes.MsgSend{
FromAddress: bechify(t, app, baseAccountAddr),
ToAddress: bechify(t, app, []byte("random-addr")),
Amount: coins(t, "100stake"),
}
sendTx(t, ctx, app, baseAccountAddr, msg)
}
func sendTx(t *testing.T, ctx sdk.Context, app *simapp.SimApp, sender []byte, msg sdk.Msg) {
tx := sign(t, ctx, app, sender, privKey, msg)
_, _, err := app.SimDeliver(app.TxEncode, tx)
require.NoError(t, err)
}
func sign(t *testing.T, ctx sdk.Context, app *simapp.SimApp, from sdk.AccAddress, privKey cryptotypes.PrivKey, msg sdk.Msg) sdk.Tx {
r := rand.New(rand.NewSource(0))
accNum, err := app.AccountsKeeper.AccountByNumber.Get(ctx, from)
require.NoError(t, err)
accSeq, err := app.AccountsKeeper.Query(ctx, from, &baseaccountv1.QuerySequence{})
require.NoError(t, err)
tx, err := sims.GenSignedMockTx(
r,
app.TxConfig(),
[]sdk.Msg{msg},
coins(t, "100stake"),
1_000_000,
app.ChainID(),
[]uint64{accNum},
[]uint64{accSeq.(*baseaccountv1.QuerySequenceResponse).Sequence},
privKey,
)
require.NoError(t, err)
return tx
}
func bechify(t *testing.T, app *simapp.SimApp, addr []byte) string {
bech32, err := app.AuthKeeper.AddressCodec().BytesToString(addr)
require.NoError(t, err)
return bech32
}
func fundAccount(t *testing.T, app *simapp.SimApp, ctx sdk.Context, addr sdk.AccAddress, amt string) {
require.NoError(t, testutil.FundAccount(ctx, app.BankKeeper, addr, coins(t, amt)))
}
func toAnyPb(t *testing.T, pm gogoproto.Message) *codectypes.Any {
t.Helper()
if gogoproto.MessageName(pm) == gogoproto.MessageName(&gogoany.Any{}) {
t.Fatal("no")
}
pb, err := codectypes.NewAnyWithValue(pm)
require.NoError(t, err)
return pb
}
@@ -0,0 +1,203 @@
package lockup
import (
"testing"
"time"
"github.com/stretchr/testify/require"
"cosmossdk.io/collections"
"cosmossdk.io/core/header"
"cosmossdk.io/math"
lockupaccount "cosmossdk.io/x/accounts/defaults/lockup"
types "cosmossdk.io/x/accounts/defaults/lockup/v1"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
func (s *IntegrationTestSuite) TestContinuousLockingAccount() {
t := s.T()
app := setupApp(t)
currentTime := time.Now()
ctx := sdk.NewContext(app.CommitMultiStore(), false, app.Logger()).WithHeaderInfo(header.Info{
Time: currentTime,
})
ownerAddrStr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
s.fundAccount(app, ctx, accOwner, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000000))})
randAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
withdrawAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
_, accountAddr, err := app.AccountsKeeper.Init(ctx, lockupaccount.CONTINUOUS_LOCKING_ACCOUNT, accOwner, &types.MsgInitLockupAccount{
Owner: ownerAddrStr,
StartTime: currentTime,
// end time in 1 minutes
EndTime: currentTime.Add(time.Minute),
}, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000))})
require.NoError(t, err)
addr, err := app.AuthKeeper.AddressCodec().BytesToString(randAcc)
require.NoError(t, err)
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
t.Run("error - execute message, wrong sender", func(t *testing.T) {
msg := &types.MsgSend{
Sender: addr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("error - execute send message, insufficient fund", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("error - execute withdraw message, no withdrawable token", func(t *testing.T) {
ownerAddr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
withdrawAddr, err := app.AuthKeeper.AddressCodec().BytesToString(withdrawAcc)
require.NoError(t, err)
msg := &types.MsgWithdraw{
Withdrawer: ownerAddr,
ToAddress: withdrawAddr,
Denoms: []string{"stake"},
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
// Update context time
// 12 sec = 1/5 of a minute so 200stake should be released
ctx = ctx.WithHeaderInfo(header.Info{
Time: currentTime.Add(time.Second * 12),
})
// Check if token is sendable
t.Run("ok - execute send message", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
balance := app.BankKeeper.GetBalance(ctx, randAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(100)))
})
t.Run("ok - execute withdraw message", func(t *testing.T) {
ownerAddr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
withdrawAddr, err := app.AuthKeeper.AddressCodec().BytesToString(withdrawAcc)
require.NoError(t, err)
msg := &types.MsgWithdraw{
Withdrawer: ownerAddr,
ToAddress: withdrawAddr,
Denoms: []string{"stake"},
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
// withdrawable amount should be 200 - 100 = 100stake
balance := app.BankKeeper.GetBalance(ctx, withdrawAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(100)))
})
t.Run("ok - execute delegate message", func(t *testing.T) {
msg := &types.MsgDelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
del, err := app.StakingKeeper.Delegations.Get(
ctx, collections.Join(sdk.AccAddress(accountAddr), sdk.ValAddress(valbz)),
)
require.NoError(t, err)
require.NotNil(t, del)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.NewInt(100)))
})
t.Run("ok - execute withdraw reward message", func(t *testing.T) {
msg := &types.MsgWithdrawReward{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
})
t.Run("ok - execute undelegate message", func(t *testing.T) {
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
msg := &types.MsgUndelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
ubd, err := app.StakingKeeper.GetUnbondingDelegation(
ctx, sdk.AccAddress(accountAddr), sdk.ValAddress(valbz),
)
require.NoError(t, err)
require.Equal(t, len(ubd.Entries), 1)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.ZeroInt()))
})
// Update context time to end time
ctx = ctx.WithHeaderInfo(header.Info{
Time: currentTime.Add(time.Minute),
})
// test if tracking delegate work perfectly
t.Run("ok - execute delegate message", func(t *testing.T) {
msg := &types.MsgDelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
del, err := app.StakingKeeper.Delegations.Get(
ctx, collections.Join(sdk.AccAddress(accountAddr), sdk.ValAddress(valbz)),
)
require.NoError(t, err)
require.NotNil(t, del)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.ZeroInt()))
delFree := lockupAccountInfoResponse.DelegatedFree
require.True(t, delFree.AmountOf("stake").Equal(math.NewInt(100)))
})
}
@@ -0,0 +1,172 @@
package lockup
import (
"testing"
"time"
"github.com/stretchr/testify/require"
"cosmossdk.io/collections"
"cosmossdk.io/core/header"
"cosmossdk.io/math"
lockupaccount "cosmossdk.io/x/accounts/defaults/lockup"
types "cosmossdk.io/x/accounts/defaults/lockup/v1"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
func (s *IntegrationTestSuite) TestDelayedLockingAccount() {
t := s.T()
app := setupApp(t)
currentTime := time.Now()
ctx := sdk.NewContext(app.CommitMultiStore(), false, app.Logger()).WithHeaderInfo(header.Info{
Time: currentTime,
})
ownerAddrStr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
s.fundAccount(app, ctx, accOwner, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000000))})
randAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
withdrawAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
_, accountAddr, err := app.AccountsKeeper.Init(ctx, lockupaccount.DELAYED_LOCKING_ACCOUNT, accOwner, &types.MsgInitLockupAccount{
Owner: ownerAddrStr,
// end time in 1 minutes
EndTime: currentTime.Add(time.Minute),
}, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000))})
require.NoError(t, err)
addr, err := app.AuthKeeper.AddressCodec().BytesToString(randAcc)
require.NoError(t, err)
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
t.Run("error - execute message, wrong sender", func(t *testing.T) {
msg := &types.MsgSend{
Sender: addr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("error - execute send message, insufficient fund", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("error - execute withdraw message, no withdrawable token", func(t *testing.T) {
ownerAddr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
withdrawAddr, err := app.AuthKeeper.AddressCodec().BytesToString(withdrawAcc)
require.NoError(t, err)
msg := &types.MsgWithdraw{
Withdrawer: ownerAddr,
ToAddress: withdrawAddr,
Denoms: []string{"stake"},
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("ok - execute delegate message", func(t *testing.T) {
msg := &types.MsgDelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
del, err := app.StakingKeeper.Delegations.Get(
ctx, collections.Join(sdk.AccAddress(accountAddr), sdk.ValAddress(valbz)),
)
require.NoError(t, err)
require.NotNil(t, del)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.NewInt(100)))
})
t.Run("ok - execute withdraw reward message", func(t *testing.T) {
msg := &types.MsgWithdrawReward{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
})
t.Run("ok - execute undelegate message", func(t *testing.T) {
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
msg := &types.MsgUndelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
ubd, err := app.StakingKeeper.GetUnbondingDelegation(
ctx, sdk.AccAddress(accountAddr), sdk.ValAddress(valbz),
)
require.NoError(t, err)
require.Equal(t, len(ubd.Entries), 1)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.ZeroInt()))
})
// Update context time
// After endtime fund should be unlock
ctx = ctx.WithHeaderInfo(header.Info{
Time: currentTime.Add(time.Second * 61),
})
// Check if token is sendable after unlock
t.Run("ok - execute send message", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
balance := app.BankKeeper.GetBalance(ctx, randAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(100)))
})
// Test to withdraw all the remain funds to an account of choice
t.Run("ok - execute withdraw message", func(t *testing.T) {
ownerAddr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
withdrawAddr, err := app.AuthKeeper.AddressCodec().BytesToString(withdrawAcc)
require.NoError(t, err)
msg := &types.MsgWithdraw{
Withdrawer: ownerAddr,
ToAddress: withdrawAddr,
Denoms: []string{"stake"},
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
// withdrawable amount should be
// 1000stake - 100stake( above sent amt ) - 100stake(above delegate amt) = 800stake
balance := app.BankKeeper.GetBalance(ctx, withdrawAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(800)))
})
}
@@ -0,0 +1,11 @@
package lockup
import (
"testing"
"github.com/stretchr/testify/suite"
)
func TestIntegrationTestSuite(t *testing.T) {
suite.Run(t, NewIntegrationTestSuite())
}
@@ -0,0 +1,223 @@
package lockup
import (
"testing"
"time"
"github.com/stretchr/testify/require"
"cosmossdk.io/collections"
"cosmossdk.io/core/header"
"cosmossdk.io/math"
lockupaccount "cosmossdk.io/x/accounts/defaults/lockup"
types "cosmossdk.io/x/accounts/defaults/lockup/v1"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
func (s *IntegrationTestSuite) TestPeriodicLockingAccount() {
t := s.T()
app := setupApp(t)
currentTime := time.Now()
ctx := sdk.NewContext(app.CommitMultiStore(), false, app.Logger()).WithHeaderInfo(header.Info{
Time: currentTime,
})
ownerAddrStr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
s.fundAccount(app, ctx, accOwner, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000000))})
randAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
withdrawAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
_, accountAddr, err := app.AccountsKeeper.Init(ctx, lockupaccount.PERIODIC_LOCKING_ACCOUNT, accOwner, &types.MsgInitPeriodicLockingAccount{
Owner: ownerAddrStr,
StartTime: currentTime,
LockingPeriods: []types.Period{
{
Amount: sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(500))),
Length: time.Minute,
},
{
Amount: sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(500))),
Length: time.Minute,
},
{
Amount: sdk.NewCoins(sdk.NewCoin("stake", math.NewInt(500))),
Length: time.Minute,
},
},
}, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1500))})
require.NoError(t, err)
addr, err := app.AuthKeeper.AddressCodec().BytesToString(randAcc)
require.NoError(t, err)
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
t.Run("error - execute message, wrong sender", func(t *testing.T) {
msg := &types.MsgSend{
Sender: addr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
// No token being unlocked yet
t.Run("error - execute send message, insufficient fund", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("error - execute withdraw message, no withdrawable token", func(t *testing.T) {
ownerAddr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
withdrawAddr, err := app.AuthKeeper.AddressCodec().BytesToString(withdrawAcc)
require.NoError(t, err)
msg := &types.MsgWithdraw{
Withdrawer: ownerAddr,
ToAddress: withdrawAddr,
Denoms: []string{"stake"},
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
// Update context time
// After first period 500stake should be unlock
ctx = ctx.WithHeaderInfo(header.Info{
Time: currentTime.Add(time.Minute),
})
// Check if 500 stake is sendable now
t.Run("ok - execute send message", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(500))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
balance := app.BankKeeper.GetBalance(ctx, randAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(500)))
})
// Update context time
// After second period 1000stake should be unlock
ctx = ctx.WithHeaderInfo(header.Info{
Time: currentTime.Add(time.Minute * 2),
})
t.Run("oke - execute withdraw message", func(t *testing.T) {
ownerAddr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
withdrawAddr, err := app.AuthKeeper.AddressCodec().BytesToString(withdrawAcc)
require.NoError(t, err)
msg := &types.MsgWithdraw{
Withdrawer: ownerAddr,
ToAddress: withdrawAddr,
Denoms: []string{"stake"},
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
// withdrawable amount should be
// 1000stake - 500stake( above sent amt ) = 500stake
balance := app.BankKeeper.GetBalance(ctx, withdrawAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(500)))
})
t.Run("ok - execute delegate message", func(t *testing.T) {
msg := &types.MsgDelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
del, err := app.StakingKeeper.Delegations.Get(
ctx, collections.Join(sdk.AccAddress(accountAddr), sdk.ValAddress(valbz)),
)
require.NoError(t, err)
require.NotNil(t, del)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.NewInt(100)))
})
t.Run("ok - execute withdraw reward message", func(t *testing.T) {
msg := &types.MsgWithdrawReward{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
})
t.Run("ok - execute undelegate message", func(t *testing.T) {
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
msg := &types.MsgUndelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
ubd, err := app.StakingKeeper.GetUnbondingDelegation(
ctx, sdk.AccAddress(accountAddr), sdk.ValAddress(valbz),
)
require.NoError(t, err)
require.Equal(t, len(ubd.Entries), 1)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.ZeroInt()))
})
// Update context time
// After third period 1500stake should be unlock
ctx = ctx.WithHeaderInfo(header.Info{
Time: currentTime.Add(time.Minute * 3),
})
t.Run("ok - execute delegate message", func(t *testing.T) {
msg := &types.MsgDelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
del, err := app.StakingKeeper.Delegations.Get(
ctx, collections.Join(sdk.AccAddress(accountAddr), sdk.ValAddress(valbz)),
)
require.NoError(t, err)
require.NotNil(t, del)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delFree := lockupAccountInfoResponse.DelegatedFree
require.True(t, delFree.AmountOf("stake").Equal(math.NewInt(100)))
})
}
@@ -0,0 +1,132 @@
package lockup
import (
"testing"
"time"
"github.com/stretchr/testify/require"
"cosmossdk.io/collections"
"cosmossdk.io/core/header"
"cosmossdk.io/math"
lockupaccount "cosmossdk.io/x/accounts/defaults/lockup"
types "cosmossdk.io/x/accounts/defaults/lockup/v1"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
func (s *IntegrationTestSuite) TestPermanentLockingAccount() {
t := s.T()
app := setupApp(t)
currentTime := time.Now()
ctx := sdk.NewContext(app.CommitMultiStore(), false, app.Logger()).WithHeaderInfo(header.Info{
Time: currentTime,
})
ownerAddrStr, err := app.AuthKeeper.AddressCodec().BytesToString(accOwner)
require.NoError(t, err)
s.fundAccount(app, ctx, accOwner, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000000))})
randAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
_, accountAddr, err := app.AccountsKeeper.Init(ctx, lockupaccount.PERMANENT_LOCKING_ACCOUNT, accOwner, &types.MsgInitLockupAccount{
Owner: ownerAddrStr,
}, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000))})
require.NoError(t, err)
addr, err := app.AuthKeeper.AddressCodec().BytesToString(randAcc)
require.NoError(t, err)
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
t.Run("error - execute message, wrong sender", func(t *testing.T) {
msg := &types.MsgSend{
Sender: addr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("error - execute send message, insufficient fund", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NotNil(t, err)
})
t.Run("ok - execute delegate message", func(t *testing.T) {
msg := &types.MsgDelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
del, err := app.StakingKeeper.Delegations.Get(
ctx, collections.Join(sdk.AccAddress(accountAddr), sdk.ValAddress(valbz)),
)
require.NoError(t, err)
require.NotNil(t, del)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.NewInt(100)))
})
t.Run("ok - execute withdraw reward message", func(t *testing.T) {
msg := &types.MsgWithdrawReward{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
})
t.Run("ok - execute undelegate message", func(t *testing.T) {
vals, err := app.StakingKeeper.GetAllValidators(ctx)
require.NoError(t, err)
val := vals[0]
msg := &types.MsgUndelegate{
Sender: ownerAddrStr,
ValidatorAddress: val.OperatorAddress,
Amount: sdk.NewCoin("stake", math.NewInt(100)),
}
err = s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
valbz, err := app.StakingKeeper.ValidatorAddressCodec().StringToBytes(val.OperatorAddress)
require.NoError(t, err)
ubd, err := app.StakingKeeper.GetUnbondingDelegation(
ctx, sdk.AccAddress(accountAddr), sdk.ValAddress(valbz),
)
require.NoError(t, err)
require.Equal(t, len(ubd.Entries), 1)
// check if tracking is updated accordingly
lockupAccountInfoResponse := s.queryLockupAccInfo(ctx, app, accountAddr)
delLocking := lockupAccountInfoResponse.DelegatedLocking
require.True(t, delLocking.AmountOf("stake").Equal(math.ZeroInt()))
})
s.fundAccount(app, ctx, accountAddr, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000))})
t.Run("ok - execute send message", func(t *testing.T) {
msg := &types.MsgSend{
Sender: ownerAddrStr,
ToAddress: addr,
Amount: sdk.Coins{sdk.NewCoin("stake", math.NewInt(100))},
}
err := s.executeTx(ctx, msg, app, accountAddr, accOwner)
require.NoError(t, err)
balance := app.BankKeeper.GetBalance(ctx, randAcc, "stake")
require.True(t, balance.Amount.Equal(math.NewInt(100)))
})
}
@@ -0,0 +1,72 @@
package lockup
import (
"testing"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
"cosmossdk.io/core/transaction"
"cosmossdk.io/simapp"
types "cosmossdk.io/x/accounts/defaults/lockup/v1"
"cosmossdk.io/x/bank/testutil"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
var (
ownerAddr = secp256k1.GenPrivKey().PubKey().Address()
accOwner = sdk.AccAddress(ownerAddr)
)
type IntegrationTestSuite struct {
suite.Suite
app *simapp.SimApp
}
func NewIntegrationTestSuite() *IntegrationTestSuite {
return &IntegrationTestSuite{}
}
func (s *IntegrationTestSuite) SetupSuite() {
s.T().Log("setting up integration test suite")
s.app = setupApp(s.T())
}
func (s *IntegrationTestSuite) TearDownSuite() {
s.T().Log("tearing down integration test suite")
}
func setupApp(t *testing.T) *simapp.SimApp {
t.Helper()
app := simapp.Setup(t, false)
return app
}
func (s *IntegrationTestSuite) executeTx(ctx sdk.Context, msg sdk.Msg, app *simapp.SimApp, accAddr, sender []byte) error {
_, err := app.AccountsKeeper.Execute(ctx, accAddr, sender, msg, nil)
return err
}
func (s *IntegrationTestSuite) queryAcc(ctx sdk.Context, req sdk.Msg, app *simapp.SimApp, accAddr []byte) (transaction.Msg, error) {
resp, err := app.AccountsKeeper.Query(ctx, accAddr, req)
return resp, err
}
func (s *IntegrationTestSuite) fundAccount(app *simapp.SimApp, ctx sdk.Context, addr sdk.AccAddress, amt sdk.Coins) {
require.NoError(s.T(), testutil.FundAccount(ctx, app.BankKeeper, addr, amt))
}
func (s *IntegrationTestSuite) queryLockupAccInfo(ctx sdk.Context, app *simapp.SimApp, accAddr []byte) *types.QueryLockupAccountInfoResponse {
req := &types.QueryLockupAccountInfoRequest{}
resp, err := s.queryAcc(ctx, req, app, accAddr)
require.NoError(s.T(), err)
require.NotNil(s.T(), resp)
lockupAccountInfoResponse, ok := resp.(*types.QueryLockupAccountInfoResponse)
require.True(s.T(), ok)
return lockupAccountInfoResponse
}
@@ -0,0 +1,265 @@
package multisig
import (
"testing"
"time"
"github.com/stretchr/testify/suite"
bankv1beta1 "cosmossdk.io/api/cosmos/bank/v1beta1"
basev1beta1 "cosmossdk.io/api/cosmos/base/v1beta1"
"cosmossdk.io/core/header"
"cosmossdk.io/math"
v1 "cosmossdk.io/x/accounts/defaults/multisig/v1"
accountsv1 "cosmossdk.io/x/accounts/v1"
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
func TestIntegrationTestSuite(t *testing.T) {
suite.Run(t, NewIntegrationTestSuite())
}
// TestSimpleSendProposal creates a multisig account with 1 member, sends a tx, votes and executes it.
func (s *IntegrationTestSuite) TestSimpleSendProposal() {
ctx := sdk.NewContext(s.app.CommitMultiStore(), false, s.app.Logger()).WithHeaderInfo(header.Info{
Time: time.Now(),
})
randAcc := sdk.AccAddress(secp256k1.GenPrivKey().PubKey().Address())
addr, err := s.app.AuthKeeper.AddressCodec().BytesToString(randAcc)
s.NoError(err)
initialMembers := map[string]uint64{
s.membersAddr[0]: 100,
}
accountAddr, accAddrStr := s.initAccount(ctx, s.members[0], initialMembers)
balance := s.app.BankKeeper.GetBalance(ctx, randAcc, "stake")
s.Equal(math.NewInt(0), balance.Amount)
// do a simple bank send
msg := &bankv1beta1.MsgSend{
FromAddress: accAddrStr,
ToAddress: addr,
Amount: []*basev1beta1.Coin{
{
Denom: "stake",
Amount: "100",
},
},
}
anyMsg, err := codectypes.NewAnyWithValue(msg)
s.NoError(err)
s.createProposal(ctx, accountAddr, s.members[0], anyMsg)
// now we vote for it
voteReq := &v1.MsgVote{
ProposalId: 0,
Vote: v1.VoteOption_VOTE_OPTION_YES,
}
err = s.executeTx(ctx, voteReq, accountAddr, s.members[0])
s.NoError(err)
// now we execute it
err = s.executeProposal(ctx, accountAddr, s.members[0], 0)
s.NoError(err)
foundPropResult := false
for _, v := range ctx.EventManager().Events() {
if v.Type == "proposal_tally" {
foundPropResult = true
status, found := v.GetAttribute("status")
s.True(found)
s.Equal(v1.ProposalStatus_PROPOSAL_STATUS_PASSED.String(), status.Value)
yesVotes, found := v.GetAttribute("yes_votes")
s.True(found)
s.Equal("100", yesVotes.Value)
noVotes, found := v.GetAttribute("no_votes")
s.True(found)
s.Equal("0", noVotes.Value)
propID, found := v.GetAttribute("proposal_id")
s.True(found)
s.Equal("0", propID.Value)
execErr, found := v.GetAttribute("exec_err")
s.True(found)
s.Equal("<nil>", execErr.Value)
rejectErr, found := v.GetAttribute("reject_err")
s.True(found)
s.Equal("<nil>", rejectErr.Value)
}
}
s.True(foundPropResult)
balance = s.app.BankKeeper.GetBalance(ctx, randAcc, "stake")
s.Equal(int64(100), balance.Amount.Int64())
// try to execute again, should fail
err = s.executeProposal(ctx, accountAddr, s.members[0], 0)
s.Error(err)
}
// TestConfigUpdate creates a multisig with 1 member, adds 2 more members and
// changes the config to require 2/3 majority (also through a proposal).
func (s *IntegrationTestSuite) TestConfigUpdate() {
ctx := sdk.NewContext(s.app.CommitMultiStore(), false, s.app.Logger()).WithHeaderInfo(header.Info{
Time: time.Now(),
})
initialMembers := map[string]uint64{
s.membersAddr[0]: 100,
}
accountAddr, accAddrStr := s.initAccount(ctx, s.members[0], initialMembers)
// Add 2 members and pass the proposal
// create proposal
updateMsg := &v1.MsgUpdateConfig{
UpdateMembers: []*v1.Member{
{
Address: s.membersAddr[1],
Weight: 100,
},
{
Address: s.membersAddr[2],
Weight: 100,
},
},
Config: &v1.Config{
Threshold: 200, // 3 members with 100 power each, 2/3 majority
Quorum: 200,
VotingPeriod: 120,
Revote: false,
EarlyExecution: false,
},
}
msgExec := &accountsv1.MsgExecute{
Sender: accAddrStr,
Target: accAddrStr,
Message: codectypes.UnsafePackAny(updateMsg),
Funds: []sdk.Coin{},
}
s.createProposal(ctx, accountAddr, s.members[0], codectypes.UnsafePackAny(msgExec))
// vote
voteReq := &v1.MsgVote{
ProposalId: 0,
Vote: v1.VoteOption_VOTE_OPTION_YES,
}
err := s.executeTx(ctx, voteReq, accountAddr, s.members[0])
s.NoError(err)
err = s.executeProposal(ctx, accountAddr, s.members[0], 0)
s.NoError(err)
// get members
res, err := s.queryAcc(ctx, &v1.QueryConfig{}, accountAddr)
s.NoError(err)
resp := res.(*v1.QueryConfigResponse)
s.Len(resp.Members, 3)
s.Equal(int64(200), resp.Config.Threshold)
// Try to remove a member, but it doesn't reach passing threshold
// create proposal
msgExec = &accountsv1.MsgExecute{
Sender: accAddrStr,
Target: accAddrStr,
Message: codectypes.UnsafePackAny(&v1.MsgUpdateConfig{
UpdateMembers: []*v1.Member{
{
Address: s.membersAddr[1],
Weight: 0,
},
},
Config: &v1.Config{
Threshold: 200, // 3 members with 100 power each, 2/3 majority
Quorum: 200,
VotingPeriod: 120,
Revote: false,
EarlyExecution: false,
},
}),
Funds: []sdk.Coin{},
}
s.createProposal(ctx, accountAddr, s.members[0], codectypes.UnsafePackAny(msgExec))
// vote
voteReq = &v1.MsgVote{
ProposalId: 1,
Vote: v1.VoteOption_VOTE_OPTION_NO,
}
err = s.executeTx(ctx, voteReq, accountAddr, s.members[0])
s.NoError(err)
// need to wait until voting period is over because we disabled early execution on the last
// config update
err = s.executeProposal(ctx, accountAddr, s.members[0], 0)
s.ErrorContains(err, "voting period has not ended yet, and early execution is not enabled")
// vote with member 1
voteReq = &v1.MsgVote{
ProposalId: 1,
Vote: v1.VoteOption_VOTE_OPTION_NO,
}
err = s.executeTx(ctx, voteReq, accountAddr, s.members[1])
s.NoError(err)
// need to wait until voting period is over because we disabled early execution on the last
// config update
err = s.executeProposal(ctx, accountAddr, s.members[0], 1)
s.ErrorContains(err, "voting period has not ended yet, and early execution is not enabled")
headerInfo := ctx.HeaderInfo()
headerInfo.Time = headerInfo.Time.Add(time.Second * 121)
ctx = ctx.WithHeaderInfo(headerInfo)
// now it should work, but the proposal will fail
err = s.executeProposal(ctx, accountAddr, s.members[0], 1)
s.NoError(err)
foundPropResult := false
for _, v := range ctx.EventManager().Events() {
if v.Type == "proposal_tally" {
propID, found := v.GetAttribute("proposal_id")
s.True(found)
if propID.Value == "1" {
foundPropResult = true
status, found := v.GetAttribute("status")
s.True(found)
s.Equal(v1.ProposalStatus_PROPOSAL_STATUS_REJECTED.String(), status.Value)
// exec_err is nil because the proposal didn't execute
execErr, found := v.GetAttribute("exec_err")
s.True(found)
s.Equal("<nil>", execErr.Value)
rejectErr, found := v.GetAttribute("reject_err")
s.True(found)
s.Equal("threshold not reached", rejectErr.Value)
}
}
}
s.True(foundPropResult)
// get members
res, err = s.queryAcc(ctx, &v1.QueryConfig{}, accountAddr)
s.NoError(err)
resp = res.(*v1.QueryConfigResponse)
s.Len(resp.Members, 3)
s.Equal(int64(200), resp.Config.Threshold)
}
@@ -0,0 +1,115 @@
package multisig
import (
"context"
"testing"
"github.com/stretchr/testify/require"
"github.com/stretchr/testify/suite"
"cosmossdk.io/core/transaction"
"cosmossdk.io/math"
"cosmossdk.io/simapp"
v1 "cosmossdk.io/x/accounts/defaults/multisig/v1"
"cosmossdk.io/x/bank/testutil"
codectypes "github.com/cosmos/cosmos-sdk/codec/types"
"github.com/cosmos/cosmos-sdk/crypto/keys/secp256k1"
sdk "github.com/cosmos/cosmos-sdk/types"
)
type IntegrationTestSuite struct {
suite.Suite
app *simapp.SimApp
members []sdk.AccAddress
membersAddr []string
}
func NewIntegrationTestSuite() *IntegrationTestSuite {
return &IntegrationTestSuite{}
}
func (s *IntegrationTestSuite) SetupSuite() {
s.app = setupApp(s.T())
s.members = []sdk.AccAddress{}
for i := 0; i < 10; i++ {
addr := secp256k1.GenPrivKey().PubKey().Address()
addrStr, err := s.app.AuthKeeper.AddressCodec().BytesToString(addr)
require.NoError(s.T(), err)
s.membersAddr = append(s.membersAddr, addrStr)
s.members = append(s.members, sdk.AccAddress(addr))
}
}
func (s *IntegrationTestSuite) TearDownSuite() {}
func setupApp(t *testing.T) *simapp.SimApp {
t.Helper()
app := simapp.Setup(t, false)
return app
}
func (s *IntegrationTestSuite) executeTx(ctx context.Context, msg sdk.Msg, accAddr, sender []byte) error {
_, err := s.app.AccountsKeeper.Execute(ctx, accAddr, sender, msg, nil)
return err
}
func (s *IntegrationTestSuite) queryAcc(ctx context.Context, req sdk.Msg, accAddr []byte) (transaction.Msg, error) {
resp, err := s.app.AccountsKeeper.Query(ctx, accAddr, req)
return resp, err
}
func (s *IntegrationTestSuite) fundAccount(ctx context.Context, addr sdk.AccAddress, amt sdk.Coins) {
require.NoError(s.T(), testutil.FundAccount(ctx, s.app.BankKeeper, addr, amt))
}
// initAccount initializes a multisig account with the given members and powers
// and returns the account address
func (s *IntegrationTestSuite) initAccount(ctx context.Context, sender []byte, membersPowers map[string]uint64) ([]byte, string) {
s.fundAccount(ctx, sender, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000000))})
members := []*v1.Member{}
for addrStr, power := range membersPowers {
members = append(members, &v1.Member{Address: addrStr, Weight: power})
}
_, accountAddr, err := s.app.AccountsKeeper.Init(ctx, "multisig", sender,
&v1.MsgInit{
Members: members,
Config: &v1.Config{
Threshold: 100,
Quorum: 100,
VotingPeriod: 120,
Revote: false,
EarlyExecution: true,
},
}, sdk.Coins{sdk.NewCoin("stake", math.NewInt(1000))})
s.NoError(err)
accountAddrStr, err := s.app.AuthKeeper.AddressCodec().BytesToString(accountAddr)
s.NoError(err)
return accountAddr, accountAddrStr
}
// createProposal
func (s *IntegrationTestSuite) createProposal(ctx context.Context, accAddr, sender []byte, msgs ...*codectypes.Any) {
propReq := &v1.MsgCreateProposal{
Proposal: &v1.Proposal{
Title: "test",
Summary: "test",
Messages: msgs,
},
}
err := s.executeTx(ctx, propReq, accAddr, sender)
s.NoError(err)
}
func (s *IntegrationTestSuite) executeProposal(ctx context.Context, accAddr, sender []byte, proposalID uint64) error {
execReq := &v1.MsgExecuteProposal{
ProposalId: proposalID,
}
return s.executeTx(ctx, execReq, accAddr, sender)
}
+67
View File
@@ -0,0 +1,67 @@
//go:build app_v1
package accounts
import (
"testing"
"cosmossdk.io/core/header"
storetypes "cosmossdk.io/store/types"
counterv1 "cosmossdk.io/x/accounts/testing/counter/v1"
"cosmossdk.io/x/bank/testutil"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/stretchr/testify/require"
)
// TestDependencies aims to test wiring between different account components,
// inherited from the runtime, specifically:
// - address codec
// - binary codec
// - header service
// - gas service
// - funds
func TestDependencies(t *testing.T) {
app := setupApp(t)
ak := app.AccountsKeeper
ctx := sdk.NewContext(app.CommitMultiStore(), false, app.Logger()).WithHeaderInfo(header.Info{ChainID: "chain-id"})
ctx = ctx.WithGasMeter(storetypes.NewGasMeter(500_000))
_, counterAddr, err := ak.Init(ctx, "counter", accCreator, &counterv1.MsgInit{
InitialValue: 0,
}, nil)
require.NoError(t, err)
// test dependencies
creatorInitFunds := sdk.NewCoins(sdk.NewInt64Coin("stake", 100_000))
err = testutil.FundAccount(ctx, app.BankKeeper, accCreator, creatorInitFunds)
require.NoError(t, err)
sentFunds := sdk.NewCoins(sdk.NewInt64Coin("stake", 50_000))
r, err := ak.Execute(
ctx,
counterAddr,
accCreator,
&counterv1.MsgTestDependencies{},
sentFunds,
)
require.NoError(t, err)
res := r.(*counterv1.MsgTestDependenciesResponse)
// test gas
require.NotZero(t, res.BeforeGas)
require.NotZero(t, res.AfterGas)
require.Equal(t, int(uint64(10)), int(res.AfterGas-res.BeforeGas))
// test header service
require.Equal(t, ctx.HeaderInfo().ChainID, res.ChainId)
// test address codec
wantAddr, err := app.AuthKeeper.AddressCodec().BytesToString(counterAddr)
require.NoError(t, err)
require.Equal(t, wantAddr, res.Address)
// test funds
creatorFunds := app.BankKeeper.GetAllBalances(ctx, accCreator)
require.Equal(t, creatorInitFunds.Sub(sentFunds...), creatorFunds)
accFunds := app.BankKeeper.GetAllBalances(ctx, counterAddr)
require.Equal(t, sentFunds, accFunds)
}
@@ -0,0 +1,46 @@
package keeper_test
import (
"testing"
"github.com/stretchr/testify/require"
authTest "github.com/cosmos/cosmos-sdk/tests/integration/auth/keeper"
"github.com/cosmos/cosmos-sdk/testutil/network"
"github.com/cosmos/cosmos-sdk/x/auth/types"
)
func TestAccountRetriever(t *testing.T) {
cfg, err := network.DefaultConfigWithAppConfig(authTest.AppConfig)
require.NoError(t, err)
cfg.NumValidators = 1
network, err := network.New(t, t.TempDir(), cfg)
require.NoError(t, err)
defer network.Cleanup()
_, err = network.WaitForHeight(3)
require.NoError(t, err)
val := network.GetValidators()[0]
clientCtx := val.GetClientCtx()
ar := types.AccountRetriever{}
clientCtx = clientCtx.WithHeight(2)
acc, err := ar.GetAccount(clientCtx, val.GetAddress())
require.NoError(t, err)
require.NotNil(t, acc)
acc, height, err := ar.GetAccountWithHeight(clientCtx, val.GetAddress())
require.NoError(t, err)
require.NotNil(t, acc)
require.Equal(t, height, int64(2))
require.NoError(t, ar.EnsureExists(clientCtx, val.GetAddress()))
accNum, accSeq, err := ar.GetAccountNumberSequence(clientCtx, val.GetAddress())
require.NoError(t, err)
require.Equal(t, accNum, uint64(0))
require.Equal(t, accSeq, uint64(1))
}
@@ -0,0 +1,26 @@
package keeper
import (
_ "cosmossdk.io/x/accounts" // import as blank for app wiring
_ "cosmossdk.io/x/bank" // import as blank for app wiring
_ "cosmossdk.io/x/consensus" // import as blank for app wiring
_ "cosmossdk.io/x/staking" // import as blank for app wiring
"github.com/cosmos/cosmos-sdk/testutil/configurator"
_ "github.com/cosmos/cosmos-sdk/x/auth" // import as blank for app wiring
_ "github.com/cosmos/cosmos-sdk/x/auth/tx/config" // import as blank for app wiring``
_ "github.com/cosmos/cosmos-sdk/x/auth/vesting" // import as blank for app wiring
_ "github.com/cosmos/cosmos-sdk/x/genutil" // import as blank for app wiring
)
var AppConfig = configurator.NewAppConfig(
configurator.AccountsModule(),
configurator.AuthModule(),
configurator.BankModule(),
configurator.VestingModule(),
configurator.StakingModule(),
configurator.TxModule(),
configurator.ValidateModule(),
configurator.ConsensusModule(),
configurator.GenutilModule(),
)
@@ -0,0 +1,67 @@
package keeper_test
import (
"testing"
"github.com/stretchr/testify/require"
"cosmossdk.io/depinject"
"cosmossdk.io/log"
authTest "github.com/cosmos/cosmos-sdk/tests/integration/auth/keeper"
simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/x/auth/keeper"
)
func BenchmarkAccountMapperGetAccountFound(b *testing.B) {
b.ReportAllocs()
var accountKeeper keeper.AccountKeeper
app, err := simtestutil.Setup(
depinject.Configs(
depinject.Supply(log.NewNopLogger()),
authTest.AppConfig,
),
&accountKeeper,
)
require.NoError(b, err)
ctx := app.BaseApp.NewContext(false)
// assumes b.N < 2**24
for i := 0; i < b.N; i++ {
arr := []byte{byte((i & 0xFF0000) >> 16), byte((i & 0xFF00) >> 8), byte(i & 0xFF)}
addr := sdk.AccAddress(arr)
acc := accountKeeper.NewAccountWithAddress(ctx, addr)
accountKeeper.SetAccount(ctx, acc)
}
b.ResetTimer()
for i := 0; i < b.N; i++ {
arr := []byte{byte((i & 0xFF0000) >> 16), byte((i & 0xFF00) >> 8), byte(i & 0xFF)}
accountKeeper.GetAccount(ctx, sdk.AccAddress(arr))
}
}
func BenchmarkAccountMapperSetAccount(b *testing.B) {
b.ReportAllocs()
var accountKeeper keeper.AccountKeeper
app, err := simtestutil.Setup(
depinject.Configs(
depinject.Supply(log.NewNopLogger()),
authTest.AppConfig,
), &accountKeeper)
require.NoError(b, err)
ctx := app.BaseApp.NewContext(false)
b.ResetTimer()
// assumes b.N < 2**24
for i := 0; i < b.N; i++ {
arr := []byte{byte((i & 0xFF0000) >> 16), byte((i & 0xFF00) >> 8), byte(i & 0xFF)}
addr := sdk.AccAddress(arr)
acc := accountKeeper.NewAccountWithAddress(ctx, addr)
accountKeeper.SetAccount(ctx, acc)
}
}
@@ -0,0 +1,30 @@
package keeper_test
import (
"testing"
"github.com/stretchr/testify/require"
"cosmossdk.io/depinject"
"cosmossdk.io/log"
authTest "github.com/cosmos/cosmos-sdk/tests/integration/auth/keeper"
simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
"github.com/cosmos/cosmos-sdk/x/auth/keeper"
"github.com/cosmos/cosmos-sdk/x/auth/types"
)
func TestItCreatesModuleAccountOnInitBlock(t *testing.T) {
var accountKeeper keeper.AccountKeeper
app, err := simtestutil.SetupAtGenesis(
depinject.Configs(
authTest.AppConfig,
depinject.Supply(log.NewNopLogger()),
),
&accountKeeper)
require.NoError(t, err)
ctx := app.BaseApp.NewContext(false)
acc := accountKeeper.GetAccount(ctx, types.NewModuleAddress(types.FeeCollectorName))
require.NotNil(t, acc)
}
@@ -0,0 +1,191 @@
package benchmark_test
import (
"context"
"testing"
"gotest.tools/v3/assert"
sdkmath "cosmossdk.io/math"
"cosmossdk.io/simapp"
banktypes "cosmossdk.io/x/bank/types"
"github.com/cosmos/cosmos-sdk/client"
clienttx "github.com/cosmos/cosmos-sdk/client/tx"
"github.com/cosmos/cosmos-sdk/testutil/cli"
"github.com/cosmos/cosmos-sdk/testutil/network"
"github.com/cosmos/cosmos-sdk/testutil/testdata"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/types/tx"
"github.com/cosmos/cosmos-sdk/types/tx/signing"
authclient "github.com/cosmos/cosmos-sdk/x/auth/client"
)
type TxBenchmarkSuite struct {
cfg network.Config
network network.NetworkI
txHeight int64
queryClient tx.ServiceClient
}
// BenchmarkTx is lifted from TestSuite from this package, with irrelevant state checks removed.
//
// Benchmark results:
//
// On a M1 macOS, on commit ab77fe20d3c00ef4cb73dfd0c803c4593a3c8233:
//
// BenchmarkTx-8 4108 268935 ns/op
//
// By comparison, the benchmark modified for v0.47.0:
//
// BenchmarkTx-8 3772 301750 ns/op
func BenchmarkTx(b *testing.B) {
s := NewTxBenchmarkSuite(b)
b.Cleanup(s.Close)
val := s.network.GetValidators()[0]
txBuilder := mkTxBuilder(b, s)
// Convert the txBuilder to a tx.Tx.
protoTx, err := txBuilder.GetTx().(interface{ AsTx() (*tx.Tx, error) }).AsTx()
assert.NilError(b, err)
// Encode the txBuilder to txBytes.
txBytes, err := val.GetClientCtx().TxConfig.TxEncoder()(txBuilder.GetTx())
assert.NilError(b, err)
testCases := []struct {
name string
req *tx.SimulateRequest
}{
{"valid request with proto tx (deprecated)", &tx.SimulateRequest{Tx: protoTx}},
{"valid request with tx_bytes", &tx.SimulateRequest{TxBytes: txBytes}},
}
b.ResetTimer()
b.ReportAllocs()
for i := 0; i < b.N; i++ {
for _, tc := range testCases {
// Broadcast the tx via gRPC via the validator's clientCtx (which goes
// through Tendermint).
res, err := s.queryClient.Simulate(context.Background(), tc.req)
assert.NilError(b, err)
// Check the result and gas used are correct.
//
// The 10 events are:
// - Sending Fee to the pool (3 events): coin_spent, coin_received, transfer
// - tx.* events (3 events): tx.fee, tx.acc_seq, tx.signature
// - Sending Amount to recipient (3 events): coin_spent, coin_received, transfer and message.sender=<val1>
// - Msg events (1 event): message.module=bank, message.action=/cosmos.bank.v1beta1.MsgSend and message.sender=<val1> (all in one event)
assert.Equal(b, 10, len(res.GetResult().GetEvents()))
assert.Assert(b, res.GetGasInfo().GetGasUsed() > 0) // Gas used sometimes change, just check it's not empty.
}
}
}
func NewTxBenchmarkSuite(tb testing.TB) *TxBenchmarkSuite {
tb.Helper()
s := new(TxBenchmarkSuite)
cfg := network.DefaultConfig(simapp.NewTestNetworkFixture)
cfg.NumValidators = 1
s.cfg = cfg
var err error
s.network, err = network.New(tb, tb.TempDir(), s.cfg)
assert.NilError(tb, err)
val := s.network.GetValidators()[0]
assert.NilError(tb, s.network.WaitForNextBlock())
s.queryClient = tx.NewServiceClient(val.GetClientCtx())
msgSend := &banktypes.MsgSend{
FromAddress: val.GetAddress().String(),
ToAddress: val.GetAddress().String(),
Amount: sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdkmath.NewInt(10))),
}
// Create a new MsgSend tx from val to itself.
out, err := cli.SubmitTestTx(
val.GetClientCtx(),
msgSend,
val.GetAddress(),
cli.TestTxConfig{
Memo: "foobar",
},
)
assert.NilError(tb, err)
var txRes sdk.TxResponse
assert.NilError(tb, val.GetClientCtx().Codec.UnmarshalJSON(out.Bytes(), &txRes))
assert.Equal(tb, uint32(0), txRes.Code, txRes)
msgSend1 := &banktypes.MsgSend{
FromAddress: val.GetAddress().String(),
ToAddress: val.GetAddress().String(),
Amount: sdk.NewCoins(sdk.NewCoin(s.cfg.BondDenom, sdkmath.NewInt(1))),
}
out, err = cli.SubmitTestTx(
val.GetClientCtx(),
msgSend1,
val.GetAddress(),
cli.TestTxConfig{
Offline: true,
AccNum: 0,
Seq: 2,
Memo: "foobar",
},
)
assert.NilError(tb, err)
var tr sdk.TxResponse
assert.NilError(tb, val.GetClientCtx().Codec.UnmarshalJSON(out.Bytes(), &tr))
assert.Equal(tb, uint32(0), tr.Code)
resp, err := cli.GetTxResponse(s.network, val.GetClientCtx(), tr.TxHash)
assert.NilError(tb, err)
s.txHeight = resp.Height
return s
}
func (s *TxBenchmarkSuite) Close() {
s.network.Cleanup()
}
func mkTxBuilder(tb testing.TB, s *TxBenchmarkSuite) client.TxBuilder {
tb.Helper()
val := s.network.GetValidators()[0]
assert.NilError(tb, s.network.WaitForNextBlock())
// prepare txBuilder with msg
txBuilder := val.GetClientCtx().TxConfig.NewTxBuilder()
feeAmount := sdk.Coins{sdk.NewInt64Coin(s.cfg.BondDenom, 10)}
gasLimit := testdata.NewTestGasLimit()
assert.NilError(tb,
txBuilder.SetMsgs(&banktypes.MsgSend{
FromAddress: val.GetAddress().String(),
ToAddress: val.GetAddress().String(),
Amount: sdk.Coins{sdk.NewInt64Coin(s.cfg.BondDenom, 10)},
}),
)
txBuilder.SetFeeAmount(feeAmount)
txBuilder.SetGasLimit(gasLimit)
txBuilder.SetMemo("foobar")
// setup txFactory
txFactory := clienttx.Factory{}.
WithChainID(val.GetClientCtx().ChainID).
WithKeybase(val.GetClientCtx().Keyring).
WithTxConfig(val.GetClientCtx().TxConfig).
WithSignMode(signing.SignMode_SIGN_MODE_DIRECT)
// Sign Tx.
err := authclient.SignTx(txFactory, val.GetClientCtx(), val.GetMoniker(), txBuilder, false, true)
assert.NilError(tb, err)
return txBuilder
}