mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-11-14 03:55:19 +00:00
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
|
package keeper_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"math/rand"
|
||
|
|
"strings"
|
||
|
|
"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/simulation"
|
||
|
|
"github.com/cosmos/cosmos-sdk/x/supply"
|
||
|
|
"github.com/kava-labs/kava/app"
|
||
|
|
"github.com/kava-labs/kava/x/auction/keeper"
|
||
|
|
"github.com/kava-labs/kava/x/auction/types"
|
||
|
|
"github.com/kava-labs/kava/x/cdp"
|
||
|
|
"github.com/stretchr/testify/suite"
|
||
|
|
abci "github.com/tendermint/tendermint/abci/types"
|
||
|
|
tmtime "github.com/tendermint/tendermint/types/time"
|
||
|
|
)
|
||
|
|
|
||
|
|
const (
|
||
|
|
custom = "custom"
|
||
|
|
TestAuctionCount = 10
|
||
|
|
)
|
||
|
|
|
||
|
|
type QuerierTestSuite struct {
|
||
|
|
suite.Suite
|
||
|
|
|
||
|
|
keeper keeper.Keeper
|
||
|
|
app app.TestApp
|
||
|
|
auctions types.Auctions
|
||
|
|
ctx sdk.Context
|
||
|
|
querier sdk.Querier
|
||
|
|
}
|
||
|
|
|
||
|
|
func (suite *QuerierTestSuite) SetupTest() {
|
||
|
|
tApp := app.NewTestApp()
|
||
|
|
ctx := tApp.NewContext(true, abci.Header{Height: 1, Time: tmtime.Now()})
|
||
|
|
|
||
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(1)
|
||
|
|
buyer := addrs[0]
|
||
|
|
modName := cdp.LiquidatorMacc
|
||
|
|
|
||
|
|
// Set up seller account
|
||
|
|
sellerAcc := supply.NewEmptyModuleAccount(modName, supply.Minter, supply.Burner)
|
||
|
|
sellerAcc.SetCoins(cs(c("token1", 1000), c("token2", 1000), c("debt", 1000)))
|
||
|
|
|
||
|
|
// Initialize genesis accounts
|
||
|
|
tApp.InitializeFromGenesisStates(
|
||
|
|
NewAuthGenStateFromAccs(authexported.GenesisAccounts{
|
||
|
|
auth.NewBaseAccount(buyer, cs(c("token1", 1000), c("token2", 1000)), nil, 0, 0),
|
||
|
|
sellerAcc,
|
||
|
|
}),
|
||
|
|
)
|
||
|
|
|
||
|
|
suite.ctx = ctx
|
||
|
|
suite.app = tApp
|
||
|
|
suite.keeper = tApp.GetAuctionKeeper()
|
||
|
|
|
||
|
|
// Populate with auctions
|
||
|
|
for j := 0; j < TestAuctionCount; j++ {
|
||
|
|
lotAmount := simulation.RandIntBetween(rand.New(rand.NewSource(int64(j))), 10, 100)
|
||
|
|
id, err := suite.keeper.StartSurplusAuction(suite.ctx, modName, c("token1", int64(lotAmount)), "token2")
|
||
|
|
suite.Nil(err)
|
||
|
|
|
||
|
|
auc, found := suite.keeper.GetAuction(suite.ctx, id)
|
||
|
|
suite.True(found)
|
||
|
|
suite.auctions = append(suite.auctions, auc)
|
||
|
|
}
|
||
|
|
|
||
|
|
suite.querier = keeper.NewQuerier(suite.keeper)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (suite *QuerierTestSuite) TestQueryAuctions() {
|
||
|
|
ctx := suite.ctx.WithIsCheckTx(false)
|
||
|
|
// Set up request query
|
||
|
|
query := abci.RequestQuery{
|
||
|
|
Path: strings.Join([]string{custom, types.QuerierRoute, types.QueryGetAuction}, "/"),
|
||
|
|
Data: types.ModuleCdc.MustMarshalJSON(types.NewQueryAllAuctionParams(1, TestAuctionCount)),
|
||
|
|
}
|
||
|
|
|
||
|
|
// Execute query and check the []byte result
|
||
|
|
bz, err := suite.querier(ctx, []string{types.QueryGetAuction}, query)
|
||
|
|
suite.NoError(err)
|
||
|
|
suite.NotNil(bz)
|
||
|
|
|
||
|
|
// Unmarshal the bytes into type Auctions
|
||
|
|
var auctions types.Auctions
|
||
|
|
suite.Nil(types.ModuleCdc.UnmarshalJSON(bz, &auctions))
|
||
|
|
|
||
|
|
// Check that each Auction has correct values
|
||
|
|
for i := 0; i < TestAuctionCount; i++ {
|
||
|
|
suite.Equal(suite.auctions[i].GetID(), auctions[i].GetID())
|
||
|
|
suite.Equal(suite.auctions[i].GetInitiator(), auctions[i].GetInitiator())
|
||
|
|
suite.Equal(suite.auctions[i].GetLot(), auctions[i].GetLot())
|
||
|
|
suite.Equal(suite.auctions[i].GetBid(), auctions[i].GetBid())
|
||
|
|
suite.Equal(suite.auctions[i].GetEndTime(), auctions[i].GetEndTime())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestQuerierTestSuite(t *testing.T) {
|
||
|
|
suite.Run(t, new(QuerierTestSuite))
|
||
|
|
}
|