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.
124 lines
3.6 KiB
124 lines
3.6 KiB
package statedb
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"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"
|
|
)
|
|
|
|
var debug = false
|
|
|
|
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, true, l1Txs[i], coordinatorL1Txs[i], poolL2Txs[i])
|
|
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, true, l1Txs[0], coordinatorL1Txs[0], poolL2Txs[0])
|
|
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, true, l1Txs[1], coordinatorL1Txs[1], poolL2Txs[1])
|
|
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, true, l1Txs[2], coordinatorL1Txs[2], poolL2Txs[2])
|
|
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())
|
|
}
|
|
|
|
func TestZKInputsGeneration(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]))
|
|
|
|
zki, _, err := sdb.ProcessTxs(false, true, l1Txs[0], coordinatorL1Txs[0], poolL2Txs[0])
|
|
require.Nil(t, err)
|
|
|
|
s, err := json.Marshal(zki)
|
|
require.Nil(t, err)
|
|
if debug {
|
|
fmt.Println(string(s))
|
|
}
|
|
}
|