0g-chain/x/savings/genesis_test.go
2024-08-02 12:22:00 +08:00

80 lines
2.0 KiB
Go

package savings_test
import (
"testing"
"time"
"github.com/stretchr/testify/suite"
sdkmath "cosmossdk.io/math"
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
tmtime "github.com/cometbft/cometbft/types/time"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/0glabs/0g-chain/app"
"github.com/0glabs/0g-chain/x/savings"
"github.com/0glabs/0g-chain/x/savings/keeper"
"github.com/0glabs/0g-chain/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", sdkmath.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))
}