mirror of
https://github.com/arnaucube/hermez-node.git
synced 2026-02-08 20:06:44 +01:00
Add StateDB txprocessor tests using txtest generators
Add StateDB tx processor tests using tx test generators: - update statedb/ tx processors - add statedb tx processors tests
This commit is contained in:
93
db/statedb/txprocessors_test.go
Normal file
93
db/statedb/txprocessors_test.go
Normal file
@@ -0,0 +1,93 @@
|
||||
package statedb
|
||||
|
||||
import (
|
||||
"io/ioutil"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/hermeznetwork/hermez-node/common"
|
||||
"github.com/hermeznetwork/hermez-node/test"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestProcessTxs(t *testing.T) {
|
||||
dir, err := ioutil.TempDir("", "tmpdb")
|
||||
require.Nil(t, err)
|
||||
|
||||
sdb, err := NewStateDB(dir, true, 32)
|
||||
assert.Nil(t, err)
|
||||
|
||||
// generate test transactions from test.SetTest0 code
|
||||
parser := test.NewParser(strings.NewReader(test.SetTest0))
|
||||
instructions, err := parser.Parse()
|
||||
assert.Nil(t, err)
|
||||
|
||||
l1Txs, coordinatorL1Txs, poolL2Txs := test.GenerateTestTxs(t, instructions)
|
||||
assert.Equal(t, 29, len(l1Txs[0]))
|
||||
assert.Equal(t, 0, len(coordinatorL1Txs[0]))
|
||||
assert.Equal(t, 21, len(poolL2Txs[0]))
|
||||
|
||||
// iterate for each batch
|
||||
for i := 0; i < len(l1Txs); i++ {
|
||||
l2Txs := common.PoolL2TxsToL2Txs(poolL2Txs[i])
|
||||
|
||||
_, _, err := sdb.ProcessTxs(true, l1Txs[i], coordinatorL1Txs[i], l2Txs)
|
||||
require.Nil(t, err)
|
||||
}
|
||||
|
||||
acc, err := sdb.GetAccount(common.Idx(1))
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "23", acc.Balance.String())
|
||||
}
|
||||
|
||||
func TestProcessTxsBatchByBatch(t *testing.T) {
|
||||
dir, err := ioutil.TempDir("", "tmpdb")
|
||||
require.Nil(t, err)
|
||||
|
||||
sdb, err := NewStateDB(dir, true, 32)
|
||||
assert.Nil(t, err)
|
||||
|
||||
// generate test transactions from test.SetTest0 code
|
||||
parser := test.NewParser(strings.NewReader(test.SetTest0))
|
||||
instructions, err := parser.Parse()
|
||||
assert.Nil(t, err)
|
||||
|
||||
l1Txs, coordinatorL1Txs, poolL2Txs := test.GenerateTestTxs(t, instructions)
|
||||
assert.Equal(t, 29, len(l1Txs[0]))
|
||||
assert.Equal(t, 0, len(coordinatorL1Txs[0]))
|
||||
assert.Equal(t, 21, len(poolL2Txs[0]))
|
||||
assert.Equal(t, 5, len(l1Txs[1]))
|
||||
assert.Equal(t, 1, len(coordinatorL1Txs[1]))
|
||||
assert.Equal(t, 55, len(poolL2Txs[1]))
|
||||
assert.Equal(t, 10, len(l1Txs[2]))
|
||||
assert.Equal(t, 0, len(coordinatorL1Txs[2]))
|
||||
assert.Equal(t, 7, len(poolL2Txs[2]))
|
||||
|
||||
// use first batch
|
||||
l2txs := common.PoolL2TxsToL2Txs(poolL2Txs[0])
|
||||
_, exitInfos, err := sdb.ProcessTxs(true, l1Txs[0], coordinatorL1Txs[0], l2txs)
|
||||
require.Nil(t, err)
|
||||
assert.Equal(t, 0, len(exitInfos))
|
||||
acc, err := sdb.GetAccount(common.Idx(1))
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "28", acc.Balance.String())
|
||||
|
||||
// use second batch
|
||||
l2txs = common.PoolL2TxsToL2Txs(poolL2Txs[1])
|
||||
_, exitInfos, err = sdb.ProcessTxs(true, l1Txs[1], coordinatorL1Txs[1], l2txs)
|
||||
require.Nil(t, err)
|
||||
assert.Equal(t, 5, len(exitInfos))
|
||||
acc, err = sdb.GetAccount(common.Idx(1))
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "48", acc.Balance.String())
|
||||
|
||||
// use third batch
|
||||
l2txs = common.PoolL2TxsToL2Txs(poolL2Txs[2])
|
||||
_, exitInfos, err = sdb.ProcessTxs(true, l1Txs[2], coordinatorL1Txs[2], l2txs)
|
||||
require.Nil(t, err)
|
||||
assert.Equal(t, 1, len(exitInfos))
|
||||
acc, err = sdb.GetAccount(common.Idx(1))
|
||||
assert.Nil(t, err)
|
||||
assert.Equal(t, "23", acc.Balance.String())
|
||||
}
|
||||
Reference in New Issue
Block a user