135 lines
4.2 KiB
Go
135 lines
4.2 KiB
Go
package simapp
|
|
|
|
import (
|
|
"encoding/json"
|
|
"os"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
"github.com/tendermint/tendermint/libs/log"
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
dbm "github.com/tendermint/tm-db"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/types/module"
|
|
)
|
|
|
|
func TestSimAppExportAndBlockedAddrs(t *testing.T) {
|
|
encCfg := MakeTestEncodingConfig()
|
|
db := dbm.NewMemDB()
|
|
app := NewSimApp(log.NewTMLogger(log.NewSyncWriter(os.Stdout)), db, nil, true, map[int64]bool{}, DefaultNodeHome, 0, encCfg, EmptyAppOptions{})
|
|
|
|
for acc := range maccPerms {
|
|
require.Equal(t, !allowedReceivingModAcc[acc], app.BankKeeper.BlockedAddr(app.AccountKeeper.GetModuleAddress(acc)),
|
|
"ensure that blocked addresses are properly set in bank keeper")
|
|
}
|
|
|
|
genesisState := NewDefaultGenesisState(encCfg.Marshaler)
|
|
stateBytes, err := json.MarshalIndent(genesisState, "", " ")
|
|
require.NoError(t, err)
|
|
|
|
// Initialize the chain
|
|
app.InitChain(
|
|
abci.RequestInitChain{
|
|
Validators: []abci.ValidatorUpdate{},
|
|
AppStateBytes: stateBytes,
|
|
},
|
|
)
|
|
app.Commit()
|
|
|
|
// Making a new app object with the db, so that initchain hasn't been called
|
|
app2 := NewSimApp(log.NewTMLogger(log.NewSyncWriter(os.Stdout)), db, nil, true, map[int64]bool{}, DefaultNodeHome, 0, encCfg, EmptyAppOptions{})
|
|
_, err = app2.ExportAppStateAndValidators(false, []string{})
|
|
require.NoError(t, err, "ExportAppStateAndValidators should not have an error")
|
|
}
|
|
|
|
func TestGetMaccPerms(t *testing.T) {
|
|
dup := GetMaccPerms()
|
|
require.Equal(t, maccPerms, dup, "duplicated module account permissions differed from actual module account permissions")
|
|
}
|
|
|
|
func TestRunMigrations(t *testing.T) {
|
|
db := dbm.NewMemDB()
|
|
encCfg := MakeTestEncodingConfig()
|
|
app := NewSimApp(log.NewTMLogger(log.NewSyncWriter(os.Stdout)), db, nil, true, map[int64]bool{}, DefaultNodeHome, 0, encCfg, EmptyAppOptions{})
|
|
|
|
// Create a new configurator for the purpose of this test.
|
|
app.configurator = module.NewConfigurator(app.MsgServiceRouter(), app.GRPCQueryRouter())
|
|
|
|
testCases := []struct {
|
|
name string
|
|
moduleName string
|
|
forVersion uint64
|
|
expRegErr bool // errors while registering migration
|
|
expRegErrMsg string
|
|
expRunErr bool // errors while running migration
|
|
expRunErrMsg string
|
|
expCalled int
|
|
}{
|
|
{
|
|
"cannot register migration for version 0",
|
|
"bank", 0,
|
|
true, "module migration versions should start at 1: invalid version", false, "", 0,
|
|
},
|
|
{
|
|
"throws error on RunMigrations if no migration registered for bank",
|
|
"", 1,
|
|
false, "", true, "no migrations found for module bank: not found", 0,
|
|
},
|
|
{
|
|
"can register and run migration handler for x/bank",
|
|
"bank", 1,
|
|
false, "", false, "", 1,
|
|
},
|
|
{
|
|
"cannot register migration handler for same module & forVersion",
|
|
"bank", 1,
|
|
true, "another migration for module bank and version 1 already exists: internal logic error", false, "", 0,
|
|
},
|
|
}
|
|
|
|
for _, tc := range testCases {
|
|
t.Run(tc.name, func(t *testing.T) {
|
|
var err error
|
|
|
|
// Since it's very hard to test actual in-place store migrations in
|
|
// tests (due to the difficulty of maintaining multiple versions of a
|
|
// module), we're just testing here that the migration logic is
|
|
// called.
|
|
called := 0
|
|
|
|
if tc.moduleName != "" {
|
|
// Register migration for module from version `forVersion` to `forVersion+1`.
|
|
err = app.configurator.RegisterMigration(tc.moduleName, tc.forVersion, func(sdk.Context) error {
|
|
called++
|
|
|
|
return nil
|
|
})
|
|
|
|
if tc.expRegErr {
|
|
require.EqualError(t, err, tc.expRegErrMsg)
|
|
|
|
return
|
|
}
|
|
}
|
|
require.NoError(t, err)
|
|
|
|
err = app.RunMigrations(
|
|
app.NewContext(true, tmproto.Header{Height: app.LastBlockHeight()}),
|
|
module.MigrationMap{
|
|
"auth": 1, "authz": 1, "bank": 1, "staking": 1, "mint": 1, "distribution": 1,
|
|
"slashing": 1, "gov": 1, "params": 1, "ibc": 1, "upgrade": 1, "vesting": 1,
|
|
"feegrant": 1, "transfer": 1, "evidence": 1, "crisis": 1, "genutil": 1, "capability": 1,
|
|
},
|
|
)
|
|
if tc.expRunErr {
|
|
require.EqualError(t, err, tc.expRunErrMsg)
|
|
} else {
|
|
require.NoError(t, err)
|
|
require.Equal(t, tc.expCalled, called)
|
|
}
|
|
})
|
|
}
|
|
}
|