Test CDP init/export genesis (#1014)

* implement basic init/export genesis test

* update expected genesis for test

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
 
+	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))
 }