mirror of
https://github.com/arnaucube/hermez-node.git
synced 2026-02-07 11:26:44 +01:00
Refactor api exits
This commit is contained in:
Arnau B
276
api/exits_test.go
Normal file
276
api/exits_test.go
Normal file
@@ -0,0 +1,276 @@
|
||||
package api
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"testing"
|
||||
|
||||
"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"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
type testCVP struct {
|
||||
Root string
|
||||
Siblings []string
|
||||
OldKey string
|
||||
OldValue string
|
||||
IsOld0 bool
|
||||
Key string
|
||||
Value string
|
||||
Fnc int
|
||||
}
|
||||
|
||||
type testExit struct {
|
||||
ItemID int `json:"itemId"`
|
||||
BatchNum common.BatchNum `json:"batchNum"`
|
||||
AccountIdx string `json:"accountIndex"`
|
||||
MerkleProof testCVP `json:"merkleProof"`
|
||||
Balance string `json:"balance"`
|
||||
InstantWithdrawn *int64 `json:"instantWithdrawn"`
|
||||
DelayedWithdrawRequest *int64 `json:"delayedWithdrawRequest"`
|
||||
DelayedWithdrawn *int64 `json:"delayedWithdrawn"`
|
||||
Token historydb.TokenWithUSD `json:"token"`
|
||||
}
|
||||
|
||||
type testExitsResponse struct {
|
||||
Exits []testExit `json:"exits"`
|
||||
Pagination *db.Pagination `json:"pagination"`
|
||||
}
|
||||
|
||||
func (t *testExitsResponse) GetPagination() *db.Pagination {
|
||||
if t.Exits[0].ItemID < t.Exits[len(t.Exits)-1].ItemID {
|
||||
t.Pagination.FirstReturnedItem = t.Exits[0].ItemID
|
||||
t.Pagination.LastReturnedItem = t.Exits[len(t.Exits)-1].ItemID
|
||||
} else {
|
||||
t.Pagination.LastReturnedItem = t.Exits[0].ItemID
|
||||
t.Pagination.FirstReturnedItem = t.Exits[len(t.Exits)-1].ItemID
|
||||
}
|
||||
return t.Pagination
|
||||
}
|
||||
|
||||
func (t *testExitsResponse) Len() int {
|
||||
return len(t.Exits)
|
||||
}
|
||||
|
||||
func genTestExits(
|
||||
commonExits []common.ExitInfo,
|
||||
tokens []historydb.TokenWithUSD,
|
||||
accs []common.Account,
|
||||
usrIdxs []string,
|
||||
) (usrExits, allExits []testExit) {
|
||||
allExits = []testExit{}
|
||||
for _, exit := range commonExits {
|
||||
token := getTokenByIdx(exit.AccountIdx, tokens, accs)
|
||||
siblings := []string{}
|
||||
for i := 0; i < len(exit.MerkleProof.Siblings); i++ {
|
||||
siblings = append(siblings, exit.MerkleProof.Siblings[i].String())
|
||||
}
|
||||
allExits = append(allExits, testExit{
|
||||
BatchNum: exit.BatchNum,
|
||||
AccountIdx: idxToHez(exit.AccountIdx, token.Symbol),
|
||||
MerkleProof: testCVP{
|
||||
Root: exit.MerkleProof.Root.String(),
|
||||
Siblings: siblings,
|
||||
OldKey: exit.MerkleProof.OldKey.String(),
|
||||
OldValue: exit.MerkleProof.OldValue.String(),
|
||||
IsOld0: exit.MerkleProof.IsOld0,
|
||||
Key: exit.MerkleProof.Key.String(),
|
||||
Value: exit.MerkleProof.Value.String(),
|
||||
Fnc: exit.MerkleProof.Fnc,
|
||||
},
|
||||
Balance: exit.Balance.String(),
|
||||
InstantWithdrawn: exit.InstantWithdrawn,
|
||||
DelayedWithdrawRequest: exit.DelayedWithdrawRequest,
|
||||
DelayedWithdrawn: exit.DelayedWithdrawn,
|
||||
Token: token,
|
||||
})
|
||||
}
|
||||
usrExits = []testExit{}
|
||||
for _, exit := range allExits {
|
||||
for _, idx := range usrIdxs {
|
||||
if idx == exit.AccountIdx {
|
||||
usrExits = append(usrExits, exit)
|
||||
break
|
||||
}
|
||||
}
|
||||
}
|
||||
return usrExits, allExits
|
||||
}
|
||||
|
||||
func TestGetExits(t *testing.T) {
|
||||
endpoint := apiURL + "exits"
|
||||
fetchedExits := []testExit{}
|
||||
appendIter := func(intr interface{}) {
|
||||
for i := 0; i < len(intr.(*testExitsResponse).Exits); i++ {
|
||||
tmp, err := copystructure.Copy(intr.(*testExitsResponse).Exits[i])
|
||||
if err != nil {
|
||||
panic(err)
|
||||
}
|
||||
fetchedExits = append(fetchedExits, tmp.(testExit))
|
||||
}
|
||||
}
|
||||
// Get all (no filters)
|
||||
limit := 8
|
||||
path := fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit)
|
||||
err := doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
assertExitAPIs(t, tc.exits, fetchedExits)
|
||||
|
||||
// Get by ethAddr
|
||||
fetchedExits = []testExit{}
|
||||
limit = 7
|
||||
path = fmt.Sprintf(
|
||||
"%s?hermezEthereumAddress=%s&limit=%d&fromItem=",
|
||||
endpoint, tc.usrAddr, limit,
|
||||
)
|
||||
err = doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
assertExitAPIs(t, tc.usrExits, fetchedExits)
|
||||
// Get by bjj
|
||||
fetchedExits = []testExit{}
|
||||
limit = 6
|
||||
path = fmt.Sprintf(
|
||||
"%s?BJJ=%s&limit=%d&fromItem=",
|
||||
endpoint, tc.usrBjj, limit,
|
||||
)
|
||||
err = doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
assertExitAPIs(t, tc.usrExits, fetchedExits)
|
||||
// Get by tokenID
|
||||
fetchedExits = []testExit{}
|
||||
limit = 5
|
||||
tokenID := tc.exits[0].Token.TokenID
|
||||
path = fmt.Sprintf(
|
||||
"%s?tokenId=%d&limit=%d&fromItem=",
|
||||
endpoint, tokenID, limit,
|
||||
)
|
||||
err = doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
tokenIDExits := []testExit{}
|
||||
for i := 0; i < len(tc.exits); i++ {
|
||||
if tc.exits[i].Token.TokenID == tokenID {
|
||||
tokenIDExits = append(tokenIDExits, tc.exits[i])
|
||||
}
|
||||
}
|
||||
assertExitAPIs(t, tokenIDExits, fetchedExits)
|
||||
// idx
|
||||
fetchedExits = []testExit{}
|
||||
limit = 4
|
||||
idx := tc.exits[0].AccountIdx
|
||||
path = fmt.Sprintf(
|
||||
"%s?accountIndex=%s&limit=%d&fromItem=",
|
||||
endpoint, idx, limit,
|
||||
)
|
||||
err = doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
idxExits := []testExit{}
|
||||
for i := 0; i < len(tc.exits); i++ {
|
||||
if tc.exits[i].AccountIdx[6:] == idx[6:] {
|
||||
idxExits = append(idxExits, tc.exits[i])
|
||||
}
|
||||
}
|
||||
assertExitAPIs(t, idxExits, fetchedExits)
|
||||
// batchNum
|
||||
fetchedExits = []testExit{}
|
||||
limit = 3
|
||||
batchNum := tc.exits[0].BatchNum
|
||||
path = fmt.Sprintf(
|
||||
"%s?batchNum=%d&limit=%d&fromItem=",
|
||||
endpoint, batchNum, limit,
|
||||
)
|
||||
err = doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
batchNumExits := []testExit{}
|
||||
for i := 0; i < len(tc.exits); i++ {
|
||||
if tc.exits[i].BatchNum == batchNum {
|
||||
batchNumExits = append(batchNumExits, tc.exits[i])
|
||||
}
|
||||
}
|
||||
assertExitAPIs(t, batchNumExits, fetchedExits)
|
||||
// Multiple filters
|
||||
fetchedExits = []testExit{}
|
||||
limit = 1
|
||||
path = fmt.Sprintf(
|
||||
"%s?batchNum=%d&tokeId=%d&limit=%d&fromItem=",
|
||||
endpoint, batchNum, tokenID, limit,
|
||||
)
|
||||
err = doGoodReqPaginated(path, historydb.OrderAsc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
mixedExits := []testExit{}
|
||||
flipedExits := []testExit{}
|
||||
for i := 0; i < len(tc.exits); i++ {
|
||||
if tc.exits[i].BatchNum == batchNum && tc.exits[i].Token.TokenID == tokenID {
|
||||
mixedExits = append(mixedExits, tc.exits[i])
|
||||
}
|
||||
flipedExits = append(flipedExits, tc.exits[len(tc.exits)-1-i])
|
||||
}
|
||||
assertExitAPIs(t, mixedExits, fetchedExits)
|
||||
// All, in reverse order
|
||||
fetchedExits = []testExit{}
|
||||
limit = 5
|
||||
path = fmt.Sprintf("%s?limit=%d&fromItem=", endpoint, limit)
|
||||
err = doGoodReqPaginated(path, historydb.OrderDesc, &testExitsResponse{}, appendIter)
|
||||
assert.NoError(t, err)
|
||||
assertExitAPIs(t, flipedExits, fetchedExits)
|
||||
// 400
|
||||
path = fmt.Sprintf(
|
||||
"%s?accountIndex=%s&hermezEthereumAddress=%s",
|
||||
endpoint, idx, tc.usrAddr,
|
||||
)
|
||||
err = doBadReq("GET", path, nil, 400)
|
||||
assert.NoError(t, err)
|
||||
path = fmt.Sprintf("%s?tokenId=X", endpoint)
|
||||
err = doBadReq("GET", path, nil, 400)
|
||||
assert.NoError(t, err)
|
||||
// 404
|
||||
path = fmt.Sprintf("%s?batchNum=999999", endpoint)
|
||||
err = doBadReq("GET", path, nil, 404)
|
||||
assert.NoError(t, err)
|
||||
path = fmt.Sprintf("%s?limit=1000&fromItem=999999", endpoint)
|
||||
err = doBadReq("GET", path, nil, 404)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func TestGetExit(t *testing.T) {
|
||||
// Get all txs by their ID
|
||||
endpoint := apiURL + "exits/"
|
||||
fetchedExits := []testExit{}
|
||||
for _, exit := range tc.exits {
|
||||
fetchedExit := testExit{}
|
||||
assert.NoError(
|
||||
t, doGoodReq(
|
||||
"GET",
|
||||
fmt.Sprintf("%s%d/%s", endpoint, exit.BatchNum, exit.AccountIdx),
|
||||
nil, &fetchedExit,
|
||||
),
|
||||
)
|
||||
fetchedExits = append(fetchedExits, fetchedExit)
|
||||
}
|
||||
assertExitAPIs(t, tc.exits, fetchedExits)
|
||||
// 400
|
||||
err := doBadReq("GET", endpoint+"1/haz:BOOM:1", nil, 400)
|
||||
assert.NoError(t, err)
|
||||
err = doBadReq("GET", endpoint+"-1/hez:BOOM:1", nil, 400)
|
||||
assert.NoError(t, err)
|
||||
// 404
|
||||
err = doBadReq("GET", endpoint+"494/hez:XXX:1", nil, 404)
|
||||
assert.NoError(t, err)
|
||||
}
|
||||
|
||||
func assertExitAPIs(t *testing.T, expected, actual []testExit) {
|
||||
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
|
||||
actual[i].Token.ItemID = 0
|
||||
if expected[i].Token.USDUpdate == nil {
|
||||
assert.Equal(t, expected[i].Token.USDUpdate, actual[i].Token.USDUpdate)
|
||||
} else {
|
||||
assert.Equal(t, expected[i].Token.USDUpdate.Unix(), actual[i].Token.USDUpdate.Unix())
|
||||
expected[i].Token.USDUpdate = actual[i].Token.USDUpdate
|
||||
}
|
||||
assert.Equal(t, expected[i], actual[i])
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user