package til import ( "math/big" "testing" "github.com/hermeznetwork/hermez-node/common" "github.com/stretchr/testify/assert" "github.com/stretchr/testify/require" ) func TestGenerateBlocksNoBatches(t *testing.T) { set := ` Type: Blockchain AddToken(1) AddToken(2) CreateAccountDeposit(1) A: 11 CreateAccountDeposit(2) B: 22 > block ` tc := NewContext(common.RollupConstMaxL1UserTx) blocks, err := tc.GenerateBlocks(set) require.NoError(t, err) assert.Equal(t, 1, len(blocks)) assert.Equal(t, 0, len(blocks[0].Rollup.Batches)) assert.Equal(t, 2, len(blocks[0].Rollup.AddedTokens)) assert.Equal(t, 2, len(blocks[0].Rollup.L1UserTxs)) } func TestGenerateBlocks(t *testing.T) { set := ` Type: Blockchain AddToken(1) AddToken(2) AddToken(3) CreateAccountDeposit(1) A: 10 CreateAccountDeposit(2) A: 20 CreateAccountDeposit(1) B: 5 CreateAccountDeposit(1) C: 5 CreateAccountDeposit(1) D: 5 > batchL1 // batchNum = 1 > batchL1 // batchNum = 2 CreateAccountDepositTransfer(1) F-A: 15, 10 Transfer(1) A-B: 6 (1) Transfer(1) B-D: 3 (1) Transfer(1) A-D: 1 (1) // set new batch > batch // batchNum = 3 CreateAccountCoordinator(1) E CreateAccountCoordinator(2) B DepositTransfer(1) A-B: 15, 10 Transfer(1) C-A : 3 (1) Transfer(2) A-B: 15 (1) Transfer(1) A-E: 1 (1) CreateAccountDeposit(1) User0: 20 CreateAccountDeposit(3) User1: 20 CreateAccountCoordinator(1) User1 CreateAccountCoordinator(3) User0 > batchL1 // batchNum = 4 Transfer(1) User0-User1: 15 (1) Transfer(3) User1-User0: 15 (1) Transfer(1) A-C: 1 (1) > batchL1 // batchNum = 5 Transfer(1) User1-User0: 1 (1) > block // Exits Transfer(1) A-B: 1 (1) Exit(1) A: 5 (1) > batch // batchNum = 6 > block // this transaction should not be generated, as it's after last // batch and last block Transfer(1) User1-User0: 1 (1) ` tc := NewContext(common.RollupConstMaxL1UserTx) blocks, err := tc.GenerateBlocks(set) require.NoError(t, err) assert.Equal(t, 2, len(blocks)) assert.Equal(t, 5, len(blocks[0].Rollup.Batches)) assert.Equal(t, 1, len(blocks[1].Rollup.Batches)) assert.Equal(t, 9, len(blocks[0].Rollup.L1UserTxs)) assert.Equal(t, 4, len(blocks[0].Rollup.Batches[3].L1CoordinatorTxs)) assert.Equal(t, 0, len(blocks[1].Rollup.L1UserTxs)) // Check expected values generated by each line // #0: Deposit(1) A: 10 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[0], common.TxTypeCreateAccountDeposit, 1, "A", "", big.NewInt(10), nil) // #1: Deposit(2) A: 20 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[1], common.TxTypeCreateAccountDeposit, 2, "A", "", big.NewInt(20), nil) // // #2: Deposit(1) A: 20 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[2], common.TxTypeCreateAccountDeposit, 1, "B", "", big.NewInt(5), nil) // // #3: CreateAccountDeposit(1) C: 5 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[3], common.TxTypeCreateAccountDeposit, 1, "C", "", big.NewInt(5), nil) // // #4: CreateAccountDeposit(1) D: 5 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[4], common.TxTypeCreateAccountDeposit, 1, "D", "", big.NewInt(5), nil) // #5: Transfer(1) A-B: 6 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[2].L2Txs[0], common.TxTypeTransfer, 1, "A", "B", big.NewInt(6), common.BatchNum(3)) // #6: Transfer(1) B-D: 3 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[2].L2Txs[1], common.TxTypeTransfer, 1, "B", "D", big.NewInt(3), common.BatchNum(3)) // #7: Transfer(1) A-D: 1 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[2].L2Txs[2], common.TxTypeTransfer, 1, "A", "D", big.NewInt(1), common.BatchNum(3)) // change of Batch // #8: CreateAccountDepositTransfer(1) F-A: 15, 10 (3) tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[5], common.TxTypeCreateAccountDepositTransfer, 1, "F", "A", big.NewInt(15), big.NewInt(10)) // #9: DepositTransfer(1) A-B: 15, 10 (1) tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[6], common.TxTypeDepositTransfer, 1, "A", "B", big.NewInt(15), big.NewInt(10)) // #11: Transfer(1) C-A : 3 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[3].L2Txs[0], common.TxTypeTransfer, 1, "C", "A", big.NewInt(3), common.BatchNum(4)) // #12: Transfer(2) A-B: 15 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[3].L2Txs[1], common.TxTypeTransfer, 2, "A", "B", big.NewInt(15), common.BatchNum(4)) // #13: Deposit(1) User0: 20 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[7], common.TxTypeCreateAccountDeposit, 1, "User0", "", big.NewInt(20), nil) // // #14: Deposit(3) User1: 20 tc.checkL1TxParams(t, blocks[0].Rollup.L1UserTxs[8], common.TxTypeCreateAccountDeposit, 3, "User1", "", big.NewInt(20), nil) // #15: Transfer(1) User0-User1: 15 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[4].L2Txs[0], common.TxTypeTransfer, 1, "User0", "User1", big.NewInt(15), common.BatchNum(5)) // #16: Transfer(3) User1-User0: 15 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[4].L2Txs[1], common.TxTypeTransfer, 3, "User1", "User0", big.NewInt(15), common.BatchNum(5)) // #17: Transfer(1) A-C: 1 (1) tc.checkL2TxParams(t, blocks[0].Rollup.Batches[4].L2Txs[2], common.TxTypeTransfer, 1, "A", "C", big.NewInt(1), common.BatchNum(5)) // change of Batch // #18: Transfer(1) User1-User0: 1 (1) tc.checkL2TxParams(t, blocks[1].Rollup.Batches[0].L2Txs[0], common.TxTypeTransfer, 1, "User1", "User0", big.NewInt(1), common.BatchNum(6)) // change of Block (implies also a change of batch) // #19: Transfer(1) A-B: 1 (1) tc.checkL2TxParams(t, blocks[1].Rollup.Batches[0].L2Txs[1], common.TxTypeTransfer, 1, "A", "B", big.NewInt(1), common.BatchNum(6)) } func (tc *Context) checkL1TxParams(t *testing.T, tx common.L1Tx, typ common.TxType, tokenID common.TokenID, from, to string, depositAmount, amount *big.Int) { assert.Equal(t, typ, tx.Type) if tx.FromIdx != common.Idx(0) { assert.Equal(t, tc.Users[from].Accounts[tokenID].Idx, tx.FromIdx) } assert.Equal(t, tc.Users[from].Addr.Hex(), tx.FromEthAddr.Hex()) assert.Equal(t, tc.Users[from].BJJ.Public().Compress(), tx.FromBJJ) if tx.ToIdx != common.Idx(0) { assert.Equal(t, tc.Users[to].Accounts[tokenID].Idx, tx.ToIdx) } if depositAmount != nil { assert.Equal(t, depositAmount, tx.DepositAmount) } if amount != nil { assert.Equal(t, amount, tx.Amount) } } func (tc *Context) checkL2TxParams(t *testing.T, tx common.L2Tx, typ common.TxType, tokenID common.TokenID, from, to string, amount *big.Int, batchNum common.BatchNum) { assert.Equal(t, typ, tx.Type) assert.Equal(t, tc.Users[from].Accounts[tokenID].Idx, tx.FromIdx) if tx.Type != common.TxTypeExit { assert.Equal(t, tc.Users[to].Accounts[tokenID].Idx, tx.ToIdx) } if amount != nil { assert.Equal(t, amount, tx.Amount) } assert.Equal(t, batchNum, tx.BatchNum) } func TestGeneratePoolL2Txs(t *testing.T) { set := ` Type: Blockchain AddToken(1) AddToken(2) AddToken(3) CreateAccountDeposit(1) A: 10 CreateAccountDeposit(2) A: 20 CreateAccountDeposit(1) B: 5 CreateAccountDeposit(1) C: 5 CreateAccountDeposit(1) User0: 5 CreateAccountDeposit(1) User1: 0 CreateAccountDeposit(3) User0: 0 CreateAccountDeposit(3) User1: 5 CreateAccountDeposit(2) B: 5 CreateAccountDeposit(2) D: 0 > batchL1 > batchL1 ` tc := NewContext(common.RollupConstMaxL1UserTx) _, err := tc.GenerateBlocks(set) require.NoError(t, err) set = ` Type: PoolL2 PoolTransfer(1) A-B: 6 (1) PoolTransfer(1) B-C: 3 (1) PoolTransfer(1) C-A: 3 (1) PoolTransfer(1) A-B: 1 (1) PoolTransfer(2) A-B: 15 (1) PoolTransfer(1) User0-User1: 15 (1) PoolTransfer(3) User1-User0: 15 (1) PoolTransfer(2) B-D: 3 (1) PoolExit(1) A: 3 (1) PoolTransferToEthAddr(1) A-B: 1 (1) PoolTransferToBJJ(1) A-B: 1 (1) ` poolL2Txs, err := tc.GeneratePoolL2Txs(set) require.NoError(t, err) assert.Equal(t, 11, len(poolL2Txs)) assert.Equal(t, common.TxTypeTransfer, poolL2Txs[0].Type) assert.Equal(t, common.TxTypeExit, poolL2Txs[8].Type) assert.Equal(t, tc.Users["B"].Addr.Hex(), poolL2Txs[0].ToEthAddr.Hex()) assert.Equal(t, tc.Users["B"].BJJ.Public().String(), poolL2Txs[0].ToBJJ.String()) assert.Equal(t, tc.Users["User1"].Addr.Hex(), poolL2Txs[5].ToEthAddr.Hex()) assert.Equal(t, tc.Users["User1"].BJJ.Public().String(), poolL2Txs[5].ToBJJ.String()) assert.Equal(t, common.Nonce(1), poolL2Txs[0].Nonce) assert.Equal(t, common.Nonce(2), poolL2Txs[3].Nonce) assert.Equal(t, common.Nonce(3), poolL2Txs[8].Nonce) assert.Equal(t, tc.Users["B"].Addr.Hex(), poolL2Txs[9].ToEthAddr.Hex()) assert.Equal(t, common.EmptyBJJComp, poolL2Txs[9].ToBJJ) assert.Equal(t, common.TxTypeTransferToEthAddr, poolL2Txs[9].Type) assert.Equal(t, common.FFAddr, poolL2Txs[10].ToEthAddr) assert.Equal(t, tc.Users["B"].BJJ.Public().String(), poolL2Txs[10].ToBJJ.String()) assert.Equal(t, common.TxTypeTransferToBJJ, poolL2Txs[10].Type) // load another set in the same Context set = ` Type: PoolL2 PoolTransfer(1) A-B: 6 (1) PoolTransfer(1) B-C: 3 (1) PoolTransfer(1) A-C: 3 (1) ` poolL2Txs, err = tc.GeneratePoolL2Txs(set) require.NoError(t, err) assert.Equal(t, common.Nonce(6), poolL2Txs[0].Nonce) assert.Equal(t, common.Nonce(2), poolL2Txs[1].Nonce) assert.Equal(t, common.Nonce(7), poolL2Txs[2].Nonce) // check that a PoolL2Tx can be done to a non existing ToIdx set = ` Type: Blockchain AddToken(1) CreateAccountDeposit(1) A: 10 > batchL1 > batchL1 > block ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.NoError(t, err) set = ` Type: PoolL2 PoolTransferToEthAddr(1) A-B: 3 (1) PoolTransferToBJJ(1) A-C: 3 (1) ` _, err = tc.GeneratePoolL2Txs(set) require.NoError(t, err) // expect error, as FromIdx=B is still not created for TokenID=1 set = ` Type: PoolL2 PoolTransferToEthAddr(1) B-A: 3 (1) PoolTransferToBJJ(1) B-A: 3 (1) ` _, err = tc.GeneratePoolL2Txs(set) require.NotNil(t, err) } func TestGeneratePoolL2TxsFromInstructions(t *testing.T) { // Generate necessary L1 data set := ` Type: Blockchain AddToken(1) CreateAccountCoordinator(1) A CreateAccountDeposit(1) B: 7 > batchL1 > batchL1 ` tc := NewContext(common.RollupConstMaxL1UserTx) _, err := tc.GenerateBlocks(set) require.NoError(t, err) // Generate Pool txs using instructions instructionSet := []Instruction{} i := 0 a := big.NewInt(3) instructionSet = append(instructionSet, Instruction{ LineNum: i, // Literal: "PoolTransferToEthAddr(1) B-A: 3 (1)", Typ: common.TxTypeTransferToEthAddr, From: "B", To: "A", TokenID: 1, Amount: a, Fee: 1, }) i++ instructionSet = append(instructionSet, Instruction{ LineNum: i, // Literal: "PoolTransferToBJJ(1) B-A: 3 (1)", Typ: common.TxTypeTransferToBJJ, From: "B", To: "A", TokenID: 1, Amount: a, Fee: 1, }) txsFromInstructions, err := tc.GeneratePoolL2TxsFromInstructions(instructionSet) require.NoError(t, err) // Generate Pool txs using string tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.NoError(t, err) stringSet := ` Type: PoolL2 PoolTransferToEthAddr(1) B-A: 3 (1) PoolTransferToBJJ(1) B-A: 3 (1) ` txsFromString, err := tc.GeneratePoolL2Txs(stringSet) require.NoError(t, err) // Compare generated txs from instructions and string // timestamps will be different for i := 0; i < len(txsFromString); i++ { txsFromInstructions[i].Timestamp = txsFromString[i].Timestamp } assert.Equal(t, txsFromString, txsFromInstructions) } func TestGenerateErrors(t *testing.T) { // unregistered token set := `Type: Blockchain CreateAccountDeposit(1) A: 5 > batchL1 ` tc := NewContext(common.RollupConstMaxL1UserTx) _, err := tc.GenerateBlocks(set) assert.Equal(t, "Line 2: Can not process CreateAccountDeposit: TokenID 1 not registered, last registered TokenID: 0", err.Error()) // ensure AddToken sequentiality and not using 0 set = ` Type: Blockchain AddToken(0) ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.Equal(t, "Line 2: AddToken can not register TokenID 0", err.Error()) set = ` Type: Blockchain AddToken(2) ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.Equal(t, "Line 2: AddToken TokenID should be sequential, expected TokenID: 1, defined TokenID: 2", err.Error()) set = ` Type: Blockchain AddToken(1) AddToken(2) AddToken(3) AddToken(5) ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.Equal(t, "Line 5: AddToken TokenID should be sequential, expected TokenID: 4, defined TokenID: 5", err.Error()) // check transactions when account is not created yet set = ` Type: Blockchain AddToken(1) CreateAccountDeposit(1) A: 10 > batchL1 CreateAccountDeposit(1) B Transfer(1) A-B: 6 (1) > batch ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.Equal(t, "Line 5: CreateAccountDeposit(1)BTransfer(1) A-B: 6 (1)\n, err: Expected ':', found 'Transfer'", err.Error()) set = ` Type: Blockchain AddToken(1) CreateAccountDeposit(1) A: 10 > batchL1 CreateAccountCoordinator(1) B > batchL1 > batch Transfer(1) A-B: 6 (1) > batch ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.NoError(t, err) // check nonces set = ` Type: Blockchain AddToken(1) CreateAccountDeposit(1) A: 10 > batchL1 CreateAccountCoordinator(1) B > batchL1 Transfer(1) A-B: 6 (1) Transfer(1) A-B: 6 (1) // on purpose this is moving more money that what it has in the account, Til should not fail Transfer(1) B-A: 6 (1) Exit(1) A: 3 (1) > batch ` tc = NewContext(common.RollupConstMaxL1UserTx) _, err = tc.GenerateBlocks(set) require.NoError(t, err) assert.Equal(t, common.Nonce(3), tc.Users["A"].Accounts[common.TokenID(1)].Nonce) assert.Equal(t, common.Idx(256), tc.Users["A"].Accounts[common.TokenID(1)].Idx) assert.Equal(t, common.Nonce(1), tc.Users["B"].Accounts[common.TokenID(1)].Nonce) assert.Equal(t, common.Idx(257), tc.Users["B"].Accounts[common.TokenID(1)].Idx) } func TestGenerateFromInstructions(t *testing.T) { // Generate block from instructions setInst := []Instruction{} i := 0 setInst = append(setInst, Instruction{ LineNum: i, // Literal: "AddToken(1)", Typ: TypeAddToken, TokenID: 1, }) i++ da := big.NewInt(10) setInst = append(setInst, Instruction{ LineNum: i, // Literal: "CreateAccountDeposit(1) A: 10", Typ: common.TxTypeCreateAccountDeposit, From: "A", TokenID: 1, DepositAmount: da, }) i++ setInst = append(setInst, Instruction{ LineNum: i, // Literal: "> batchL1", Typ: TypeNewBatchL1, }) i++ setInst = append(setInst, Instruction{ LineNum: i, // Literal: "CreateAccountCoordinator(1) B", Typ: TxTypeCreateAccountDepositCoordinator, From: "B", TokenID: 1, }) i++ setInst = append(setInst, Instruction{ LineNum: i, // Literal: "> batchL1", Typ: TypeNewBatchL1, }) i++ a := big.NewInt(6) setInst = append(setInst, Instruction{ LineNum: i, // 5 // Literal: "Transfer(1) A-B: 6 (1)", Typ: common.TxTypeTransfer, From: "A", To: "B", TokenID: 1, Amount: a, Fee: 1, }) i++ setInst = append(setInst, Instruction{ LineNum: i, // Literal: "Transfer(1) A-B: 6 (1)", Typ: common.TxTypeTransfer, From: "A", To: "B", TokenID: 1, Amount: a, Fee: 1, }) i++ setInst = append(setInst, Instruction{ LineNum: i, // Literal: "Transfer(1) B-A: 6 (1)", Typ: common.TxTypeTransfer, From: "B", To: "A", TokenID: 1, Amount: a, Fee: 1, }) i++ a = big.NewInt(3) setInst = append(setInst, Instruction{ LineNum: i, // Literal: "Exit(1) A: 3 (1)", Typ: common.TxTypeExit, From: "A", TokenID: 1, Amount: a, Fee: 1, }) i++ setInst = append(setInst, Instruction{ LineNum: i, // Literal: "> batch", Typ: TypeNewBatch, }) setInst = append(setInst, Instruction{ LineNum: i, // Literal: "> block", Typ: TypeNewBlock, }) tc := NewContext(common.RollupConstMaxL1UserTx) blockFromInstructions, err := tc.GenerateBlocksFromInstructions(setInst) require.NoError(t, err) // Generate block from string setString := ` Type: Blockchain AddToken(1) CreateAccountDeposit(1) A: 10 > batchL1 CreateAccountCoordinator(1) B > batchL1 Transfer(1) A-B: 6 (1) Transfer(1) A-B: 6 (1) // on purpose this is moving more money that what it has in the account, Til should not fail Transfer(1) B-A: 6 (1) Exit(1) A: 3 (1) > batch > block ` tc = NewContext(common.RollupConstMaxL1UserTx) blockFromString, err := tc.GenerateBlocks(setString) require.NoError(t, err) // Generated data should be equivalent, except for Eth Addrs and BJJs for i, strBatch := range blockFromString[0].Rollup.Batches { // instBatch := blockFromInstructions[0].Rollup.Batches[i] for j := 0; j < len(strBatch.L1CoordinatorTxs); j++ { blockFromInstructions[0].Rollup.Batches[i].L1CoordinatorTxs[j].FromEthAddr = blockFromString[0].Rollup.Batches[i].L1CoordinatorTxs[j].FromEthAddr blockFromInstructions[0].Rollup.Batches[i].L1CoordinatorTxs[j].FromBJJ = blockFromString[0].Rollup.Batches[i].L1CoordinatorTxs[j].FromBJJ } for j := 0; j < len(strBatch.L1UserTxs); j++ { blockFromInstructions[0].Rollup.Batches[i].L1UserTxs[j].FromEthAddr = blockFromString[0].Rollup.Batches[i].L1UserTxs[j].FromEthAddr blockFromInstructions[0].Rollup.Batches[i].L1UserTxs[j].FromBJJ = blockFromString[0].Rollup.Batches[i].L1UserTxs[j].FromBJJ } } for i := 0; i < len(blockFromString[0].Rollup.L1UserTxs); i++ { blockFromInstructions[0].Rollup.L1UserTxs[i].FromEthAddr = blockFromString[0].Rollup.L1UserTxs[i].FromEthAddr blockFromInstructions[0].Rollup.L1UserTxs[i].FromBJJ = blockFromString[0].Rollup.L1UserTxs[i].FromBJJ } assert.Equal(t, blockFromString, blockFromInstructions) }