From 19cc90772bbd7933f5badd084bc8c80847672cd9 Mon Sep 17 00:00:00 2001
From: Marcos Nicolau <marcosnicolau@lambdaclass.com>
Date: Wed, 28 Aug 2024 10:08:31 -0300
Subject: [PATCH] Add parallel execution test

---
 .../multivm/src/versions/era_vm/tests/mod.rs  |  1 +
 .../era_vm/tests/parallel_execution.rs        | 73 +++++++++++++++++++
 2 files changed, 74 insertions(+)
 create mode 100644 core/lib/multivm/src/versions/era_vm/tests/parallel_execution.rs

diff --git a/core/lib/multivm/src/versions/era_vm/tests/mod.rs b/core/lib/multivm/src/versions/era_vm/tests/mod.rs
index 801b6450ef8..a248eda064d 100644
--- a/core/lib/multivm/src/versions/era_vm/tests/mod.rs
+++ b/core/lib/multivm/src/versions/era_vm/tests/mod.rs
@@ -11,6 +11,7 @@ mod is_write_initial;
 mod l1_tx_execution;
 mod l2_blocks;
 mod nonce_holder;
+mod parallel_execution;
 mod precompiles;
 mod refunds;
 mod require_eip712;
diff --git a/core/lib/multivm/src/versions/era_vm/tests/parallel_execution.rs b/core/lib/multivm/src/versions/era_vm/tests/parallel_execution.rs
new file mode 100644
index 00000000000..876bea4fb0d
--- /dev/null
+++ b/core/lib/multivm/src/versions/era_vm/tests/parallel_execution.rs
@@ -0,0 +1,73 @@
+use zksync_types::Transaction;
+
+use super::tester::VmTester;
+use crate::{
+    era_vm::tests::tester::{TxType, VmTesterBuilder},
+    interface::{VmExecutionMode, VmInterface},
+};
+
+fn prepare_test() -> (VmTester, [Transaction; 3]) {
+    let mut vm_tester = VmTesterBuilder::new()
+        .with_empty_in_memory_storage()
+        .with_deployer()
+        .with_random_rich_accounts(1)
+        .build();
+
+    vm_tester.deploy_test_contract();
+
+    let account = &mut vm_tester.rich_accounts[0];
+
+    let txs = [
+        account.get_test_contract_transaction(
+            vm_tester.test_contract.unwrap(),
+            false,
+            Default::default(),
+            false,
+            TxType::L1 { serial_id: 1 },
+        ),
+        account.get_test_contract_transaction(
+            vm_tester.test_contract.unwrap(),
+            true,
+            Default::default(),
+            false,
+            TxType::L1 { serial_id: 1 },
+        ),
+        account.get_test_contract_transaction(
+            vm_tester.test_contract.unwrap(),
+            false,
+            Default::default(),
+            false,
+            TxType::L1 { serial_id: 1 },
+        ),
+    ];
+
+    (vm_tester, txs)
+}
+
+#[test]
+fn parallel_execution() {
+    let normal_execution = {
+        let (mut vm, txs) = prepare_test();
+        let vm = &mut vm.vm;
+        for tx in &txs {
+            vm.push_transaction_inner(tx.clone(), 0, true);
+        }
+        vm.execute(VmExecutionMode::Batch)
+    };
+
+    let parallel_execution = {
+        let (mut vm, txs) = prepare_test();
+        let vm = &mut vm.vm;
+        for tx in txs {
+            vm.push_parallel_transaction(tx, 0, true);
+        }
+        vm.execute_parallel()
+    };
+
+    // we don't assert if statistics are equal since that would require
+    // sharing the gas in parallel,
+    // assert!(1 == 2);
+    assert_eq!(normal_execution.logs, parallel_execution.logs);
+    // assert_eq!(normal_execution.result, parallel_execution.result);
+    // assert_eq!(normal_execution.refunds, parallel_execution.refunds);
+}