0g-chain/x/auction/abci_test.go

67 lines
2.3 KiB
Go
Raw Normal View History

2019-12-05 13:53:28 +00:00
package auction_test
2019-11-25 19:46:02 +00:00
import (
"testing"
sdk "github.com/cosmos/cosmos-sdk/types"
"github.com/cosmos/cosmos-sdk/x/auth"
authexported "github.com/cosmos/cosmos-sdk/x/auth/exported"
"github.com/cosmos/cosmos-sdk/x/supply"
2019-11-25 19:46:02 +00:00
"github.com/stretchr/testify/require"
abci "github.com/tendermint/tendermint/abci/types"
2019-12-05 13:53:28 +00:00
"github.com/kava-labs/kava/app"
"github.com/kava-labs/kava/x/auction"
"github.com/kava-labs/kava/x/cdp"
2019-11-25 19:46:02 +00:00
)
func TestKeeper_EndBlocker(t *testing.T) {
2019-12-05 13:53:28 +00:00
// Setup
_, addrs := app.GeneratePrivKeyAddressPairs(2)
buyer := addrs[0]
returnAddrs := addrs[1:]
returnWeights := []sdk.Int{sdk.NewInt(1)}
sellerModName := cdp.LiquidatorMacc
2019-12-05 13:53:28 +00:00
tApp := app.NewTestApp()
sellerAcc := supply.NewEmptyModuleAccount(sellerModName)
2020-01-12 14:17:47 +00:00
require.NoError(t, sellerAcc.SetCoins(cs(c("token1", 100), c("token2", 100), c("debt", 100))))
2019-12-07 01:25:45 +00:00
tApp.InitializeFromGenesisStates(
NewAuthGenStateFromAccs(authexported.GenesisAccounts{
auth.NewBaseAccount(buyer, cs(c("token1", 100), c("token2", 100)), nil, 0, 0),
sellerAcc,
}),
2019-12-07 01:25:45 +00:00
)
2019-12-05 13:53:28 +00:00
ctx := tApp.NewContext(true, abci.Header{})
keeper := tApp.GetAuctionKeeper()
2019-11-25 19:46:02 +00:00
2020-01-12 14:17:47 +00:00
auctionID, err := keeper.StartCollateralAuction(ctx, sellerModName, c("token1", 20), c("token2", 50), returnAddrs, returnWeights, c("debt", 40))
2019-12-05 13:53:28 +00:00
require.NoError(t, err)
require.NoError(t, keeper.PlaceBid(ctx, auctionID, buyer, c("token2", 30)))
2019-11-25 19:46:02 +00:00
// Run the endblocker, simulating a block time 1ns before auction expiry
preExpiryTime := ctx.BlockTime().Add(auction.DefaultBidDuration - 1)
auction.EndBlocker(ctx.WithBlockTime(preExpiryTime), keeper)
2019-11-25 19:46:02 +00:00
2019-12-05 13:53:28 +00:00
// Check auction has not been closed yet
_, found := keeper.GetAuction(ctx, auctionID)
require.True(t, found)
// Run the endblocker, simulating a block time equal to auction expiry
expiryTime := ctx.BlockTime().Add(auction.DefaultBidDuration)
auction.EndBlocker(ctx.WithBlockTime(expiryTime), keeper)
2019-12-05 13:53:28 +00:00
// Check auction has been closed
_, found = keeper.GetAuction(ctx, auctionID)
2019-11-25 19:46:02 +00:00
require.False(t, found)
}
2019-12-05 13:53:28 +00:00
func c(denom string, amount int64) sdk.Coin { return sdk.NewInt64Coin(denom, amount) }
func cs(coins ...sdk.Coin) sdk.Coins { return sdk.NewCoins(coins...) }
func NewAuthGenStateFromAccs(accounts authexported.GenesisAccounts) app.GenesisState {
authGenesis := auth.NewGenesisState(auth.DefaultParams(), accounts)
return app.GenesisState{auth.ModuleName: auth.ModuleCdc.MustMarshalJSON(authGenesis)}
}