hermez-node/test/l2db.go

package test

import (
	"math/big"
	"strconv"
	"time"

	ethCommon "github.com/ethereum/go-ethereum/common"
	"github.com/hermeznetwork/hermez-node/common"
	"github.com/iden3/go-iden3-crypto/babyjub"
	"github.com/jmoiron/sqlx"
)

// CleanL2DB deletes 'tx_pool' and 'account_creation_auth' from the given DB
func CleanL2DB(db *sqlx.DB) {
	if _, err := db.Exec("DELETE FROM tx_pool;"); err != nil {
		panic(err)
	}
	if _, err := db.Exec("DELETE FROM account_creation_auth;"); err != nil {
		panic(err)
	}
}

// GenPoolTxs generates L2 pool txs.
// WARNING: This tx doesn't follow the protocol (signature, txID, ...)
// it's just to test getting/setting from/to the DB.
func GenPoolTxs(n int, tokens []common.Token) []*common.PoolL2Tx {
	txs := make([]*common.PoolL2Tx, 0, n)
	privK := babyjub.NewRandPrivKey()
	for i := 256; i < 256+n; i++ {
		var state common.PoolL2TxState
		//nolint:gomnd
		if i%4 == 0 {
			state = common.PoolL2TxStatePending
			//nolint:gomnd
		} else if i%4 == 1 {
			state = common.PoolL2TxStateInvalid
			//nolint:gomnd
		} else if i%4 == 2 {
			state = common.PoolL2TxStateForging
			//nolint:gomnd
		} else if i%4 == 3 {
			state = common.PoolL2TxStateForged
		}
		fee := common.FeeSelector(i % 255) //nolint:gomnd
		token := tokens[i%len(tokens)]
		tx := &common.PoolL2Tx{
			FromIdx:   common.Idx(i),
			ToIdx:     common.Idx(i + 1),
			ToEthAddr: ethCommon.BigToAddress(big.NewInt(int64(i))),
			ToBJJ:     privK.Public(),
			TokenID:   token.TokenID,
			Amount:    big.NewInt(int64(i)),
			Fee:       fee,
			Nonce:     common.Nonce(i),
			State:     state,
			Signature: privK.SignPoseidon(big.NewInt(int64(i))),
		}
		var err error
		tx, err = common.NewPoolL2Tx(tx)
		if err != nil {
			panic(err)
		}
		if i%2 == 0 { // Optional parameters: rq
			tx.RqFromIdx = common.Idx(i)
			tx.RqToIdx = common.Idx(i + 1)
			tx.RqToEthAddr = ethCommon.BigToAddress(big.NewInt(int64(i)))
			tx.RqToBJJ = privK.Public()
			tx.RqTokenID = common.TokenID(i)
			tx.RqAmount = big.NewInt(int64(i))
			tx.RqFee = common.FeeSelector(i)
			tx.RqNonce = common.Nonce(i)
		}
		txs = append(txs, tx)
	}
	return txs
}

// GenAuths generates account creation authorizations
func GenAuths(nAuths int) []*common.AccountCreationAuth {
	auths := []*common.AccountCreationAuth{}
	for i := 0; i < nAuths; i++ {
		privK := babyjub.NewRandPrivKey()
		auths = append(auths, &common.AccountCreationAuth{
			EthAddr:   ethCommon.BigToAddress(big.NewInt(int64(i))),
			BJJ:       privK.Public(),
			Signature: []byte(strconv.Itoa(i)),
			Timestamp: time.Now(),
		})
	}
	return auths
}