Implement L2DB

test/l2db.go

@@ -1,6 +1,15 @@
 package test
 
-import "github.com/jmoiron/sqlx"
+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) {
@@ -11,3 +20,77 @@ func CleanL2DB(db *sqlx.DB) {
 		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) []*common.PoolL2Tx {
+	txs := make([]*common.PoolL2Tx, 0, n)
+	privK := babyjub.NewRandPrivKey()
+	for i := 0; i < 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
+		}
+		f := new(big.Float).SetInt(big.NewInt(int64(i)))
+		amountF, _ := f.Float64()
+		tx := &common.PoolL2Tx{
+			TxID:        common.TxID(common.Hash([]byte(strconv.Itoa(i)))),
+			FromIdx:     common.Idx(i),
+			ToIdx:       common.Idx(i + 1),
+			ToEthAddr:   ethCommon.BigToAddress(big.NewInt(int64(i))),
+			ToBJJ:       privK.Public(),
+			TokenID:     common.TokenID(i),
+			Amount:      big.NewInt(int64(i)),
+			AmountFloat: amountF,
+			//nolint:gomnd
+			Fee:       common.FeeSelector(i % 255),
+			Nonce:     common.Nonce(i),
+			State:     state,
+			Signature: privK.SignPoseidon(big.NewInt(int64(i))),
+			Timestamp: time.Now().UTC(),
+		}
+		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 = uint64(i)
+		}
+		if i%3 == 0 { // Optional parameters: things that get updated "a posteriori"
+			tx.BatchNum = 489
+			tx.AbsoluteFee = 39.12345
+			tx.AbsoluteFeeUpdate = time.Now().UTC()
+		}
+		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
+}