mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-26 08:15:19 +00:00
d45fa58f5c
* wip: add swap state persistent to genesis * separate pool record constructors; add tests for json and yaml encoding of record structs * beef up validation checks for state records * fix integration with master - renamed method * add test coverage for basic state array validations * extra test around pool record reserve and id ordering to ensure no regressions in the future * add validations to ensure pool records and share records are unique within the collection types * test genesis json and yaml encoding * validate in genesis that the total shares owned for each pool is equal to the total shares of each pool * update alias * nit lint * test genesis init and export * add migration todo Co-authored-by: Nick DeLuca <nickdeluca08@gmail.com>
74 lines
2.4 KiB
Go
74 lines
2.4 KiB
Go
package swap_test
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/kava-labs/kava/x/swap"
|
|
"github.com/kava-labs/kava/x/swap/testutil"
|
|
"github.com/kava-labs/kava/x/swap/types"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
)
|
|
|
|
type genesisTestSuite struct {
|
|
testutil.Suite
|
|
}
|
|
|
|
func (suite *genesisTestSuite) Test_InitGenesis_ValidationPanic() {
|
|
invalidState := types.NewGenesisState(
|
|
types.Params{
|
|
SwapFee: sdk.NewDec(-1),
|
|
},
|
|
types.PoolRecords{},
|
|
types.ShareRecords{},
|
|
)
|
|
|
|
suite.Panics(func() {
|
|
swap.InitGenesis(suite.Ctx, suite.Keeper, invalidState)
|
|
}, "expected init genesis to panic with invalid state")
|
|
}
|
|
|
|
func (suite *genesisTestSuite) Test_InitAndExportGenesis() {
|
|
depositor_1, err := sdk.AccAddressFromBech32("kava1mq9qxlhze029lm0frzw2xr6hem8c3k9ts54w0w")
|
|
suite.Require().NoError(err)
|
|
depositor_2, err := sdk.AccAddressFromBech32("kava1esagqd83rhqdtpy5sxhklaxgn58k2m3s3mnpea")
|
|
suite.Require().NoError(err)
|
|
|
|
// slices are sorted by key as stored in the data store, so init and export can be compared with equal
|
|
state := types.NewGenesisState(
|
|
types.Params{
|
|
AllowedPools: swap.AllowedPools{swap.NewAllowedPool("ukava", "usdx")},
|
|
SwapFee: sdk.MustNewDecFromStr("0.00255"),
|
|
},
|
|
types.PoolRecords{
|
|
swap.NewPoolRecord(sdk.NewCoins(sdk.NewCoin("hard", sdk.NewInt(1e6)), sdk.NewCoin("usdx", sdk.NewInt(2e6))), sdk.NewInt(1e6)),
|
|
swap.NewPoolRecord(sdk.NewCoins(sdk.NewCoin("ukava", sdk.NewInt(1e6)), sdk.NewCoin("usdx", sdk.NewInt(5e6))), sdk.NewInt(3e6)),
|
|
},
|
|
types.ShareRecords{
|
|
types.NewShareRecord(depositor_2, "hard/usdx", sdk.NewInt(1e6)),
|
|
types.NewShareRecord(depositor_1, "ukava/usdx", sdk.NewInt(3e6)),
|
|
},
|
|
)
|
|
|
|
swap.InitGenesis(suite.Ctx, suite.Keeper, state)
|
|
suite.Equal(state.Params, suite.Keeper.GetParams(suite.Ctx))
|
|
|
|
poolRecord1, _ := suite.Keeper.GetPool(suite.Ctx, "hard/usdx")
|
|
suite.Equal(state.PoolRecords[0], poolRecord1)
|
|
poolRecord2, _ := suite.Keeper.GetPool(suite.Ctx, "ukava/usdx")
|
|
suite.Equal(state.PoolRecords[1], poolRecord2)
|
|
|
|
shareRecord1, _ := suite.Keeper.GetDepositorShares(suite.Ctx, depositor_2, "hard/usdx")
|
|
suite.Equal(state.ShareRecords[0], shareRecord1)
|
|
shareRecord2, _ := suite.Keeper.GetDepositorShares(suite.Ctx, depositor_1, "ukava/usdx")
|
|
suite.Equal(state.ShareRecords[1], shareRecord2)
|
|
|
|
exportedState := swap.ExportGenesis(suite.Ctx, suite.Keeper)
|
|
suite.Equal(state, exportedState)
|
|
}
|
|
|
|
func TestGenesisTestSuite(t *testing.T) {
|
|
suite.Run(t, new(genesisTestSuite))
|
|
}
|