|
|
package api
import ( "encoding/base64" "math/big" "strconv" "testing"
ethCommon "github.com/ethereum/go-ethereum/common" "github.com/hermeznetwork/hermez-node/common" "github.com/iden3/go-iden3-crypto/babyjub" "github.com/stretchr/testify/assert")
type queryParser struct { m map[string]string}
func (qp *queryParser) Query(query string) string { if val, ok := qp.m[query]; ok { return val } return ""}
func TestParseQueryUint(t *testing.T) { name := "foo" c := &queryParser{} c.m = make(map[string]string) var min uint = 1 var max uint = 10 var dflt *uint // Not uint
c.m[name] = "-1" _, err := parseQueryUint(name, dflt, min, max, c) assert.Error(t, err) c.m[name] = "a" _, err = parseQueryUint(name, dflt, min, max, c) assert.Error(t, err) c.m[name] = "0.1" _, err = parseQueryUint(name, dflt, min, max, c) assert.Error(t, err) c.m[name] = "1.0" _, err = parseQueryUint(name, dflt, min, max, c) assert.Error(t, err) // Out of range
c.m[name] = strconv.Itoa(int(min) - 1) _, err = parseQueryUint(name, dflt, min, max, c) assert.Error(t, err) c.m[name] = strconv.Itoa(int(max) + 1) _, err = parseQueryUint(name, dflt, min, max, c) assert.Error(t, err) // Default nil
c.m[name] = "" res, err := parseQueryUint(name, dflt, min, max, c) assert.NoError(t, err) assert.Nil(t, res) // Default not nil
dflt = new(uint) *dflt = uint(min) res, err = parseQueryUint(name, dflt, min, max, c) assert.NoError(t, err) assert.Equal(t, uint(min), *res) // Correct
c.m[name] = strconv.Itoa(int(max)) res, err = parseQueryUint(name, res, min, max, c) assert.NoError(t, err) assert.Equal(t, uint(max), *res)}
func TestParseQueryBool(t *testing.T) { name := "foo" c := &queryParser{} c.m = make(map[string]string) var dflt *bool // Not bool
c.m[name] = "x" _, err := parseQueryBool(name, dflt, c) assert.Error(t, err) c.m[name] = "False" _, err = parseQueryBool(name, dflt, c) assert.Error(t, err) c.m[name] = "0" _, err = parseQueryBool(name, dflt, c) assert.Error(t, err) c.m[name] = "1" _, err = parseQueryBool(name, dflt, c) assert.Error(t, err) // Default nil
c.m[name] = "" res, err := parseQueryBool(name, dflt, c) assert.NoError(t, err) assert.Nil(t, res) // Default not nil
dflt = new(bool) *dflt = true res, err = parseQueryBool(name, dflt, c) assert.NoError(t, err) assert.True(t, *res) // Correct
c.m[name] = "false" res, err = parseQueryBool(name, dflt, c) assert.NoError(t, err) assert.False(t, *res) c.m[name] = "true" res, err = parseQueryBool(name, dflt, c) assert.NoError(t, err) assert.True(t, *res)}
func TestParsePagination(t *testing.T) { c := &queryParser{} c.m = make(map[string]string) // fromItem out of range
c.m["fromItem"] = "-1" _, _, _, err := parsePagination(c) assert.Error(t, err) c.m["fromItem"] = strconv.Itoa(maxUint32 + 1) _, _, _, err = parsePagination(c) assert.Error(t, err) c.m["fromItem"] = "" // Bad order
c.m["order"] = "0" _, _, _, err = parsePagination(c) assert.Error(t, err) c.m["order"] = strconv.Itoa(int(maxLimit) + 1) _, _, _, err = parsePagination(c) assert.Error(t, err) // Default
c.m["fromItem"] = "" c.m["order"] = "" c.m["limit"] = "" fromItem, order, limit, err := parsePagination(c) assert.NoError(t, err) assert.Nil(t, fromItem) assert.Equal(t, dfltOrder, order) assert.Equal(t, dfltLimit, *limit) // Correct
c.m["fromItem"] = "" c.m["order"] = "ASC" c.m["limit"] = "25" fromItem, order, limit, err = parsePagination(c) assert.NoError(t, err) assert.Nil(t, fromItem) assert.Equal(t, "ASC", order) assert.Equal(t, 25, int(*limit)) c.m["fromItem"] = "25" c.m["order"] = "DESC" c.m["limit"] = "50" fromItem, order, limit, err = parsePagination(c) assert.NoError(t, err) assert.Equal(t, 25, int(*fromItem)) assert.Equal(t, "DESC", order) assert.Equal(t, 50, int(*limit))}
func TestParseQueryHezEthAddr(t *testing.T) { name := "hermezEthereumAddress" c := &queryParser{} c.m = make(map[string]string) ethAddr := ethCommon.BigToAddress(big.NewInt(int64(347683))) // Not HEZ Eth addr
c.m[name] = "hez:0xf" _, err := parseQueryHezEthAddr(c) assert.Error(t, err) c.m[name] = ethAddr.String() _, err = parseQueryHezEthAddr(c) assert.Error(t, err) c.m[name] = "hez:0xXX942cfcd25ad4d90a62358b0dd84f33b39826XX" _, err = parseQueryHezEthAddr(c) assert.Error(t, err) // Default
c.m[name] = "" res, err := parseQueryHezEthAddr(c) assert.NoError(t, err) assert.Nil(t, res) // Correct
c.m[name] = "hez:" + ethAddr.String() res, err = parseQueryHezEthAddr(c) assert.NoError(t, err) assert.Equal(t, ethAddr, *res)}
func TestParseQueryBJJ(t *testing.T) { name := "BJJ" c := &queryParser{} c.m = make(map[string]string) privK := babyjub.NewRandPrivKey() pubK := privK.Public() pkComp := [32]byte(pubK.Compress()) // Not HEZ Eth addr
c.m[name] = "hez:abcd" _, err := parseQueryBJJ(c) assert.Error(t, err) c.m[name] = pubK.String() _, err = parseQueryBJJ(c) assert.Error(t, err) // Wrong checksum
bjjSum := append(pkComp[:], byte(1)) c.m[name] = "hez:" + base64.RawStdEncoding.EncodeToString(bjjSum) _, err = parseQueryBJJ(c) assert.Error(t, err) // Default
c.m[name] = "" res, err := parseQueryBJJ(c) assert.NoError(t, err) assert.Nil(t, res) // Correct
c.m[name] = bjjToString(pubK) res, err = parseQueryBJJ(c) assert.NoError(t, err) assert.Equal(t, *pubK, *res)}
func TestParseQueryTxType(t *testing.T) { name := "type" c := &queryParser{} c.m = make(map[string]string) // Incorrect values
c.m[name] = "deposit" _, err := parseQueryTxType(c) assert.Error(t, err) c.m[name] = "1" _, err = parseQueryTxType(c) assert.Error(t, err) // Default
c.m[name] = "" res, err := parseQueryTxType(c) assert.NoError(t, err) assert.Nil(t, res) // Correct values
c.m[name] = string(common.TxTypeExit) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeExit, *res) c.m[name] = string(common.TxTypeTransfer) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeTransfer, *res) c.m[name] = string(common.TxTypeDeposit) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeDeposit, *res) c.m[name] = string(common.TxTypeCreateAccountDeposit) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeCreateAccountDeposit, *res) c.m[name] = string(common.TxTypeCreateAccountDepositTransfer) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeCreateAccountDepositTransfer, *res) c.m[name] = string(common.TxTypeDepositTransfer) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeDepositTransfer, *res) c.m[name] = string(common.TxTypeForceTransfer) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeForceTransfer, *res) c.m[name] = string(common.TxTypeForceExit) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeForceExit, *res) c.m[name] = string(common.TxTypeTransferToEthAddr) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeTransferToEthAddr, *res) c.m[name] = string(common.TxTypeTransferToBJJ) res, err = parseQueryTxType(c) assert.NoError(t, err) assert.Equal(t, common.TxTypeTransferToBJJ, *res)}
|
