mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-26 16:25:21 +00:00
138 lines
3.0 KiB
Go
138 lines
3.0 KiB
Go
package keeper_test
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
sdkmath "cosmossdk.io/math"
|
|
tmproto "github.com/cometbft/cometbft/proto/tendermint/types"
|
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
|
|
|
"github.com/0glabs/0g-chain/app"
|
|
"github.com/0glabs/0g-chain/x/auction/keeper"
|
|
"github.com/0glabs/0g-chain/x/auction/types"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestGrpcAuctionsFilter(t *testing.T) {
|
|
// setup
|
|
tApp := app.NewTestApp()
|
|
tApp.InitializeFromGenesisStates()
|
|
auctionsKeeper := tApp.GetAuctionKeeper()
|
|
ctx := tApp.NewContext(true, tmproto.Header{Height: 1})
|
|
_, addrs := app.GeneratePrivKeyAddressPairs(2)
|
|
|
|
auctions := []types.Auction{
|
|
types.NewSurplusAuction(
|
|
"sellerMod",
|
|
c("swp", 12345678),
|
|
"usdx",
|
|
time.Date(1998, time.January, 1, 0, 0, 0, 0, time.UTC),
|
|
).WithID(0),
|
|
types.NewDebtAuction(
|
|
"buyerMod",
|
|
c("hard", 12345678),
|
|
c("usdx", 12345678),
|
|
time.Date(1998, time.January, 1, 0, 0, 0, 0, time.UTC),
|
|
c("debt", 12345678),
|
|
).WithID(1),
|
|
types.NewCollateralAuction(
|
|
"sellerMod",
|
|
c("ukava", 12345678),
|
|
time.Date(1998, time.January, 1, 0, 0, 0, 0, time.UTC),
|
|
c("usdx", 12345678),
|
|
types.WeightedAddresses{
|
|
Addresses: addrs,
|
|
Weights: []sdkmath.Int{sdkmath.NewInt(100)},
|
|
},
|
|
c("debt", 12345678),
|
|
).WithID(2),
|
|
types.NewCollateralAuction(
|
|
"sellerMod",
|
|
c("hard", 12345678),
|
|
time.Date(1998, time.January, 1, 0, 0, 0, 0, time.UTC),
|
|
c("usdx", 12345678),
|
|
types.WeightedAddresses{
|
|
Addresses: addrs,
|
|
Weights: []sdkmath.Int{sdkmath.NewInt(100)},
|
|
},
|
|
c("debt", 12345678),
|
|
).WithID(3),
|
|
}
|
|
for _, a := range auctions {
|
|
auctionsKeeper.SetAuction(ctx, a)
|
|
}
|
|
|
|
qs := keeper.NewQueryServerImpl(auctionsKeeper)
|
|
|
|
tests := []struct {
|
|
giveName string
|
|
giveRequest types.QueryAuctionsRequest
|
|
wantResponse []types.Auction
|
|
}{
|
|
{
|
|
"empty request",
|
|
types.QueryAuctionsRequest{},
|
|
auctions,
|
|
},
|
|
{
|
|
"denom query swp",
|
|
types.QueryAuctionsRequest{
|
|
Denom: "swp",
|
|
},
|
|
auctions[0:1],
|
|
},
|
|
{
|
|
"denom query usdx all",
|
|
types.QueryAuctionsRequest{
|
|
Denom: "usdx",
|
|
},
|
|
auctions,
|
|
},
|
|
{
|
|
"owner",
|
|
types.QueryAuctionsRequest{
|
|
Owner: addrs[0].String(),
|
|
},
|
|
auctions[2:4],
|
|
},
|
|
{
|
|
"owner and denom",
|
|
types.QueryAuctionsRequest{
|
|
Owner: addrs[0].String(),
|
|
Denom: "hard",
|
|
},
|
|
auctions[3:4],
|
|
},
|
|
{
|
|
"owner, denom, type, phase",
|
|
types.QueryAuctionsRequest{
|
|
Owner: addrs[0].String(),
|
|
Denom: "hard",
|
|
Type: types.CollateralAuctionType,
|
|
Phase: types.ForwardAuctionPhase,
|
|
},
|
|
auctions[3:4],
|
|
},
|
|
}
|
|
|
|
for _, tc := range tests {
|
|
t.Run(tc.giveName, func(t *testing.T) {
|
|
res, err := qs.Auctions(sdk.WrapSDKContext(ctx), &tc.giveRequest)
|
|
require.NoError(t, err)
|
|
|
|
var unpackedAuctions []types.Auction
|
|
|
|
for _, anyAuction := range res.Auctions {
|
|
var auction types.Auction
|
|
err := tApp.AppCodec().UnpackAny(anyAuction, &auction)
|
|
require.NoError(t, err)
|
|
|
|
unpackedAuctions = append(unpackedAuctions, auction)
|
|
}
|
|
|
|
require.Equal(t, tc.wantResponse, unpackedAuctions)
|
|
})
|
|
}
|
|
}
|