mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-28 17:25:19 +00:00
80 lines
2.0 KiB
Go
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))
|
|
}
|