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())
}