TxSelector include L1CoordinatorTxs in selections

txselector/txselector_test.go

@@ -1,7 +1,26 @@
 package txselector
 
-/*
-   TODO update transactions generation
+import (
+	"io/ioutil"
+	"os"
+	"strconv"
+	"testing"
+	"time"
+
+	ethCommon "github.com/ethereum/go-ethereum/common"
+	"github.com/hermeznetwork/hermez-node/common"
+	dbUtils "github.com/hermeznetwork/hermez-node/db"
+	"github.com/hermeznetwork/hermez-node/db/historydb"
+	"github.com/hermeznetwork/hermez-node/db/l2db"
+	"github.com/hermeznetwork/hermez-node/db/statedb"
+	"github.com/hermeznetwork/hermez-node/eth"
+	"github.com/hermeznetwork/hermez-node/test"
+	"github.com/hermeznetwork/hermez-node/test/til"
+	"github.com/jmoiron/sqlx"
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
 func initTest(t *testing.T, testSet string, maxL1UserTxs, maxL1OperatorTxs, maxTxs uint64) *TxSelector {
 	pass := os.Getenv("POSTGRES_PASS")
 	db, err := dbUtils.InitSQLDB(5432, "localhost", "hermez", pass, "hermez")
@@ -42,34 +61,67 @@ func TestGetL2TxSelection(t *testing.T) {
 	txsel := initTest(t, til.SetPool0, 5, 5, 10)
 	test.CleanL2DB(txsel.l2db.DB())
 
-	   	// generate test transactions
-	   	l1Txs, _, poolL2Txs, tokens := test.GenerateTestTxsFromSet(t, test.SetTest0)
+	tc := til.NewContext(eth.RollupConstMaxL1UserTx)
+	// generate test transactions
+	blocks, err := tc.GenerateBlocks(til.SetBlockchain0)
+	assert.Nil(t, err)
+	// poolL2Txs, err := tc.GeneratePoolL2Txs(til.SetPool0)
+	// assert.Nil(t, err)
 
-	   	// add tokens to HistoryDB to avoid breaking FK constrains
-	   	addTokens(t, tokens, txsel.l2db.DB())
-	   	// add the first batch of transactions to the TxSelector
-	   	addL2Txs(t, txsel, poolL2Txs[0])
+	coordIdxs := []common.Idx{256, 257, 258, 259}
 
-	   	_, err := txsel.GetL2TxSelection(0)
-	   	assert.Nil(t, err)
+	// add tokens to HistoryDB to avoid breaking FK constrains
+	var tokens []common.Token
+	for i := 0; i < int(tc.LastRegisteredTokenID); i++ {
+		tokens = append(tokens, common.Token{
+			TokenID:     common.TokenID(i),
+			EthBlockNum: 1,
+			EthAddr:     ethCommon.BytesToAddress([]byte{byte(i)}),
+			Name:        strconv.Itoa(i),
+			Symbol:      strconv.Itoa(i),
+			Decimals:    18,
+		})
+	}
+	addTokens(t, tokens, txsel.l2db.DB())
 
-	   	_, _, _, err = txsel.GetL1L2TxSelection(0, l1Txs[0])
-	   	assert.Nil(t, err)
+	// Process the 1st batch, which contains the L1CoordinatorTxs necessary
+	// to create the Coordinator accounts to receive the fees
+	_, err = txsel.localAccountsDB.ProcessTxs(nil, nil, blocks[0].Batches[0].L1CoordinatorTxs, nil)
+	require.Nil(t, err)
 
-		// TODO once L2DB is updated to return error in case that AddTxTest
-		// fails, and the Til is updated, update this test, checking that the
-		// selected PoolL2Tx are correctly sorted by Nonce
+	// add the 1st batch of transactions to the TxSelector
+	addL2Txs(t, txsel, common.L2TxsToPoolL2Txs(blocks[0].Batches[0].L2Txs))
 
+	l1CoordTxs, l2Txs, err := txsel.GetL2TxSelection(coordIdxs, 0)
+	assert.Nil(t, err)
+	assert.Equal(t, 0, len(l2Txs))
+	assert.Equal(t, 0, len(l1CoordTxs))
 
+	_, _, _, err = txsel.GetL1L2TxSelection(coordIdxs, 0, blocks[0].L1UserTxs)
+	assert.Nil(t, err)
 
-	   	// txs, err := txsel.GetL2TxSelection(0)
-	   	// assert.Nil(t, err)
-	   	// for _, tx := range txs {
-	   	//         fmt.Println(tx.FromIdx, tx.ToIdx, tx.AbsoluteFee)
-	   	// }
-	   	// assert.Equal(t, 3, len(txs))
-	   	// assert.Equal(t, uint64(6), txs[0].AbsoluteFee)
-	   	// assert.Equal(t, uint64(5), txs[1].AbsoluteFee)
-	   	// assert.Equal(t, uint64(4), txs[2].AbsoluteFee)
+	// TODO once L2DB is updated to return error in case that AddTxTest
+	// fails, and the Til is updated, update this test, checking that the
+	// selected PoolL2Tx are correctly sorted by Nonce
+
+	// TODO once L2DB is updated to store the parameter AbsoluteFee (which
+	// is used by TxSelector to sort L2Txs), uncomment this next lines of
+	// test, and put the expected value for
+	// l2Txs[len(l2Txs)-1].AbsoluteFee, which is the Tx which has the
+	// Fee==192.
+	/*
+		// add the 3rd batch of transactions to the TxSelector
+		addL2Txs(t, txsel, common.L2TxsToPoolL2Txs(blocks[0].Batches[2].L2Txs))
+
+		_, l2Txs, err = txsel.GetL2TxSelection(coordIdxs, 0)
+		assert.Nil(t, err)
+		for _, tx := range l2Txs {
+			fmt.Println(tx.FromIdx, tx.ToIdx, tx.AbsoluteFee)
+		}
+		require.Equal(t, 10, len(l2Txs))
+		assert.Equal(t, float64(0), l2Txs[0].AbsoluteFee)
+
+		fmt.Println(l2Txs[len(l2Txs)-1].Amount)
+		assert.Equal(t, float64(4), l2Txs[len(l2Txs)-1].AbsoluteFee)
+	*/
 }
-*/