package api import ( "fmt" "testing" "github.com/hermeznetwork/hermez-node/common" "github.com/hermeznetwork/hermez-node/db/historydb" "github.com/mitchellh/copystructure" "github.com/stretchr/testify/assert" ) type testCoordinatorsResponse struct { Coordinators []historydb.CoordinatorAPI `json:"coordinators"` PendingItems uint64 `json:"pendingItems"` } func (t testCoordinatorsResponse) GetPending() (pendingItems, lastItemID uint64) { if len(t.Coordinators) == 0 { return 0, 0 } pendingItems = t.PendingItems lastItemID = t.Coordinators[len(t.Coordinators)-1].ItemID return pendingItems, lastItemID } func (t *testCoordinatorsResponse) Len() int { return len(t.Coordinators) } func (t testCoordinatorsResponse) New() Pendinger { return &testCoordinatorsResponse{} } func genTestCoordinators(coordinators []common.Coordinator) []historydb.CoordinatorAPI { testCoords := []historydb.CoordinatorAPI{} for i := 0; i < len(coordinators); i++ { testCoords = append(testCoords, historydb.CoordinatorAPI{ Bidder: coordinators[i].Bidder, Forger: coordinators[i].Forger, EthBlockNum: coordinators[i].EthBlockNum, URL: coordinators[i].URL, }) } return testCoords } func TestGetCoordinators(t *testing.T) { endpoint := apiURL + "coordinators" fetchedCoordinators := []historydb.CoordinatorAPI{} appendIter := func(intr interface{}) { for i := 0; i < len(intr.(*testCoordinatorsResponse).Coordinators); i++ { tmp, err := copystructure.Copy(intr.(*testCoordinatorsResponse).Coordinators[i]) if err != nil { panic(err) } fetchedCoordinators = append(fetchedCoordinators, tmp.(historydb.CoordinatorAPI)) } } // All limit := 5 path := fmt.Sprintf("%s?limit=%d", endpoint, limit) err := doGoodReqPaginated(path, historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter) assert.NoError(t, err) assertCoordinators(t, tc.coordinators, fetchedCoordinators) // All in reverse order fetchedCoordinators = []historydb.CoordinatorAPI{} err = doGoodReqPaginated(path, historydb.OrderDesc, &testCoordinatorsResponse{}, appendIter) assert.NoError(t, err) reversedCoordinators := []historydb.CoordinatorAPI{} for i := 0; i < len(tc.coordinators); i++ { reversedCoordinators = append(reversedCoordinators, tc.coordinators[len(tc.coordinators)-1-i]) } assertCoordinators(t, reversedCoordinators, fetchedCoordinators) for _, filteredCoord := range tc.coordinators { // By bidder fetchedCoordinators = []historydb.CoordinatorAPI{} err = doGoodReqPaginated( fmt.Sprintf(path+"&bidderAddr=%s", filteredCoord.Bidder.String()), historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter, ) assert.NoError(t, err) assertCoordinators(t, []historydb.CoordinatorAPI{filteredCoord}, fetchedCoordinators) // By forger fetchedCoordinators = []historydb.CoordinatorAPI{} err = doGoodReqPaginated( fmt.Sprintf(path+"&forgerAddr=%s", filteredCoord.Forger.String()), historydb.OrderAsc, &testCoordinatorsResponse{}, appendIter, ) assert.NoError(t, err) assertCoordinators(t, []historydb.CoordinatorAPI{filteredCoord}, fetchedCoordinators) } // Empty array fetchedCoordinators = []historydb.CoordinatorAPI{} path = fmt.Sprintf("%s?bidderAddr=0xaa942cfcd25ad4d90a62358b0dd84f33b398262a", endpoint) err = doGoodReqPaginated(path, historydb.OrderDesc, &testCoordinatorsResponse{}, appendIter) assert.NoError(t, err) assertCoordinators(t, []historydb.CoordinatorAPI{}, fetchedCoordinators) // 400 path = fmt.Sprintf("%s?bidderAddr=0x001", endpoint) err = doBadReq("GET", path, nil, 400) assert.NoError(t, err) } func assertCoordinator(t *testing.T, expected, actual historydb.CoordinatorAPI) { actual.ItemID = 0 assert.Equal(t, expected, actual) } func assertCoordinators(t *testing.T, expected, actual []historydb.CoordinatorAPI) { assert.Equal(t, len(expected), len(actual)) for i := 0; i < len(expected); i++ { assertCoordinator(t, expected[i], actual[i]) } }