mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-29 09:45:18 +00:00
119 lines
3.5 KiB
Go
119 lines
3.5 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
"github.com/stretchr/testify/require"
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
|
|
|
"github.com/kava-labs/kava/app"
|
|
"github.com/kava-labs/kava/x/liquidator/types"
|
|
)
|
|
|
|
func TestKeeper_SeizeAndStartCollateralAuction(t *testing.T) {
|
|
// Setup
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(1)
|
|
|
|
tApp := app.NewTestApp()
|
|
tApp.InitializeFromGenesisStates(
|
|
app.NewAuthGenState(addrs, []sdk.Coins{cs(c("btc", 100))}),
|
|
NewPFGenState("btc", sdk.MustNewDecFromStr("8000.00")),
|
|
NewCDPGenState(),
|
|
NewLiquidatorGenState(),
|
|
)
|
|
ctx := tApp.NewContext(false, abci.Header{})
|
|
|
|
require.NoError(t, tApp.GetCDPKeeper().ModifyCDP(ctx, addrs[0], "btc", i(3), i(16000)))
|
|
|
|
_, err := tApp.GetPriceFeedKeeper().SetPrice(ctx, addrs[0], "btc", sdk.MustNewDecFromStr("7999.99"), time.Unix(9999999999, 0))
|
|
require.NoError(t, err)
|
|
require.NoError(t, tApp.GetPriceFeedKeeper().SetCurrentPrices(ctx, "btc"))
|
|
|
|
// Run test function
|
|
auctionID, err := tApp.GetLiquidatorKeeper().SeizeAndStartCollateralAuction(ctx, addrs[0], "btc")
|
|
|
|
// Check CDP
|
|
require.NoError(t, err)
|
|
cdp, found := tApp.GetCDPKeeper().GetCDP(ctx, addrs[0], "btc")
|
|
require.True(t, found)
|
|
require.Equal(t, cdp.CollateralAmount, i(2)) // original amount - params.CollateralAuctionSize
|
|
require.Equal(t, cdp.Debt, i(10667)) // original debt scaled by amount of collateral removed
|
|
// Check auction exists
|
|
_, found = tApp.GetAuctionKeeper().GetAuction(ctx, auctionID)
|
|
require.True(t, found)
|
|
}
|
|
|
|
func TestKeeper_StartDebtAuction(t *testing.T) {
|
|
// Setup
|
|
tApp := app.NewTestApp()
|
|
tApp.InitializeFromGenesisStates(
|
|
NewLiquidatorGenState(),
|
|
)
|
|
keeper := tApp.GetLiquidatorKeeper()
|
|
ctx := tApp.NewContext(false, abci.Header{})
|
|
|
|
initSDebt := types.SeizedDebt{i(2000), i(0)}
|
|
keeper.SetSeizedDebt(ctx, initSDebt)
|
|
|
|
// Execute
|
|
auctionID, err := keeper.StartDebtAuction(ctx)
|
|
|
|
// Check
|
|
require.NoError(t, err)
|
|
require.Equal(t,
|
|
types.SeizedDebt{
|
|
initSDebt.Total,
|
|
initSDebt.SentToAuction.Add(keeper.GetParams(ctx).DebtAuctionSize),
|
|
},
|
|
keeper.GetSeizedDebt(ctx),
|
|
)
|
|
_, found := tApp.GetAuctionKeeper().GetAuction(ctx, auctionID)
|
|
require.True(t, found)
|
|
}
|
|
|
|
func TestKeeper_partialSeizeCDP(t *testing.T) {
|
|
// Setup
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(1)
|
|
|
|
tApp := app.NewTestApp()
|
|
tApp.InitializeFromGenesisStates(
|
|
app.NewAuthGenState(addrs, []sdk.Coins{cs(c("btc", 100))}),
|
|
NewPFGenState("btc", sdk.MustNewDecFromStr("8000.00")),
|
|
NewCDPGenState(),
|
|
NewLiquidatorGenState(),
|
|
)
|
|
ctx := tApp.NewContext(false, abci.Header{})
|
|
|
|
tApp.GetCDPKeeper().ModifyCDP(ctx, addrs[0], "btc", i(3), i(16000))
|
|
|
|
tApp.GetPriceFeedKeeper().SetPrice(ctx, addrs[0], "btc", sdk.MustNewDecFromStr("7999.99"), tmtime.Now().Add(time.Hour*1))
|
|
tApp.GetPriceFeedKeeper().SetCurrentPrices(ctx, "btc")
|
|
|
|
// Run test function
|
|
err := tApp.GetLiquidatorKeeper().PartialSeizeCDP(ctx, addrs[0], "btc", i(2), i(10000))
|
|
|
|
// Check
|
|
require.NoError(t, err)
|
|
cdp, found := tApp.GetCDPKeeper().GetCDP(ctx, addrs[0], "btc")
|
|
require.True(t, found)
|
|
require.Equal(t, i(1), cdp.CollateralAmount)
|
|
require.Equal(t, i(6000), cdp.Debt)
|
|
}
|
|
|
|
func TestKeeper_GetSetSeizedDebt(t *testing.T) {
|
|
// Setup
|
|
tApp := app.NewTestApp()
|
|
ctx := tApp.NewContext(true, abci.Header{})
|
|
debt := types.SeizedDebt{i(234247645), i(2343)}
|
|
|
|
// Run test function
|
|
tApp.GetLiquidatorKeeper().SetSeizedDebt(ctx, debt)
|
|
readDebt := tApp.GetLiquidatorKeeper().GetSeizedDebt(ctx)
|
|
|
|
// Check
|
|
require.Equal(t, debt, readDebt)
|
|
}
|