package api import ( "encoding/base64" "math/big" "strconv" "strings" "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 (qp *queryParser) Param(param string) string { if val, ok := qp.m[param]; 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 := "hezEthereumAddress" 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) } func TestParseTokenFilters(t *testing.T) { ids := "ids" symbols := "symbols" name := "name" nameValue := "1" symbolsValue := "1,2,3" idsValue := "2,3,4" c := &queryParser{} c.m = make(map[string]string) // Incorrect values c.m[name] = nameValue c.m[ids] = idsValue c.m[symbols] = symbolsValue idsParse, symbolsParse, nameParse, err := parseTokenFilters(c) assert.NoError(t, err) // Correct values var tokenIds []common.TokenID = []common.TokenID{2, 3, 4} assert.Equal(t, tokenIds, idsParse) var symbolsArray []string = []string{"1", "2", "3"} assert.Equal(t, symbolsArray, symbolsParse) assert.Equal(t, nameValue, nameParse) } func TestParseEthAddr(t *testing.T) { name := "forgerAddr" c := &queryParser{} c.m = make(map[string]string) ethAddr := ethCommon.BigToAddress(big.NewInt(int64(123456))) // Default c.m[name] = "" res, err := parseQueryEthAddr(name, c) assert.NoError(t, err) assert.Nil(t, res) // Incorrect c.m[name] = "0x12345678" _, err = parseQueryEthAddr(name, c) assert.Error(t, err) // Correct c.m[name] = ethAddr.String() res, err = parseQueryEthAddr(name, c) assert.NoError(t, err) assert.Equal(t, ethAddr, *res) } func TestParseBidFilters(t *testing.T) { slotNum := "slotNum" bidderAddr := "bidderAddr" slotNumValue := "2" bidderAddrValue := "0xaa942cfcd25ad4d90a62358b0dd84f33b398262a" c := &queryParser{} c.m = make(map[string]string) // Incorrect values c.m[slotNum] = slotNumValue c.m[bidderAddr] = bidderAddrValue slotNumParse, bidderAddrParse, err := parseBidFilters(c) assert.NoError(t, err) // Correct values assert.Equal(t, strings.ToLower(bidderAddrParse.Hex()), bidderAddrValue) assert.Equal(t, slotNumValue, strconv.FormatUint(uint64(*slotNumParse), 10)) }