You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.8 KiB
93 lines
2.8 KiB
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())
|
|
}
|