package keeper_test import ( "fmt" "sort" "strconv" "testing" "github.com/stretchr/testify/suite" sdk "github.com/cosmos/cosmos-sdk/types" authexported "github.com/cosmos/cosmos-sdk/x/auth/exported" supplyexported "github.com/cosmos/cosmos-sdk/x/supply/exported" abci "github.com/tendermint/tendermint/abci/types" tmtime "github.com/tendermint/tendermint/types/time" "github.com/kava-labs/kava/app" aucKeeper "github.com/kava-labs/kava/x/auction/keeper" "github.com/kava-labs/kava/x/harvest/keeper" "github.com/kava-labs/kava/x/harvest/types" ) // Test suite used for all keeper tests type KeeperTestSuite struct { suite.Suite keeper keeper.Keeper auctionKeeper aucKeeper.Keeper app app.TestApp ctx sdk.Context addrs []sdk.AccAddress } // The default state used by each test func (suite *KeeperTestSuite) SetupTest() { config := sdk.GetConfig() app.SetBech32AddressPrefixes(config) tApp := app.NewTestApp() ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: tmtime.Now()}) tApp.InitializeFromGenesisStates() _, addrs := app.GeneratePrivKeyAddressPairs(1) keeper := tApp.GetHarvestKeeper() suite.app = tApp suite.ctx = ctx suite.keeper = keeper suite.addrs = addrs } func (suite *KeeperTestSuite) TestGetSetPreviousBlockTime() { now := tmtime.Now() _, f := suite.keeper.GetPreviousBlockTime(suite.ctx) suite.Require().False(f) suite.NotPanics(func() { suite.keeper.SetPreviousBlockTime(suite.ctx, now) }) pbt, f := suite.keeper.GetPreviousBlockTime(suite.ctx) suite.True(f) suite.Equal(now, pbt) } func (suite *KeeperTestSuite) TestGetSetPreviousDelegatorDistribution() { now := tmtime.Now() _, f := suite.keeper.GetPreviousDelegatorDistribution(suite.ctx, suite.keeper.BondDenom(suite.ctx)) suite.Require().False(f) suite.NotPanics(func() { suite.keeper.SetPreviousDelegationDistribution(suite.ctx, now, suite.keeper.BondDenom(suite.ctx)) }) pdt, f := suite.keeper.GetPreviousDelegatorDistribution(suite.ctx, suite.keeper.BondDenom(suite.ctx)) suite.True(f) suite.Equal(now, pdt) } func (suite *KeeperTestSuite) TestGetSetDeleteDeposit() { dep := types.NewDeposit(sdk.AccAddress("test"), sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(100)))) _, f := suite.keeper.GetDeposit(suite.ctx, sdk.AccAddress("test")) suite.Require().False(f) suite.keeper.SetDeposit(suite.ctx, dep) testDeposit, f := suite.keeper.GetDeposit(suite.ctx, sdk.AccAddress("test")) suite.Require().True(f) suite.Require().Equal(dep, testDeposit) suite.Require().NotPanics(func() { suite.keeper.DeleteDeposit(suite.ctx, dep) }) _, f = suite.keeper.GetDeposit(suite.ctx, sdk.AccAddress("test")) suite.Require().False(f) } func (suite *KeeperTestSuite) TestIterateDeposits() { for i := 0; i < 5; i++ { dep := types.NewDeposit(sdk.AccAddress("test"+fmt.Sprint(i)), sdk.NewCoins(sdk.NewCoin("bnb", sdk.NewInt(100)))) suite.Require().NotPanics(func() { suite.keeper.SetDeposit(suite.ctx, dep) }) } var deposits []types.Deposit suite.keeper.IterateDeposits(suite.ctx, func(d types.Deposit) bool { deposits = append(deposits, d) return false }) suite.Require().Equal(5, len(deposits)) } func (suite *KeeperTestSuite) TestGetSetDeleteClaim() { claim := types.NewClaim(sdk.AccAddress("test"), "bnb", sdk.NewCoin("hard", sdk.NewInt(100)), "lp") _, f := suite.keeper.GetClaim(suite.ctx, sdk.AccAddress("test"), "bnb", "lp") suite.Require().False(f) suite.Require().NotPanics(func() { suite.keeper.SetClaim(suite.ctx, claim) }) testClaim, f := suite.keeper.GetClaim(suite.ctx, sdk.AccAddress("test"), "bnb", "lp") suite.Require().True(f) suite.Require().Equal(claim, testClaim) suite.Require().NotPanics(func() { suite.keeper.DeleteClaim(suite.ctx, claim) }) _, f = suite.keeper.GetClaim(suite.ctx, sdk.AccAddress("test"), "bnb", "lp") suite.Require().False(f) } func (suite *KeeperTestSuite) TestGetSetDeleteInterestRateModel() { denom := "test" model := types.NewInterestRateModel(sdk.MustNewDecFromStr("0.05"), sdk.MustNewDecFromStr("2"), sdk.MustNewDecFromStr("0.8"), sdk.MustNewDecFromStr("10")) borrowLimit := types.NewBorrowLimit(false, sdk.MustNewDecFromStr("0.2"), sdk.MustNewDecFromStr("0.5")) moneyMarket := types.NewMoneyMarket(denom, borrowLimit, denom+":usd", sdk.NewInt(1000000), sdk.NewInt(1000000000), model, sdk.MustNewDecFromStr("0.05"), sdk.ZeroDec()) _, f := suite.keeper.GetMoneyMarket(suite.ctx, denom) suite.Require().False(f) suite.keeper.SetMoneyMarket(suite.ctx, denom, moneyMarket) testMoneyMarket, f := suite.keeper.GetMoneyMarket(suite.ctx, denom) suite.Require().True(f) suite.Require().Equal(moneyMarket, testMoneyMarket) suite.Require().NotPanics(func() { suite.keeper.DeleteMoneyMarket(suite.ctx, denom) }) _, f = suite.keeper.GetMoneyMarket(suite.ctx, denom) suite.Require().False(f) } func (suite *KeeperTestSuite) TestIterateInterestRateModels() { testDenom := "test" var setMMs types.MoneyMarkets var setDenoms []string for i := 0; i < 5; i++ { // Initialize a new money market denom := testDenom + strconv.Itoa(i) model := types.NewInterestRateModel(sdk.MustNewDecFromStr("0.05"), sdk.MustNewDecFromStr("2"), sdk.MustNewDecFromStr("0.8"), sdk.MustNewDecFromStr("10")) borrowLimit := types.NewBorrowLimit(false, sdk.MustNewDecFromStr("0.2"), sdk.MustNewDecFromStr("0.5")) moneyMarket := types.NewMoneyMarket(denom, borrowLimit, denom+":usd", sdk.NewInt(1000000), sdk.NewInt(1000000000), model, sdk.MustNewDecFromStr("0.05"), sdk.ZeroDec()) // Store money market in the module's store suite.Require().NotPanics(func() { suite.keeper.SetMoneyMarket(suite.ctx, denom, moneyMarket) }) // Save the denom and model setDenoms = append(setDenoms, denom) setMMs = append(setMMs, moneyMarket) } var seenMMs types.MoneyMarkets var seenDenoms []string suite.keeper.IterateMoneyMarkets(suite.ctx, func(denom string, i types.MoneyMarket) bool { seenDenoms = append(seenDenoms, denom) seenMMs = append(seenMMs, i) return false }) suite.Require().Equal(setMMs, seenMMs) suite.Require().Equal(setDenoms, seenDenoms) } func (suite *KeeperTestSuite) TestSetDeleteLtvIndex() { // LTV index should have 0 items firstAddrs := suite.keeper.GetLtvIndexSlice(suite.ctx, 10) suite.Require().Equal(0, len(firstAddrs)) // Add an item to the LTV index addr := sdk.AccAddress("test") ltv := sdk.MustNewDecFromStr("1.1") suite.Require().NotPanics(func() { suite.keeper.InsertIntoLtvIndex(suite.ctx, ltv, addr) }) // LTV index should have 1 item secondAddrs := suite.keeper.GetLtvIndexSlice(suite.ctx, 10) suite.Require().Equal(1, len(secondAddrs)) // Attempt to remove invalid item from LTV index fakeLtv := sdk.MustNewDecFromStr("1.2") suite.Require().NotPanics(func() { suite.keeper.RemoveFromLtvIndex(suite.ctx, fakeLtv, addr) }) // LTV index should still have 1 item thirdAddrs := suite.keeper.GetLtvIndexSlice(suite.ctx, 10) suite.Require().Equal(1, len(thirdAddrs)) // Attempt to remove valid item from LTV index suite.Require().NotPanics(func() { suite.keeper.RemoveFromLtvIndex(suite.ctx, ltv, addr) }) // LTV index should still have 0 items fourthAddrs := suite.keeper.GetLtvIndexSlice(suite.ctx, 10) suite.Require().Equal(0, len(fourthAddrs)) } func (suite *KeeperTestSuite) TestIterateLtvIndex() { var setAddrs []sdk.AccAddress for i := 1; i <= 20; i++ { addr := sdk.AccAddress("test" + fmt.Sprint(i)) incrementalDec := sdk.NewDec(int64(i)).Quo(sdk.NewDec(10)) ltv := sdk.OneDec().Add(incrementalDec) // Set the ltv-address pair in the store suite.Require().NotPanics(func() { suite.keeper.InsertIntoLtvIndex(suite.ctx, ltv, addr) }) setAddrs = append(setAddrs, addr) } // Only the first 10 addresses should be returned sliceAddrs := suite.keeper.GetLtvIndexSlice(suite.ctx, 10) suite.Require().Equal(addressSort(setAddrs[10:20]), addressSort(sliceAddrs)) // Insert an additional item into the LTV index that should be returned in the first 10 elements addr := sdk.AccAddress("test" + fmt.Sprint(21)) ltv := sdk.OneDec().Add(sdk.MustNewDecFromStr("15").Quo(sdk.NewDec(10))) suite.Require().NotPanics(func() { suite.keeper.InsertIntoLtvIndex(suite.ctx, ltv, addr) }) // Fetch the updated LTV index updatedSliceAddrs := suite.keeper.GetLtvIndexSlice(suite.ctx, 10) sawAddr := false for _, updatedSliceAddr := range updatedSliceAddrs { if updatedSliceAddr.Equals(addr) { sawAddr = true } } suite.Require().Equal(true, sawAddr) } func (suite *KeeperTestSuite) getAccount(addr sdk.AccAddress) authexported.Account { ak := suite.app.GetAccountKeeper() return ak.GetAccount(suite.ctx, addr) } func (suite *KeeperTestSuite) getAccountAtCtx(addr sdk.AccAddress, ctx sdk.Context) authexported.Account { ak := suite.app.GetAccountKeeper() return ak.GetAccount(ctx, addr) } func (suite *KeeperTestSuite) getModuleAccount(name string) supplyexported.ModuleAccountI { sk := suite.app.GetSupplyKeeper() return sk.GetModuleAccount(suite.ctx, name) } func (suite *KeeperTestSuite) getModuleAccountAtCtx(name string, ctx sdk.Context) supplyexported.ModuleAccountI { sk := suite.app.GetSupplyKeeper() return sk.GetModuleAccount(ctx, name) } func addressSort(addrs []sdk.AccAddress) (sortedAddrs []sdk.AccAddress) { addrStrs := []string{} for _, addr := range addrs { addrStrs = append(addrStrs, addr.String()) } sort.Strings(addrStrs) for _, addrStr := range addrStrs { addr, _ := sdk.AccAddressFromBech32(addrStr) sortedAddrs = append(sortedAddrs, addr) } return sortedAddrs } func TestKeeperTestSuite(t *testing.T) { suite.Run(t, new(KeeperTestSuite)) }