You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 

165 lines
5.2 KiB

package api
import (
"fmt"
"testing"
"time"
ethCommon "github.com/ethereum/go-ethereum/common"
"github.com/hermeznetwork/hermez-node/common"
"github.com/hermeznetwork/hermez-node/db"
"github.com/hermeznetwork/hermez-node/db/historydb"
"github.com/mitchellh/copystructure"
"github.com/stretchr/testify/assert"
)
type testBid struct {
ItemID int `json:"itemId"`
SlotNum int64 `json:"slotNum"`
BidValue string `json:"bidValue"`
EthBlockNum int64 `json:"ethereumBlockNum"`
Bidder ethCommon.Address `json:"bidderAddr"`
Forger ethCommon.Address `json:"forgerAddr"`
URL string `json:"URL"`
Timestamp time.Time `json:"timestamp"`
}
type testBidsResponse struct {
Bids []testBid `json:"bids"`
Pagination *db.Pagination `json:"pagination"`
}
func (t testBidsResponse) GetPagination() *db.Pagination {
if t.Bids[0].ItemID < t.Bids[len(t.Bids)-1].ItemID {
t.Pagination.FirstReturnedItem = t.Bids[0].ItemID
t.Pagination.LastReturnedItem = t.Bids[len(t.Bids)-1].ItemID
} else {
t.Pagination.LastReturnedItem = t.Bids[0].ItemID
t.Pagination.FirstReturnedItem = t.Bids[len(t.Bids)-1].ItemID
}
return t.Pagination
}
func (t testBidsResponse) Len() int {
return len(t.Bids)
}
func genTestBids(blocks []common.Block, coordinators []historydb.CoordinatorAPI, bids []common.Bid) []testBid {
tBids := []testBid{}
for _, bid := range bids {
block := getBlockByNum(bid.EthBlockNum, blocks)
coordinator := getCoordinatorByBidder(bid.Bidder, coordinators)
tBid := testBid{
SlotNum: bid.SlotNum,
BidValue: bid.BidValue.String(),
EthBlockNum: bid.EthBlockNum,
Bidder: bid.Bidder,
Forger: coordinator.Forger,
URL: coordinator.URL,
Timestamp: block.Timestamp,
}
tBids = append(tBids, tBid)
}
return tBids
}
func TestGetBids(t *testing.T) {
endpoint := apiURL + "bids"
fetchedBids := []testBid{}
appendIter := func(intr interface{}) {
for i := 0; i < len(intr.(*testBidsResponse).Bids); i++ {
tmp, err := copystructure.Copy(intr.(*testBidsResponse).Bids[i])
if err != nil {
panic(err)
}
fetchedBids = append(fetchedBids, tmp.(testBid))
}
}
limit := 3
// bidderAddress
fetchedBids = []testBid{}
bidderAddress := tc.bids[3].Bidder
path := fmt.Sprintf("%s?bidderAddr=%s&limit=%d&fromItem=", endpoint, bidderAddress.String(), limit)
err := doGoodReqPaginated(path, historydb.OrderAsc, &testBidsResponse{}, appendIter)
assert.NoError(t, err)
bidderAddrBids := []testBid{}
for i := 0; i < len(tc.bids); i++ {
if tc.bids[i].Bidder == bidderAddress {
bidderAddrBids = append(bidderAddrBids, tc.bids[i])
}
}
assertBids(t, bidderAddrBids, fetchedBids)
// slotNum
fetchedBids = []testBid{}
slotNum := tc.bids[3].SlotNum
path = fmt.Sprintf("%s?slotNum=%d&limit=%d&fromItem=", endpoint, slotNum, limit)
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBidsResponse{}, appendIter)
assert.NoError(t, err)
slotNumBids := []testBid{}
for i := 0; i < len(tc.bids); i++ {
if tc.bids[i].SlotNum == slotNum {
slotNumBids = append(slotNumBids, tc.bids[i])
}
}
assertBids(t, slotNumBids, fetchedBids)
// slotNum, in reverse order
fetchedBids = []testBid{}
path = fmt.Sprintf("%s?slotNum=%d&limit=%d&fromItem=", endpoint, slotNum, limit)
err = doGoodReqPaginated(path, historydb.OrderDesc, &testBidsResponse{}, appendIter)
assert.NoError(t, err)
flippedBids := []testBid{}
for i := len(slotNumBids) - 1; i >= 0; i-- {
flippedBids = append(flippedBids, slotNumBids[i])
}
assertBids(t, flippedBids, fetchedBids)
// Mixed filters
fetchedBids = []testBid{}
bidderAddress = tc.bids[9].Bidder
slotNum = tc.bids[4].SlotNum
path = fmt.Sprintf("%s?bidderAddr=%s&slotNum=%d&limit=%d&fromItem=", endpoint, bidderAddress.String(), slotNum, limit)
err = doGoodReqPaginated(path, historydb.OrderAsc, &testBidsResponse{}, appendIter)
assert.NoError(t, err)
slotNumBidderAddrBids := []testBid{}
for i := 0; i < len(tc.bids); i++ {
if tc.bids[i].Bidder == bidderAddress && tc.bids[i].SlotNum == slotNum {
slotNumBidderAddrBids = append(slotNumBidderAddrBids, tc.bids[i])
}
}
assertBids(t, slotNumBidderAddrBids, fetchedBids)
// 400
// No filters
path = fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit)
err = doBadReq("GET", path, nil, 400)
assert.NoError(t, err)
// Invalid slotNum
path = fmt.Sprintf("%s?slotNum=%d", endpoint, -2)
err = doBadReq("GET", path, nil, 400)
assert.NoError(t, err)
// Invalid bidderAddress
path = fmt.Sprintf("%s?bidderAddr=%s", endpoint, "0xG0000001")
err = doBadReq("GET", path, nil, 400)
assert.NoError(t, err)
// 404
path = fmt.Sprintf("%s?slotNum=%d&bidderAddr=%s", endpoint, tc.bids[0].SlotNum, tc.bids[1].Bidder.String())
err = doBadReq("GET", path, nil, 404)
assert.NoError(t, err)
}
func assertBids(t *testing.T, expected, actual []testBid) {
assert.Equal(t, len(expected), len(actual))
for i := 0; i < len(expected); i++ {
assertBid(t, expected[i], actual[i])
}
}
func assertBid(t *testing.T, expected, actual testBid) {
assert.Equal(t, expected.Timestamp.Unix(), actual.Timestamp.Unix())
expected.Timestamp = actual.Timestamp
actual.ItemID = expected.ItemID
assert.Equal(t, expected, actual)
}