laisolizq
4 years ago
9 changed files with 303 additions and 302 deletions
Unified View
Diff Options
-
+5 -126api/api_test.go
-
+48 -96api/dbtoapistructs.go
-
+0 -49api/handlers.go
-
+66 -0api/token.go
-
+153 -0api/token_test.go
-
+8 -8db/historydb/historydb.go
-
+14 -14db/historydb/views.go
-
+6 -6db/l2db/l2db_test.go
-
+3 -3synchronizer/synchronizer_test.go
@ -0,0 +1,66 @@ |
|||||
|
package api |
||||
|
|
||||
|
import ( |
||||
|
"errors" |
||||
|
"net/http" |
||||
|
|
||||
|
"github.com/gin-gonic/gin" |
||||
|
"github.com/hermeznetwork/hermez-node/common" |
||||
|
"github.com/hermeznetwork/hermez-node/db" |
||||
|
"github.com/hermeznetwork/hermez-node/db/historydb" |
||||
|
) |
||||
|
|
||||
|
func getToken(c *gin.Context) { |
||||
|
// Get TokenID
|
||||
|
tokenIDUint, err := parseParamUint("id", nil, 0, maxUint32, c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
if tokenIDUint == nil { // tokenID is required
|
||||
|
retBadReq(errors.New("Invalid tokenID"), c) |
||||
|
return |
||||
|
} |
||||
|
tokenID := common.TokenID(*tokenIDUint) |
||||
|
// Fetch token from historyDB
|
||||
|
token, err := h.GetToken(tokenID) |
||||
|
if err != nil { |
||||
|
retSQLErr(err, c) |
||||
|
return |
||||
|
} |
||||
|
c.JSON(http.StatusOK, token) |
||||
|
} |
||||
|
|
||||
|
func getTokens(c *gin.Context) { |
||||
|
// Account filters
|
||||
|
tokenIDs, symbols, name, err := parseTokenFilters(c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// Pagination
|
||||
|
fromItem, order, limit, err := parsePagination(c) |
||||
|
if err != nil { |
||||
|
retBadReq(err, c) |
||||
|
return |
||||
|
} |
||||
|
// Fetch exits from historyDB
|
||||
|
tokens, pagination, err := h.GetTokens( |
||||
|
tokenIDs, symbols, name, fromItem, limit, order, |
||||
|
) |
||||
|
if err != nil { |
||||
|
retSQLErr(err, c) |
||||
|
return |
||||
|
} |
||||
|
|
||||
|
// Build succesfull response
|
||||
|
type tokensResponse struct { |
||||
|
Tokens []historydb.TokenWithUSD `json:"tokens"` |
||||
|
Pagination *db.Pagination `json:"pagination"` |
||||
|
} |
||||
|
c.JSON(http.StatusOK, &tokensResponse{ |
||||
|
Tokens: tokens, |
||||
|
Pagination: pagination, |
||||
|
}) |
||||
|
} |
||||
@ -0,0 +1,153 @@ |
|||||
|
package api |
||||
|
|
||||
|
import ( |
||||
|
"fmt" |
||||
|
"strconv" |
||||
|
"testing" |
||||
|
|
||||
|
"github.com/hermeznetwork/hermez-node/db" |
||||
|
"github.com/hermeznetwork/hermez-node/db/historydb" |
||||
|
"github.com/mitchellh/copystructure" |
||||
|
"github.com/stretchr/testify/assert" |
||||
|
"github.com/stretchr/testify/require" |
||||
|
) |
||||
|
|
||||
|
type testTokensResponse struct { |
||||
|
Tokens []historydb.TokenWithUSD `json:"tokens"` |
||||
|
Pagination *db.Pagination `json:"pagination"` |
||||
|
} |
||||
|
|
||||
|
func (t *testTokensResponse) GetPagination() *db.Pagination { |
||||
|
if t.Tokens[0].ItemID < t.Tokens[len(t.Tokens)-1].ItemID { |
||||
|
t.Pagination.FirstReturnedItem = t.Tokens[0].ItemID |
||||
|
t.Pagination.LastReturnedItem = t.Tokens[len(t.Tokens)-1].ItemID |
||||
|
} else { |
||||
|
t.Pagination.LastReturnedItem = t.Tokens[0].ItemID |
||||
|
t.Pagination.FirstReturnedItem = t.Tokens[len(t.Tokens)-1].ItemID |
||||
|
} |
||||
|
return t.Pagination |
||||
|
} |
||||
|
|
||||
|
func (t *testTokensResponse) Len() int { |
||||
|
return len(t.Tokens) |
||||
|
} |
||||
|
|
||||
|
func TestGetToken(t *testing.T) { |
||||
|
// Get all txs by their ID
|
||||
|
endpoint := apiURL + "tokens/" |
||||
|
fetchedTokens := []historydb.TokenWithUSD{} |
||||
|
for _, token := range tc.tokens { |
||||
|
fetchedToken := historydb.TokenWithUSD{} |
||||
|
assert.NoError(t, doGoodReq("GET", endpoint+strconv.Itoa(int(token.TokenID)), nil, &fetchedToken)) |
||||
|
fetchedTokens = append(fetchedTokens, fetchedToken) |
||||
|
} |
||||
|
assertTokensAPIs(t, tc.tokens, fetchedTokens) |
||||
|
} |
||||
|
|
||||
|
func TestGetTokens(t *testing.T) { |
||||
|
endpoint := apiURL + "tokens" |
||||
|
fetchedTokens := []historydb.TokenWithUSD{} |
||||
|
appendIter := func(intr interface{}) { |
||||
|
for i := 0; i < len(intr.(*testTokensResponse).Tokens); i++ { |
||||
|
tmp, err := copystructure.Copy(intr.(*testTokensResponse).Tokens[i]) |
||||
|
if err != nil { |
||||
|
panic(err) |
||||
|
} |
||||
|
fetchedTokens = append(fetchedTokens, tmp.(historydb.TokenWithUSD)) |
||||
|
} |
||||
|
} |
||||
|
// Get all (no filters)
|
||||
|
limit := 8 |
||||
|
path := fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit) |
||||
|
err := doGoodReqPaginated(path, historydb.OrderAsc, &testTokensResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
assertTokensAPIs(t, tc.tokens, fetchedTokens) |
||||
|
|
||||
|
// Get by tokenIds
|
||||
|
fetchedTokens = []historydb.TokenWithUSD{} |
||||
|
limit = 7 |
||||
|
stringIds := strconv.Itoa(int(tc.tokens[2].TokenID)) + "," + strconv.Itoa(int(tc.tokens[5].TokenID)) + "," + strconv.Itoa(int(tc.tokens[6].TokenID)) |
||||
|
path = fmt.Sprintf( |
||||
|
"%s?ids=%s&limit=%d&fromItem=", |
||||
|
endpoint, stringIds, limit, |
||||
|
) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testTokensResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
var tokensFiltered []historydb.TokenWithUSD |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[2]) |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[5]) |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[6]) |
||||
|
assertTokensAPIs(t, tokensFiltered, fetchedTokens) |
||||
|
|
||||
|
// Get by symbols
|
||||
|
fetchedTokens = []historydb.TokenWithUSD{} |
||||
|
limit = 7 |
||||
|
stringSymbols := tc.tokens[1].Symbol + "," + tc.tokens[3].Symbol |
||||
|
path = fmt.Sprintf( |
||||
|
"%s?symbols=%s&limit=%d&fromItem=", |
||||
|
endpoint, stringSymbols, limit, |
||||
|
) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testTokensResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
tokensFiltered = nil |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[1]) |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[3]) |
||||
|
assertTokensAPIs(t, tokensFiltered, fetchedTokens) |
||||
|
|
||||
|
// Get by name
|
||||
|
fetchedTokens = []historydb.TokenWithUSD{} |
||||
|
limit = 5 |
||||
|
stringName := tc.tokens[8].Name[4:5] |
||||
|
path = fmt.Sprintf( |
||||
|
"%s?name=%s&limit=%d&fromItem=", |
||||
|
endpoint, stringName, limit, |
||||
|
) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testTokensResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
tokensFiltered = nil |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[8]) |
||||
|
assertTokensAPIs(t, tokensFiltered, fetchedTokens) |
||||
|
|
||||
|
// Multiple filters
|
||||
|
fetchedTokens = []historydb.TokenWithUSD{} |
||||
|
limit = 5 |
||||
|
stringSymbols = tc.tokens[2].Symbol + "," + tc.tokens[6].Symbol |
||||
|
stringIds = strconv.Itoa(int(tc.tokens[2].TokenID)) + "," + strconv.Itoa(int(tc.tokens[5].TokenID)) + "," + strconv.Itoa(int(tc.tokens[6].TokenID)) |
||||
|
path = fmt.Sprintf( |
||||
|
"%s?symbols=%s&ids=%s&limit=%d&fromItem=", |
||||
|
endpoint, stringSymbols, stringIds, limit, |
||||
|
) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderAsc, &testTokensResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
|
||||
|
tokensFiltered = nil |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[2]) |
||||
|
tokensFiltered = append(tokensFiltered, tc.tokens[6]) |
||||
|
assertTokensAPIs(t, tokensFiltered, fetchedTokens) |
||||
|
|
||||
|
// All, in reverse order
|
||||
|
fetchedTokens = []historydb.TokenWithUSD{} |
||||
|
limit = 5 |
||||
|
path = fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit) |
||||
|
err = doGoodReqPaginated(path, historydb.OrderDesc, &testTokensResponse{}, appendIter) |
||||
|
assert.NoError(t, err) |
||||
|
flipedTokens := []historydb.TokenWithUSD{} |
||||
|
for i := 0; i < len(tc.tokens); i++ { |
||||
|
flipedTokens = append(flipedTokens, tc.tokens[len(tc.tokens)-1-i]) |
||||
|
} |
||||
|
assertTokensAPIs(t, flipedTokens, fetchedTokens) |
||||
|
} |
||||
|
|
||||
|
func assertTokensAPIs(t *testing.T, expected, actual []historydb.TokenWithUSD) { |
||||
|
require.Equal(t, len(expected), len(actual)) |
||||
|
for i := 0; i < len(actual); i++ { //nolint len(actual) won't change within the loop
|
||||
|
actual[i].ItemID = 0 |
||||
|
if expected[i].USDUpdate == nil { |
||||
|
assert.Equal(t, expected[i].USDUpdate, actual[i].USDUpdate) |
||||
|
} else { |
||||
|
assert.Equal(t, expected[i].USDUpdate.Unix(), actual[i].USDUpdate.Unix()) |
||||
|
expected[i].USDUpdate = actual[i].USDUpdate |
||||
|
} |
||||
|
assert.Equal(t, expected[i], actual[i]) |
||||
|
} |
||||
|
} |
||||