51 lines
1.6 KiB
Go
51 lines
1.6 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
|
|
|
|
"github.com/cosmos/cosmos-sdk/simapp"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/cosmos/cosmos-sdk/x/crisis/types"
|
|
)
|
|
|
|
func TestLogger(t *testing.T) {
|
|
app := simapp.Setup(t, false)
|
|
|
|
ctx := app.NewContext(true, tmproto.Header{})
|
|
|
|
require.Equal(t,
|
|
ctx.Logger().With("module", "x/"+types.ModuleName),
|
|
app.CrisisKeeper.Logger(ctx))
|
|
}
|
|
|
|
func TestInvariants(t *testing.T) {
|
|
app := simapp.Setup(t, false)
|
|
app.Commit()
|
|
app.BeginBlock(abci.RequestBeginBlock{Header: tmproto.Header{Height: app.LastBlockHeight() + 1}})
|
|
|
|
require.Equal(t, app.CrisisKeeper.InvCheckPeriod(), uint(5))
|
|
|
|
// SimApp has 11 registered invariants
|
|
orgInvRoutes := app.CrisisKeeper.Routes()
|
|
app.CrisisKeeper.RegisterRoute("testModule", "testRoute", func(sdk.Context) (string, bool) { return "", false })
|
|
require.Equal(t, len(app.CrisisKeeper.Routes()), len(orgInvRoutes)+1)
|
|
}
|
|
|
|
func TestAssertInvariants(t *testing.T) {
|
|
app := simapp.Setup(t, false)
|
|
app.Commit()
|
|
app.BeginBlock(abci.RequestBeginBlock{Header: tmproto.Header{Height: app.LastBlockHeight() + 1}})
|
|
|
|
ctx := app.NewContext(true, tmproto.Header{})
|
|
|
|
app.CrisisKeeper.RegisterRoute("testModule", "testRoute1", func(sdk.Context) (string, bool) { return "", false })
|
|
require.NotPanics(t, func() { app.CrisisKeeper.AssertInvariants(ctx) })
|
|
|
|
app.CrisisKeeper.RegisterRoute("testModule", "testRoute2", func(sdk.Context) (string, bool) { return "", true })
|
|
require.Panics(t, func() { app.CrisisKeeper.AssertInvariants(ctx) })
|
|
}
|