mirror of
https://github.com/0glabs/0g-chain.git
synced 2024-12-29 01:35:18 +00:00
89 lines
2.6 KiB
Go
89 lines
2.6 KiB
Go
|
package keeper
|
||
|
|
||
|
import (
|
||
|
"testing"
|
||
|
|
||
|
sdk "github.com/cosmos/cosmos-sdk/types"
|
||
|
"github.com/kava-labs/kava/x/auction/types"
|
||
|
"github.com/stretchr/testify/require"
|
||
|
abci "github.com/tendermint/tendermint/abci/types"
|
||
|
)
|
||
|
|
||
|
func TestKeeper_SetGetDeleteAuction(t *testing.T) {
|
||
|
// setup keeper, create auction
|
||
|
mapp, keeper, addresses, _ := setUpMockApp()
|
||
|
header := abci.Header{Height: mapp.LastBlockHeight() + 1}
|
||
|
mapp.BeginBlock(abci.RequestBeginBlock{Header: header}) // Without this it panics about "invalid memory address or nil pointer dereference"
|
||
|
ctx := mapp.BaseApp.NewContext(false, header)
|
||
|
auction, _ := types.NewForwardAuction(addresses[0], sdk.NewInt64Coin("usdx", 100), sdk.NewInt64Coin("kava", 0), types.EndTime(1000))
|
||
|
id := types.ID(5)
|
||
|
auction.SetID(id)
|
||
|
|
||
|
// write and read from store
|
||
|
keeper.SetAuction(ctx, &auction)
|
||
|
readAuction, found := keeper.GetAuction(ctx, id)
|
||
|
|
||
|
// check before and after match
|
||
|
require.True(t, found)
|
||
|
require.Equal(t, &auction, readAuction)
|
||
|
t.Log(auction)
|
||
|
t.Log(readAuction.GetID())
|
||
|
// check auction is in queue
|
||
|
iter := keeper.GetQueueIterator(ctx, 100000)
|
||
|
require.Equal(t, 1, len(convertIteratorToSlice(keeper, iter)))
|
||
|
iter.Close()
|
||
|
|
||
|
// delete auction
|
||
|
keeper.deleteAuction(ctx, id)
|
||
|
|
||
|
// check auction does not exist
|
||
|
_, found = keeper.GetAuction(ctx, id)
|
||
|
require.False(t, found)
|
||
|
// check auction not in queue
|
||
|
iter = keeper.GetQueueIterator(ctx, 100000)
|
||
|
require.Equal(t, 0, len(convertIteratorToSlice(keeper, iter)))
|
||
|
iter.Close()
|
||
|
|
||
|
}
|
||
|
|
||
|
// TODO convert to table driven test with more test cases
|
||
|
func TestKeeper_ExpiredAuctionQueue(t *testing.T) {
|
||
|
// setup keeper
|
||
|
mapp, keeper, _, _ := setUpMockApp()
|
||
|
header := abci.Header{Height: mapp.LastBlockHeight() + 1}
|
||
|
mapp.BeginBlock(abci.RequestBeginBlock{Header: header})
|
||
|
ctx := mapp.BaseApp.NewContext(false, header)
|
||
|
// create an example queue
|
||
|
type queue []struct {
|
||
|
endTime types.EndTime
|
||
|
auctionID types.ID
|
||
|
}
|
||
|
q := queue{{1000, 0}, {1300, 2}, {5200, 1}}
|
||
|
|
||
|
// write and read queue
|
||
|
for _, v := range q {
|
||
|
keeper.insertIntoQueue(ctx, v.endTime, v.auctionID)
|
||
|
}
|
||
|
iter := keeper.GetQueueIterator(ctx, 1000)
|
||
|
|
||
|
// check before and after match
|
||
|
i := 0
|
||
|
for ; iter.Valid(); iter.Next() {
|
||
|
var auctionID types.ID
|
||
|
keeper.cdc.MustUnmarshalBinaryLengthPrefixed(iter.Value(), &auctionID)
|
||
|
require.Equal(t, q[i].auctionID, auctionID)
|
||
|
i++
|
||
|
}
|
||
|
|
||
|
}
|
||
|
|
||
|
func convertIteratorToSlice(keeper Keeper, iterator sdk.Iterator) []types.ID {
|
||
|
var queue []types.ID
|
||
|
for ; iterator.Valid(); iterator.Next() {
|
||
|
var auctionID types.ID
|
||
|
keeper.cdc.MustUnmarshalBinaryLengthPrefixed(iterator.Value(), &auctionID)
|
||
|
queue = append(queue, auctionID)
|
||
|
}
|
||
|
return queue
|
||
|
}
|