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		alreadyRegistered := false
 35		for j := 0; j < len(testCoords); j++ {
 36			// coordinator already registered
 37			if coordinators[i].Bidder == testCoords[j].Bidder {
 38				alreadyRegistered = true
 39				if coordinators[i].EthBlockNum > testCoords[j].EthBlockNum {
 40					// This occurrence is more updated, delete current and append new
 41					testCoords = append(testCoords[:j], testCoords[j+1:]...)
 42					alreadyRegistered = false
 43				}
 44				break
 45			}
 46		}
 47		if !alreadyRegistered {
 48			testCoords = append(testCoords, historydb.CoordinatorAPI{
 49				Bidder:      coordinators[i].Bidder,
 50				Forger:      coordinators[i].Forger,
 51				EthBlockNum: coordinators[i].EthBlockNum,
 52				URL:         coordinators[i].URL,
 53			})
 54		}
 55	}
 56	return testCoords
 57}
 58
 59func TestGetCoordinators(t *testing.T) {
 60	endpoint := apiURL + "coordinators"
 61	fetchedCoordinators := []historydb.CoordinatorAPI{}
 62	appendIter := func(intr interface{}) {
 63		for i := 0; i < len(intr.(*testCoordinatorsResponse).Coordinators); i++ {
 64			tmp, err := copystructure.Copy(intr.(*testCoordinatorsResponse).Coordinators[i])
 65			if err != nil {
 66				panic(err)
 67			}
 68			fetchedCoordinators = append(fetchedCoordinators, tmp.(historydb.CoordinatorAPI))
 69		}
 70	}
 71
 72	// All
 73	limit := 5
 74	path := fmt.Sprintf("%s?limit=%d", endpoint, limit)
 75	err := doGoodReqPaginated(path, historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter)
 76	assert.NoError(t, err)
 77	assertCoordinators(t, tc.coordinators, fetchedCoordinators)
 78
 79	// All in reverse order
 80	fetchedCoordinators = []historydb.CoordinatorAPI{}
 81	err = doGoodReqPaginated(path, historydb.OrderDesc, &testCoordinatorsResponse{}, appendIter)
 82	assert.NoError(t, err)
 83	reversedCoordinators := []historydb.CoordinatorAPI{}
 84	for i := 0; i < len(tc.coordinators); i++ {
 85		reversedCoordinators = append(reversedCoordinators, tc.coordinators[len(tc.coordinators)-1-i])
 86	}
 87	assertCoordinators(t, reversedCoordinators, fetchedCoordinators)
 88	for _, filteredCoord := range tc.coordinators {
 89		// By bidder
 90		fetchedCoordinators = []historydb.CoordinatorAPI{}
 91		err = doGoodReqPaginated(
 92			fmt.Sprintf(path+"&bidderAddr=%s", filteredCoord.Bidder.String()),
 93			historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter,
 94		)
 95		assert.NoError(t, err)
 96		assertCoordinators(t, []historydb.CoordinatorAPI{filteredCoord}, fetchedCoordinators)
 97		// By forger
 98		fetchedCoordinators = []historydb.CoordinatorAPI{}
 99		err = doGoodReqPaginated(
100			fmt.Sprintf(path+"&forgerAddr=%s", filteredCoord.Forger.String()),
101			historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter,
102		)
103		assert.NoError(t, err)
104		assertCoordinators(t, []historydb.CoordinatorAPI{filteredCoord}, fetchedCoordinators)
105	}
106
107	// Empty array
108	fetchedCoordinators = []historydb.CoordinatorAPI{}
109	path = fmt.Sprintf("%s?bidderAddr=0xaa942cfcd25ad4d90a62358b0dd84f33b398262a", endpoint)
110	err = doGoodReqPaginated(path, historydb.OrderDesc, &testCoordinatorsResponse{}, appendIter)
111	assert.NoError(t, err)
112	assertCoordinators(t, []historydb.CoordinatorAPI{}, fetchedCoordinators)
113
114	// 400
115	path = fmt.Sprintf("%s?bidderAddr=0x001", endpoint)
116	err = doBadReq("GET", path, nil, 400)
117	assert.NoError(t, err)
118}
119
120func assertCoordinator(t *testing.T, expected, actual historydb.CoordinatorAPI) {
121	actual.ItemID = 0
122	assert.Equal(t, expected, actual)
123}
124
125func assertCoordinators(t *testing.T, expected, actual []historydb.CoordinatorAPI) {
126	assert.Equal(t, len(expected), len(actual))
127	for i := 0; i < len(expected); i++ {
128		assertCoordinator(t, expected[i], actual[i])
129	}
130}