mirror of
https://github.com/arnaucube/hermez-node.git
synced 2026-02-07 11:26:44 +01:00
41715e47ba
- GetL2TxSelection & GetL1L2TxSelection integrated with dbs - Create L1CoordinatorTx of type CreateAccountDeposit when a L2 requires it (and the AccountCreationAuth exists)
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
package txselector
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/hermeznetwork/hermez-node/common"
|
|
"github.com/hermeznetwork/hermez-node/db/l2db"
|
|
"github.com/hermeznetwork/hermez-node/db/statedb"
|
|
"github.com/hermeznetwork/hermez-node/test"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func initTest(t *testing.T, testSet string) *TxSelector {
|
|
pass := os.Getenv("POSTGRES_PASS")
|
|
l2DB, err := l2db.NewL2DB(5432, "localhost", "hermez", pass, "l2", 10, 512, 24*time.Hour)
|
|
require.Nil(t, err)
|
|
|
|
dir, err := ioutil.TempDir("", "tmpdb")
|
|
require.Nil(t, err)
|
|
sdb, err := statedb.NewStateDB(dir, false, 0)
|
|
require.Nil(t, err)
|
|
|
|
txselDir, err := ioutil.TempDir("", "tmpTxSelDB")
|
|
require.Nil(t, err)
|
|
txsel, err := NewTxSelector(txselDir, sdb, l2DB, 100, 100, 1000)
|
|
require.Nil(t, err)
|
|
|
|
return txsel
|
|
}
|
|
func addL2Txs(t *testing.T, txsel *TxSelector, poolL2Txs []*common.PoolL2Tx) {
|
|
for i := 0; i < len(poolL2Txs); i++ {
|
|
err := txsel.l2db.AddTx(poolL2Txs[i])
|
|
require.Nil(t, err)
|
|
}
|
|
}
|
|
|
|
func TestGetL2TxSelection(t *testing.T) {
|
|
txsel := initTest(t, test.SetTest0)
|
|
test.CleanL2DB(txsel.l2db.DB())
|
|
|
|
// generate test transactions
|
|
l1Txs, _, poolL2Txs := test.GenerateTestTxsFromSet(t, test.SetTest0)
|
|
|
|
// add the first batch of transactions to the TxSelector
|
|
addL2Txs(t, txsel, poolL2Txs[0])
|
|
|
|
_, err := txsel.GetL2TxSelection(0)
|
|
assert.Nil(t, err)
|
|
|
|
_, _, _, err = txsel.GetL1L2TxSelection(0, l1Txs[0])
|
|
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)
|
|
}
|