From 0ee449e5a9aeb7fe038ce04d5806d0e131541c48 Mon Sep 17 00:00:00 2001 From: Denali Marsh Date: Thu, 9 Sep 2021 19:54:39 +0200 Subject: [PATCH] Test CDP init/export genesis (#1014) * implement basic init/export genesis test * update expected genesis for test --- x/cdp/genesis_test.go | 188 ++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 182 insertions(+), 6 deletions(-) diff --git a/x/cdp/genesis_test.go b/x/cdp/genesis_test.go index e12b2516..f49bb8a2 100644 --- a/x/cdp/genesis_test.go +++ b/x/cdp/genesis_test.go @@ -1,11 +1,15 @@ package cdp_test import ( + "fmt" + "sort" "strings" "testing" "time" "github.com/stretchr/testify/suite" + abci "github.com/tendermint/tendermint/abci/types" + tmtime "github.com/tendermint/tendermint/types/time" sdk "github.com/cosmos/cosmos-sdk/types" @@ -16,8 +20,22 @@ import ( type GenesisTestSuite struct { suite.Suite - ctx sdk.Context - keeper cdp.Keeper + app app.TestApp + ctx sdk.Context + genTime time.Time + keeper cdp.Keeper + addrs []sdk.AccAddress +} + +func (suite *GenesisTestSuite) SetupTest() { + tApp := app.NewTestApp() + suite.genTime = tmtime.Canonical(time.Date(2021, 1, 1, 1, 1, 1, 1, time.UTC)) + suite.ctx = tApp.NewContext(true, abci.Header{Height: 1, Time: suite.genTime}) + suite.keeper = tApp.GetCDPKeeper() + suite.app = tApp + + _, addrs := app.GeneratePrivKeyAddressPairs(3) + suite.addrs = addrs } func (suite *GenesisTestSuite) TestInvalidGenState() { @@ -126,10 +144,8 @@ func (suite *GenesisTestSuite) TestInvalidGenState() { } func (suite *GenesisTestSuite) TestValidGenState() { - tApp := app.NewTestApp() - suite.NotPanics(func() { - tApp.InitializeFromGenesisStates( + suite.app.InitializeFromGenesisStates( NewPricefeedGenStateMulti(), NewCDPGenStateMulti(), ) @@ -142,13 +158,173 @@ func (suite *GenesisTestSuite) TestValidGenState() { gs.StartingCdpID = uint64(5) appGS := app.GenesisState{"cdp": cdp.ModuleCdc.MustMarshalJSON(gs)} suite.NotPanics(func() { - tApp.InitializeFromGenesisStates( + suite.app.InitializeFromGenesisStates( NewPricefeedGenStateMulti(), appGS, ) }) } +func (suite *GenesisTestSuite) Test_InitExportGenesis() { + + cdps := cdp.CDPs{ + { + ID: 2, + Owner: suite.addrs[0], + Type: "xrp-a", + Collateral: c("xrp", 200000000), + Principal: c("usdx", 10000000), + AccumulatedFees: c("usdx", 0), + FeesUpdated: suite.genTime, + InterestFactor: sdk.NewDec(1), + }, + } + + genTotalPrincipals := cdp.GenesisTotalPrincipals{ + cdp.NewGenesisTotalPrincipal("btc-a", sdk.ZeroInt()), + cdp.NewGenesisTotalPrincipal("xrp-a", sdk.ZeroInt()), + } + + var deposits cdp.Deposits + for _, c := range cdps { + deposit := cdp.Deposit{ + CdpID: c.ID, + Depositor: c.Owner, + Amount: c.Collateral, + } + deposits = append(deposits, deposit) + + for i, p := range genTotalPrincipals { + if p.CollateralType == c.Type { + genTotalPrincipals[i].TotalPrincipal = genTotalPrincipals[i].TotalPrincipal.Add(c.Principal.Amount) + } + } + } + + cdpGenesis := cdp.GenesisState{ + Params: cdp.Params{ + GlobalDebtLimit: sdk.NewInt64Coin("usdx", 1000000000000), + SurplusAuctionThreshold: cdp.DefaultSurplusThreshold, + SurplusAuctionLot: cdp.DefaultSurplusLot, + DebtAuctionThreshold: cdp.DefaultDebtThreshold, + DebtAuctionLot: cdp.DefaultDebtLot, + CollateralParams: cdp.CollateralParams{ + { + Denom: "xrp", + Type: "xrp-a", + LiquidationRatio: sdk.MustNewDecFromStr("2.0"), + DebtLimit: sdk.NewInt64Coin("usdx", 500000000000), + StabilityFee: sdk.MustNewDecFromStr("1.000000001547125958"), // 5% apr + LiquidationPenalty: d("0.05"), + AuctionSize: i(7000000000), + Prefix: 0x20, + SpotMarketID: "xrp:usd", + LiquidationMarketID: "xrp:usd", + KeeperRewardPercentage: d("0.01"), + CheckCollateralizationIndexCount: i(10), + ConversionFactor: i(6), + }, + { + Denom: "btc", + Type: "btc-a", + LiquidationRatio: sdk.MustNewDecFromStr("1.5"), + DebtLimit: sdk.NewInt64Coin("usdx", 500000000000), + StabilityFee: sdk.MustNewDecFromStr("1.000000000782997609"), // 2.5% apr + LiquidationPenalty: d("0.025"), + AuctionSize: i(10000000), + Prefix: 0x21, + SpotMarketID: "btc:usd", + LiquidationMarketID: "btc:usd", + KeeperRewardPercentage: d("0.01"), + CheckCollateralizationIndexCount: i(10), + ConversionFactor: i(8), + }, + }, + DebtParam: cdp.DebtParam{ + Denom: "usdx", + ReferenceAsset: "usd", + ConversionFactor: i(6), + DebtFloor: i(10000000), + }, + }, + StartingCdpID: cdp.DefaultCdpStartingID, + DebtDenom: cdp.DefaultDebtDenom, + GovDenom: cdp.DefaultGovDenom, + CDPs: cdps, + Deposits: deposits, + PreviousAccumulationTimes: cdp.GenesisAccumulationTimes{ + cdp.NewGenesisAccumulationTime("btc-a", suite.genTime, sdk.OneDec()), + cdp.NewGenesisAccumulationTime("xrp-a", suite.genTime, sdk.OneDec()), + }, + TotalPrincipals: genTotalPrincipals, + } + + suite.NotPanics(func() { + suite.app.InitializeFromGenesisStatesWithTime( + suite.genTime, + NewPricefeedGenStateMulti(), + app.GenesisState{cdp.ModuleName: cdp.ModuleCdc.MustMarshalJSON(cdpGenesis)}, + ) + }) + + // We run the BeginBlocker at time.Now() to accumulate interest + suite.ctx = suite.ctx.WithBlockTime(time.Now()) + cdp.BeginBlocker(suite.ctx, abci.RequestBeginBlock{}, suite.keeper) + + expectedGenesis := cdpGenesis + + // Update previous accrual times in expected genesis + var expectedPrevAccTimes cdp.GenesisAccumulationTimes + for _, prevAccTime := range cdpGenesis.PreviousAccumulationTimes { + time, found := suite.keeper.GetPreviousAccrualTime(suite.ctx, prevAccTime.CollateralType) + if !found { + panic(fmt.Sprintf("couldn't find previous accrual time for %s", prevAccTime.CollateralType)) + } + prevAccTime.PreviousAccumulationTime = time + + interestFactor, found := suite.keeper.GetInterestFactor(suite.ctx, prevAccTime.CollateralType) + if !found { + panic(fmt.Sprintf("couldn't find interest factor for %s", prevAccTime.CollateralType)) + } + prevAccTime.InterestFactor = interestFactor + + expectedPrevAccTimes = append(expectedPrevAccTimes, prevAccTime) + } + expectedGenesis.PreviousAccumulationTimes = expectedPrevAccTimes + + // Update total principals + var totalPrincipals cdp.GenesisTotalPrincipals + for _, p := range expectedGenesis.TotalPrincipals { + totalPrincipal := suite.keeper.GetTotalPrincipal(suite.ctx, p.CollateralType, "usdx") + p.TotalPrincipal = totalPrincipal + totalPrincipals = append(totalPrincipals, p) + } + expectedGenesis.TotalPrincipals = totalPrincipals + + // Update CDPs + expectedGenesis.CDPs = suite.keeper.GetAllCdps(suite.ctx) + + exportedGenesis := cdp.ExportGenesis(suite.ctx, suite.keeper) + + // Sort TotalPrincipals in both genesis files so slice order matches + sort.SliceStable(expectedGenesis.TotalPrincipals, func(i, j int) bool { + return expectedGenesis.TotalPrincipals[i].CollateralType < expectedGenesis.TotalPrincipals[j].CollateralType + }) + sort.SliceStable(exportedGenesis.TotalPrincipals, func(i, j int) bool { + return exportedGenesis.TotalPrincipals[i].CollateralType < exportedGenesis.TotalPrincipals[j].CollateralType + }) + + // Sort PreviousAccumulationTimes in both genesis files so slice order matches + sort.SliceStable(expectedGenesis.PreviousAccumulationTimes, func(i, j int) bool { + return expectedGenesis.PreviousAccumulationTimes[i].CollateralType < expectedGenesis.PreviousAccumulationTimes[j].CollateralType + }) + sort.SliceStable(exportedGenesis.PreviousAccumulationTimes, func(i, j int) bool { + return exportedGenesis.PreviousAccumulationTimes[i].CollateralType < exportedGenesis.PreviousAccumulationTimes[j].CollateralType + }) + + suite.Equal(expectedGenesis, exportedGenesis) +} + func TestGenesisTestSuite(t *testing.T) { suite.Run(t, new(GenesisTestSuite)) }