15b04c2a87
## Description There are still some references to simapp in `x/bank` and `x/slashing`. This patch removes them. Somewhat related to forming tests also: - Flags `LegacySubspace exported.Subspace` as ``optional:"true"`` since x/params is moving to deprecation --- ### Author Checklist *All items are required. Please add a note to the item if the item is not applicable and please add links to any relevant follow up issues.* I have... - [ ] included the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title - [ ] added `!` to the type prefix if API or client breaking change - [ ] targeted the correct branch (see [PR Targeting](https://github.com/cosmos/cosmos-sdk/blob/main/CONTRIBUTING.md#pr-targeting)) - [ ] provided a link to the relevant issue or specification - [ ] followed the guidelines for [building modules](https://github.com/cosmos/cosmos-sdk/blob/main/docs/building-modules) - [ ] included the necessary unit and integration [tests](https://github.com/cosmos/cosmos-sdk/blob/main/CONTRIBUTING.md#testing) - [ ] added a changelog entry to `CHANGELOG.md` - [ ] included comments for [documenting Go code](https://blog.golang.org/godoc) - [ ] updated the relevant documentation or specification - [ ] reviewed "Files changed" and left comments if necessary - [ ] confirmed all CI checks have passed ### Reviewers Checklist *All items are required. Please add a note if the item is not applicable and please add your handle next to the items reviewed if you only reviewed selected items.* I have... - [ ] confirmed the correct [type prefix](https://github.com/commitizen/conventional-commit-types/blob/v3.0.0/index.json) in the PR title - [ ] confirmed `!` in the type prefix if API or client breaking change - [ ] confirmed all author checklist items have been addressed - [ ] reviewed state machine logic - [ ] reviewed API design and naming - [ ] reviewed documentation is accurate - [ ] reviewed tests and test coverage - [ ] manually tested (if applicable)
105 lines
3.9 KiB
Go
105 lines
3.9 KiB
Go
package capability_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/suite"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/baseapp"
|
|
"github.com/cosmos/cosmos-sdk/codec"
|
|
"github.com/cosmos/cosmos-sdk/runtime"
|
|
storetypes "github.com/cosmos/cosmos-sdk/store/types"
|
|
simtestutil "github.com/cosmos/cosmos-sdk/testutil/sims"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
banktypes "github.com/cosmos/cosmos-sdk/x/bank/types"
|
|
"github.com/cosmos/cosmos-sdk/x/capability"
|
|
"github.com/cosmos/cosmos-sdk/x/capability/keeper"
|
|
"github.com/cosmos/cosmos-sdk/x/capability/testutil"
|
|
"github.com/cosmos/cosmos-sdk/x/capability/types"
|
|
)
|
|
|
|
type CapabilityTestSuite struct {
|
|
suite.Suite
|
|
|
|
app *runtime.App
|
|
cdc codec.Codec
|
|
ctx sdk.Context
|
|
keeper *keeper.Keeper
|
|
memKey *storetypes.MemoryStoreKey
|
|
}
|
|
|
|
func (suite *CapabilityTestSuite) SetupTest() {
|
|
suite.memKey = storetypes.NewMemoryStoreKey("testingkey")
|
|
|
|
startupCfg := simtestutil.DefaultStartUpConfig()
|
|
startupCfg.BaseAppOption = func(ba *baseapp.BaseApp) {
|
|
ba.MountStores(suite.memKey)
|
|
}
|
|
|
|
app, err := simtestutil.SetupWithConfiguration(testutil.AppConfig,
|
|
startupCfg,
|
|
&suite.cdc,
|
|
&suite.keeper,
|
|
)
|
|
suite.Require().NoError(err)
|
|
|
|
suite.app = app
|
|
suite.ctx = app.BaseApp.NewContext(false, tmproto.Header{Height: 1})
|
|
}
|
|
|
|
// The following test case mocks a specific bug discovered in https://github.com/cosmos/cosmos-sdk/issues/9800
|
|
// and ensures that the current code successfully fixes the issue.
|
|
func (suite *CapabilityTestSuite) TestInitializeMemStore() {
|
|
sk1 := suite.keeper.ScopeToModule(banktypes.ModuleName)
|
|
|
|
cap1, err := sk1.NewCapability(suite.ctx, "transfer")
|
|
suite.Require().NoError(err)
|
|
suite.Require().NotNil(cap1)
|
|
|
|
// mock statesync by creating new keeper that shares persistent state but loses in-memory map
|
|
newKeeper := keeper.NewKeeper(suite.cdc, suite.app.UnsafeFindStoreKey(types.StoreKey).(*storetypes.KVStoreKey), suite.memKey)
|
|
newSk1 := newKeeper.ScopeToModule(banktypes.ModuleName)
|
|
|
|
// Mock App startup
|
|
ctx := suite.app.BaseApp.NewUncachedContext(false, tmproto.Header{})
|
|
newKeeper.Seal()
|
|
suite.Require().False(newKeeper.IsInitialized(ctx), "memstore initialized flag set before BeginBlock")
|
|
|
|
// Mock app beginblock and ensure that no gas has been consumed and memstore is initialized
|
|
ctx = suite.app.BaseApp.NewContext(false, tmproto.Header{}).WithBlockGasMeter(sdk.NewGasMeter(50))
|
|
prevGas := ctx.BlockGasMeter().GasConsumed()
|
|
restartedModule := capability.NewAppModule(suite.cdc, *newKeeper, true)
|
|
restartedModule.BeginBlock(ctx, abci.RequestBeginBlock{})
|
|
suite.Require().True(newKeeper.IsInitialized(ctx), "memstore initialized flag not set")
|
|
gasUsed := ctx.BlockGasMeter().GasConsumed()
|
|
|
|
suite.Require().Equal(prevGas, gasUsed, "beginblocker consumed gas during execution")
|
|
|
|
// Mock the first transaction getting capability and subsequently failing
|
|
// by using a cached context and discarding all cached writes.
|
|
cacheCtx, _ := ctx.CacheContext()
|
|
_, ok := newSk1.GetCapability(cacheCtx, "transfer")
|
|
suite.Require().True(ok)
|
|
|
|
// Ensure that the second transaction can still receive capability even if first tx fails.
|
|
ctx = suite.app.BaseApp.NewContext(false, tmproto.Header{})
|
|
|
|
cap1, ok = newSk1.GetCapability(ctx, "transfer")
|
|
suite.Require().True(ok)
|
|
|
|
// Ensure the capabilities don't get reinitialized on next BeginBlock
|
|
// by testing to see if capability returns same pointer
|
|
// also check that initialized flag is still set
|
|
restartedModule.BeginBlock(ctx, abci.RequestBeginBlock{})
|
|
recap, ok := newSk1.GetCapability(ctx, "transfer")
|
|
suite.Require().True(ok)
|
|
suite.Require().Equal(cap1, recap, "capabilities got reinitialized after second BeginBlock")
|
|
suite.Require().True(newKeeper.IsInitialized(ctx), "memstore initialized flag not set")
|
|
}
|
|
|
|
func TestCapabilityTestSuite(t *testing.T) {
|
|
suite.Run(t, new(CapabilityTestSuite))
|
|
}
|