arnaucube/hermez-node
1
0
Fork 0
mirror of https://github.com/arnaucube/hermez-node.git synced 2026-02-08 20:06:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add StateDB ZKInputs generation for L1 & L2 Txs

Browse Source
This commit is contained in:
arnaucube
2020-09-09 14:40:55 +02:00
parent a7fe80f150
commit aa0bde61d2
7 changed files with 251 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
db/statedb/txprocessors_test.go
View File

@@ -1,6 +1,8 @@
package statedb
import (
"encoding/json"
"fmt"
"io/ioutil"
"strings"
"testing"
@@ -11,6 +13,8 @@ import (
"github.com/stretchr/testify/require"
)
var debug = false
func TestProcessTxs(t *testing.T) {
dir, err := ioutil.TempDir("", "tmpdb")
require.Nil(t, err)
@@ -30,9 +34,9 @@ func TestProcessTxs(t *testing.T) {
// iterate for each batch
for i := 0; i < len(l1Txs); i++ {
l2Txs := common.PoolL2TxsToL2Txs(poolL2Txs[i])
// l2Txs := common.PoolL2TxsToL2Txs(poolL2Txs[i])
_, _, err := sdb.ProcessTxs(true, l1Txs[i], coordinatorL1Txs[i], l2Txs)
_, _, err := sdb.ProcessTxs(true, true, l1Txs[i], coordinatorL1Txs[i], poolL2Txs[i])
require.Nil(t, err)
}
@@ -65,8 +69,8 @@ func TestProcessTxsBatchByBatch(t *testing.T) {
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)
// 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))
@@ -74,8 +78,8 @@ func TestProcessTxsBatchByBatch(t *testing.T) {
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)
// 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))
@@ -83,11 +87,38 @@ func TestProcessTxsBatchByBatch(t *testing.T) {
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)
// 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))
}
}