mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-24 23:35:19 +00:00
Test CDP init/export genesis (#1014)
* implement basic init/export genesis test * update expected genesis for test
This commit is contained in:
Denali Marsh
GitHub
parent
6710e424ee
commit
0ee449e5a9
@ -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))
|
||||
}
|
||||
|
||||
Loading…
Reference in New Issue
Block a user