0g-chain/x/savings/genesis_test.go
Denali Marsh 988836dee0
Savings module invariants (#1199)
* add deposits to genesis state

* import/export genesis with deposits

* add helper keeper method + update tests

* invariants + tests

* register invariants on module

* fix genesis test invariant init

* clean up invariants test

* remove comment from test file

* fix invariants test

* run 'make proto-all'
2022-03-30 13:51:06 +02:00

79 lines
2.0 KiB
Go

package savings_test
import (
"testing"
"time"
"github.com/stretchr/testify/suite"
sdk "github.com/cosmos/cosmos-sdk/types"
tmproto "github.com/tendermint/tendermint/proto/tendermint/types"
tmtime "github.com/tendermint/tendermint/types/time"
"github.com/kava-labs/kava/app"
"github.com/kava-labs/kava/x/savings"
"github.com/kava-labs/kava/x/savings/keeper"
"github.com/kava-labs/kava/x/savings/types"
)
type GenesisTestSuite struct {
suite.Suite
app app.TestApp
genTime time.Time
ctx sdk.Context
keeper keeper.Keeper
addrs []sdk.AccAddress
}
func (suite *GenesisTestSuite) SetupTest() {
tApp := app.NewTestApp()
suite.genTime = tmtime.Canonical(time.Date(2022, 1, 1, 1, 1, 1, 1, time.UTC))
suite.ctx = tApp.NewContext(true, tmproto.Header{Height: 1, Time: suite.genTime})
suite.keeper = tApp.GetSavingsKeeper()
suite.app = tApp
_, addrs := app.GeneratePrivKeyAddressPairs(3)
suite.addrs = addrs
}
func (suite *GenesisTestSuite) TestInitExportGenesis() {
params := types.NewParams(
[]string{"btc", "ukava", "bnb"},
)
depositAmt := sdk.NewCoins(sdk.NewCoin("ukava", sdk.NewInt(1e8)))
deposits := types.Deposits{
types.NewDeposit(
suite.addrs[0],
depositAmt, // 100 ukava
),
}
savingsGenesis := types.NewGenesisState(params, deposits)
authBuilder := app.NewAuthBankGenesisBuilder().
WithSimpleModuleAccount(types.ModuleAccountName, depositAmt)
cdc := suite.app.AppCodec()
suite.NotPanics(
func() {
suite.app.InitializeFromGenesisStatesWithTime(
suite.genTime,
authBuilder.BuildMarshalled(cdc),
app.GenesisState{types.ModuleName: cdc.MustMarshalJSON(&savingsGenesis)},
)
},
)
expectedDeposits := suite.keeper.GetAllDeposits(suite.ctx)
expectedGenesis := savingsGenesis
expectedGenesis.Deposits = expectedDeposits
exportedGenesis := savings.ExportGenesis(suite.ctx, suite.keeper)
suite.Equal(expectedGenesis, exportedGenesis)
}
func TestGenesisTestSuite(t *testing.T) {
suite.Run(t, new(GenesisTestSuite))
}