1package api
  2
  3import (
  4	"fmt"
  5	"testing"
  6
  7	"github.com/hermeznetwork/hermez-node/common"
  8	"github.com/hermeznetwork/hermez-node/db/historydb"
  9	"github.com/mitchellh/copystructure"
 10	"github.com/stretchr/testify/assert"
 11)
 12
 13type testCoordinatorsResponse struct {
 14	Coordinators []historydb.CoordinatorAPI `json:"coordinators"`
 15	PendingItems uint64                     `json:"pendingItems"`
 16}
 17
 18func (t testCoordinatorsResponse) GetPending() (pendingItems, lastItemID uint64) {
 19	if len(t.Coordinators) == 0 {
 20		return 0, 0
 21	}
 22	pendingItems = t.PendingItems
 23	lastItemID = t.Coordinators[len(t.Coordinators)-1].ItemID
 24	return pendingItems, lastItemID
 25}
 26
 27func (t *testCoordinatorsResponse) Len() int { return len(t.Coordinators) }
 28
 29func (t testCoordinatorsResponse) New() Pendinger { return &testCoordinatorsResponse{} }
 30
 31func genTestCoordinators(coordinators []common.Coordinator) []historydb.CoordinatorAPI {
 32	testCoords := []historydb.CoordinatorAPI{}
 33	for i := 0; i < len(coordinators); i++ {
 34		testCoords = append(testCoords, historydb.CoordinatorAPI{
 35			Bidder:      coordinators[i].Bidder,
 36			Forger:      coordinators[i].Forger,
 37			EthBlockNum: coordinators[i].EthBlockNum,
 38			URL:         coordinators[i].URL,
 39		})
 40	}
 41	return testCoords
 42}
 43
 44func TestGetCoordinators(t *testing.T) {
 45	endpoint := apiURL + "coordinators"
 46	fetchedCoordinators := []historydb.CoordinatorAPI{}
 47	appendIter := func(intr interface{}) {
 48		for i := 0; i < len(intr.(*testCoordinatorsResponse).Coordinators); i++ {
 49			tmp, err := copystructure.Copy(intr.(*testCoordinatorsResponse).Coordinators[i])
 50			if err != nil {
 51				panic(err)
 52			}
 53			fetchedCoordinators = append(fetchedCoordinators, tmp.(historydb.CoordinatorAPI))
 54		}
 55	}
 56
 57	// All
 58	limit := 5
 59	path := fmt.Sprintf("%s?limit=%d", endpoint, limit)
 60	err := doGoodReqPaginated(path, historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter)
 61	assert.NoError(t, err)
 62	assertCoordinators(t, tc.coordinators, fetchedCoordinators)
 63
 64	// All in reverse order
 65	fetchedCoordinators = []historydb.CoordinatorAPI{}
 66	err = doGoodReqPaginated(path, historydb.OrderDesc, &testCoordinatorsResponse{}, appendIter)
 67	assert.NoError(t, err)
 68	reversedCoordinators := []historydb.CoordinatorAPI{}
 69	for i := 0; i < len(tc.coordinators); i++ {
 70		reversedCoordinators = append(reversedCoordinators, tc.coordinators[len(tc.coordinators)-1-i])
 71	}
 72	assertCoordinators(t, reversedCoordinators, fetchedCoordinators)
 73	for _, filteredCoord := range tc.coordinators {
 74		// By bidder
 75		fetchedCoordinators = []historydb.CoordinatorAPI{}
 76		err = doGoodReqPaginated(
 77			fmt.Sprintf(path+"&bidderAddr=%s", filteredCoord.Bidder.String()),
 78			historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter,
 79		)
 80		assert.NoError(t, err)
 81		assertCoordinators(t, []historydb.CoordinatorAPI{filteredCoord}, fetchedCoordinators)
 82		// By forger
 83		fetchedCoordinators = []historydb.CoordinatorAPI{}
 84		err = doGoodReqPaginated(
 85			fmt.Sprintf(path+"&forgerAddr=%s", filteredCoord.Forger.String()),
 86			historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter,
 87		)
 88		assert.NoError(t, err)
 89		assertCoordinators(t, []historydb.CoordinatorAPI{filteredCoord}, fetchedCoordinators)
 90	}
 91
 92	// Empty array
 93	fetchedCoordinators = []historydb.CoordinatorAPI{}
 94	path = fmt.Sprintf("%s?bidderAddr=0xaa942cfcd25ad4d90a62358b0dd84f33b398262a", endpoint)
 95	err = doGoodReqPaginated(path, historydb.OrderDesc, &testCoordinatorsResponse{}, appendIter)
 96	assert.NoError(t, err)
 97	assertCoordinators(t, []historydb.CoordinatorAPI{}, fetchedCoordinators)
 98
 99	// 400
100	path = fmt.Sprintf("%s?bidderAddr=0x001", endpoint)
101	err = doBadReq("GET", path, nil, 400)
102	assert.NoError(t, err)
103}
104
105func assertCoordinator(t *testing.T, expected, actual historydb.CoordinatorAPI) {
106	actual.ItemID = 0
107	assert.Equal(t, expected, actual)
108}
109
110func assertCoordinators(t *testing.T, expected, actual []historydb.CoordinatorAPI) {
111	assert.Equal(t, len(expected), len(actual))
112	for i := 0; i < len(expected); i++ {
113		assertCoordinator(t, expected[i], actual[i])
114	}
115}