package cdp_test import ( "math/rand" "testing" "time" "github.com/stretchr/testify/suite" sdk "github.com/cosmos/cosmos-sdk/types" "github.com/cosmos/cosmos-sdk/types/simulation" abci "github.com/cometbft/cometbft/abci/types" tmproto "github.com/cometbft/cometbft/proto/tendermint/types" tmtime "github.com/cometbft/cometbft/types/time" "github.com/kava-labs/kava/app" auctiontypes "github.com/kava-labs/kava/x/auction/types" "github.com/kava-labs/kava/x/cdp" "github.com/kava-labs/kava/x/cdp/keeper" "github.com/kava-labs/kava/x/cdp/types" ) type ModuleTestSuite struct { suite.Suite keeper keeper.Keeper addrs []sdk.AccAddress app app.TestApp cdps types.CDPs ctx sdk.Context liquidations liquidationTracker } type liquidationTracker struct { xrp []uint64 btc []uint64 debt int64 } func (suite *ModuleTestSuite) SetupTest() { tApp := app.NewTestApp() ctx := tApp.NewContext(true, tmproto.Header{Height: 1, Time: tmtime.Now()}) tracker := liquidationTracker{} coins := cs(c("btc", 100000000), c("xrp", 10000000000)) _, addrs := app.GeneratePrivKeyAddressPairs(100) authGS := app.NewFundedGenStateWithSameCoins(tApp.AppCodec(), coins, addrs) tApp.InitializeFromGenesisStates( authGS, NewPricefeedGenStateMulti(tApp.AppCodec()), NewCDPGenStateMulti(tApp.AppCodec()), ) suite.ctx = ctx suite.app = tApp suite.keeper = tApp.GetCDPKeeper() suite.cdps = types.CDPs{} suite.addrs = addrs suite.liquidations = tracker } func (suite *ModuleTestSuite) createCdps() { tApp := app.NewTestApp() ctx := tApp.NewContext(true, tmproto.Header{Height: 1, Time: tmtime.Now()}) cdps := make(types.CDPs, 100) tracker := liquidationTracker{} coins := cs(c("btc", 100000000), c("xrp", 10000000000)) _, addrs := app.GeneratePrivKeyAddressPairs(100) authGS := app.NewFundedGenStateWithSameCoins(tApp.AppCodec(), coins, addrs) tApp.InitializeFromGenesisStates( authGS, NewPricefeedGenStateMulti(tApp.AppCodec()), NewCDPGenStateMulti(tApp.AppCodec()), ) suite.ctx = ctx suite.app = tApp suite.keeper = tApp.GetCDPKeeper() // create 100 cdps for j := 0; j < 100; j++ { // 50 of the cdps will be collateralized with xrp collateral := "xrp" amount := 10000000000 debt := simulation.RandIntBetween(rand.New(rand.NewSource(int64(j))), 750000000, 1249000000) // the other half (50) will be collateralized with btc if j%2 == 0 { collateral = "btc" amount = 100000000 debt = simulation.RandIntBetween(rand.New(rand.NewSource(int64(j))), 2700000000, 5332000000) if debt >= 4000000000 { tracker.btc = append(tracker.btc, uint64(j+1)) tracker.debt += int64(debt) } } else { if debt >= 1000000000 { tracker.xrp = append(tracker.xrp, uint64(j+1)) tracker.debt += int64(debt) } } suite.Nil(suite.keeper.AddCdp(suite.ctx, addrs[j], c(collateral, int64(amount)), c("usdx", int64(debt)), collateral+"-a")) c, f := suite.keeper.GetCDP(suite.ctx, collateral+"-a", uint64(j+1)) suite.True(f) cdps[j] = c } suite.cdps = cdps suite.addrs = addrs suite.liquidations = tracker } func (suite *ModuleTestSuite) setPrice(price sdk.Dec, market string) { pfKeeper := suite.app.GetPriceFeedKeeper() _, err := pfKeeper.SetPrice(suite.ctx, sdk.AccAddress{}, market, price, suite.ctx.BlockTime().Add(time.Hour*3)) suite.NoError(err) err = pfKeeper.SetCurrentPrices(suite.ctx, market) suite.NoError(err) pp, err := pfKeeper.GetCurrentPrice(suite.ctx, market) suite.NoError(err) suite.Equal(price, pp.Price) } func (suite *ModuleTestSuite) TestBeginBlock() { // test setup, creating // 50 xrp cdps each with // collateral: 10000000000 // debt: between 750000000 - 1249000000 // if debt above 10000000000, // cdp added to tracker / liquidation list // debt total added to trackers debt total // 50 btc cdps each with // collateral: 10000000000 // debt: between 2700000000 - 5332000000 // if debt above 4000000000, // cdp added to tracker / liquidation list // debt total added to trackers debt total // naively we expect roughly half of the cdps to be above the debt tracking floor, roughly 25 of them collaterallized with xrp, the other 25 with btcb // usdx is the principal for all cdps suite.createCdps() ak := suite.app.GetAccountKeeper() bk := suite.app.GetBankKeeper() // test case 1 setup acc := ak.GetModuleAccount(suite.ctx, types.ModuleName) // track how much xrp collateral exists in the cdp module originalXrpCollateral := bk.GetBalance(suite.ctx, acc.GetAddress(), "xrp").Amount // set the trading price for xrp:usd pools suite.setPrice(d("0.2"), "xrp:usd") // test case 1 execution cdp.BeginBlocker(suite.ctx, abci.RequestBeginBlock{Header: suite.ctx.BlockHeader()}, suite.keeper) // test case 1 assert acc = ak.GetModuleAccount(suite.ctx, types.ModuleName) // get the current amount of xrp held by the cdp module finalXrpCollateral := bk.GetBalance(suite.ctx, acc.GetAddress(), "xrp").Amount seizedXrpCollateral := originalXrpCollateral.Sub(finalXrpCollateral) // calculate the number of cdps that were liquidated based on the total // seized collateral divided by the size of each cdp when it was created xrpLiquidations := int(seizedXrpCollateral.Quo(i(10000000000)).Int64()) // should be 10 because...? suite.Equal(10, xrpLiquidations) // btc collateral test case setup acc = ak.GetModuleAccount(suite.ctx, types.ModuleName) originalBtcCollateral := bk.GetBalance(suite.ctx, acc.GetAddress(), "btc").Amount // set the trading price for btc:usd pools suite.setPrice(d("6000"), "btc:usd") // btc collateral test case execution cdp.BeginBlocker(suite.ctx, abci.RequestBeginBlock{Header: suite.ctx.BlockHeader()}, suite.keeper) // btc collateral test case assertion 1 acc = ak.GetModuleAccount(suite.ctx, types.ModuleName) finalBtcCollateral := bk.GetBalance(suite.ctx, acc.GetAddress(), "btc").Amount seizedBtcCollateral := originalBtcCollateral.Sub(finalBtcCollateral) // calculate the number of btc cdps that were liquidated based on the // total seized collateral divided by the fixed size of each cdp // when it was created during test setup btcLiquidations := int(seizedBtcCollateral.Quo(i(100000000)).Int64()) suite.Equal(10, btcLiquidations) // btc collateral test case assertion 2 // test that the auction module has a balance equal to the amount of collateral seized acc = ak.GetModuleAccount(suite.ctx, auctiontypes.ModuleName) // should be this exact value because...? suite.Equal(int64(71955653865), bk.GetBalance(suite.ctx, acc.GetAddress(), "debt").Amount.Int64()) } func (suite *ModuleTestSuite) TestSeizeSingleCdpWithFees() { // test setup // starting with zero cdps, add a single cdp of // xrp backed 1:1 with usdx err := suite.keeper.AddCdp(suite.ctx, suite.addrs[0], c("xrp", 10000000000), c("usdx", 1000000000), "xrp-a") suite.NoError(err) // verify the total value of all assets in cdps composed of xrp-a/usdx pair equals the amount of the single cdp we just added above suite.Equal(i(1000000000), suite.keeper.GetTotalPrincipal(suite.ctx, "xrp-a", "usdx")) ak := suite.app.GetAccountKeeper() bk := suite.app.GetBankKeeper() cdpMacc := ak.GetModuleAccount(suite.ctx, types.ModuleName) suite.Equal(i(1000000000), bk.GetBalance(suite.ctx, cdpMacc.GetAddress(), "debt").Amount) for i := 0; i < 100; i++ { suite.ctx = suite.ctx.WithBlockTime(suite.ctx.BlockTime().Add(time.Second * 6)) cdp.BeginBlocker(suite.ctx, abci.RequestBeginBlock{Header: suite.ctx.BlockHeader()}, suite.keeper) } cdpMacc = ak.GetModuleAccount(suite.ctx, types.ModuleName) suite.Equal(i(1000000891), (bk.GetBalance(suite.ctx, cdpMacc.GetAddress(), "debt").Amount)) cdp, _ := suite.keeper.GetCDP(suite.ctx, "xrp-a", 1) err = suite.keeper.SeizeCollateral(suite.ctx, cdp) suite.NoError(err) _, found := suite.keeper.GetCDP(suite.ctx, "xrp-a", 1) suite.False(found) } func (suite *ModuleTestSuite) TestCDPBeginBlockerRunsOnlyOnConfiguredInterval() { // test setup, creating // 50 xrp cdps each with // collateral: 10000000000 // debt: between 750000000 - 1249000000 // if debt above 10000000000, // cdp added to tracker / liquidation list // debt total added to trackers debt total // 50 btc cdps each with // collateral: 10000000000 // debt: between 2700000000 - 5332000000 // if debt above 4000000000, // cdp added to tracker / liquidation list // debt total added to trackers debt total // naively we expect roughly half of the cdps to be above the debt tracking floor, roughly 25 of them collaterallized with xrp, the other 25 with btcb // usdx is the principal for all cdps suite.createCdps() ak := suite.app.GetAccountKeeper() bk := suite.app.GetBankKeeper() // set the cdp begin blocker to run every other block params := suite.keeper.GetParams(suite.ctx) params.LiquidationBlockInterval = 2 suite.keeper.SetParams(suite.ctx, params) // test case 1 setup acc := ak.GetModuleAccount(suite.ctx, types.ModuleName) // track how much xrp collateral exists in the cdp module originalXrpCollateral := bk.GetBalance(suite.ctx, acc.GetAddress(), "xrp").Amount // set the trading price for xrp:usd pools suite.setPrice(d("0.2"), "xrp:usd") // test case 1 execution cdp.BeginBlocker(suite.ctx, abci.RequestBeginBlock{Header: suite.ctx.BlockHeader()}, suite.keeper) // test case 1 assert acc = ak.GetModuleAccount(suite.ctx, types.ModuleName) // get the current amount of xrp held by the cdp module finalXrpCollateral := bk.GetBalance(suite.ctx, acc.GetAddress(), "xrp").Amount seizedXrpCollateral := originalXrpCollateral.Sub(finalXrpCollateral) // calculate the number of cdps that were liquidated based on the total // seized collateral divided by the size of each cdp when it was created xrpLiquidations := int(seizedXrpCollateral.Quo(i(10000000000)).Int64()) // should be 0 because the cdp begin blocker is configured to // skip execution every odd numbered block suite.Equal(0, xrpLiquidations, "expected cdp begin blocker not to run liqudations") // test case 2 setup // simulate running the second block of the chain suite.ctx = suite.ctx.WithBlockHeight(2) // test case 2 execution cdp.BeginBlocker(suite.ctx, abci.RequestBeginBlock{Header: suite.ctx.BlockHeader()}, suite.keeper) // test case 2 assert acc = ak.GetModuleAccount(suite.ctx, types.ModuleName) // get the current amount of xrp held by the cdp module finalXrpCollateral = bk.GetBalance(suite.ctx, acc.GetAddress(), "xrp").Amount seizedXrpCollateral = originalXrpCollateral.Sub(finalXrpCollateral) // calculate the number of cdps that were liquidated based on the total // seized collateral divided by the size of each cdp when it was created xrpLiquidations = int(seizedXrpCollateral.Quo(i(10000000000)).Int64()) suite.Greater(xrpLiquidations, 0, "expected cdp begin blocker to run liquidations") } func TestModuleTestSuite(t *testing.T) { suite.Run(t, new(ModuleTestSuite)) }