From b642f961e7f92f8705bf212ac4815c1dec082575 Mon Sep 17 00:00:00 2001
From: renancloudwalk <53792026+renancloudwalk@users.noreply.github.com>
Date: Wed, 26 Jun 2024 13:39:14 -0300
Subject: [PATCH] Test append entries (#1252)

* chore: test append entries as a whole

* lint
---
 src/eth/consensus/mod.rs                      |  1 +
 src/eth/consensus/tests/factories.rs          |  6 +-
 src/eth/consensus/tests/test_simple_blocks.rs | 72 +++++++++++++++++++
 3 files changed, 76 insertions(+), 3 deletions(-)
 create mode 100644 src/eth/consensus/tests/test_simple_blocks.rs

diff --git a/src/eth/consensus/mod.rs b/src/eth/consensus/mod.rs
index 4bc25c08c..b96d157cd 100644
--- a/src/eth/consensus/mod.rs
+++ b/src/eth/consensus/mod.rs
@@ -736,6 +736,7 @@ impl Consensus {
 mod tests {
     use super::*;
     pub mod factories;
+    mod test_simple_blocks;
 
     #[test]
     fn test_peer_address_from_string_valid_http() {
diff --git a/src/eth/consensus/tests/factories.rs b/src/eth/consensus/tests/factories.rs
index a792bae1d..e5a51b093 100644
--- a/src/eth/consensus/tests/factories.rs
+++ b/src/eth/consensus/tests/factories.rs
@@ -22,7 +22,7 @@ use crate::eth::consensus::Role;
 use crate::eth::consensus::TransactionExecutionEntry;
 use crate::eth::storage::StratusStorage;
 
-pub fn create_mock_block_entry() -> BlockEntry {
+pub fn create_mock_block_entry(transaction_hashes: Vec<String>) -> BlockEntry {
     BlockEntry {
         number: rand::thread_rng().gen(),
         hash: H256::random().to_string(),
@@ -42,7 +42,7 @@ pub fn create_mock_block_entry() -> BlockEntry {
         timestamp: rand::thread_rng().gen(),
         bloom: H256::random().to_string(),
         author: H256::random().to_string(),
-        transaction_hashes: vec![H256::random().to_string()],
+        transaction_hashes,
     }
 }
 
@@ -83,7 +83,7 @@ pub fn create_mock_transaction_execution_entry() -> TransactionExecutionEntry {
 }
 
 pub fn create_mock_log_entry_data_block() -> LogEntryData {
-    LogEntryData::BlockEntry(create_mock_block_entry())
+    LogEntryData::BlockEntry(create_mock_block_entry(vec![]))
 }
 
 pub fn create_mock_log_entry_data_transactions() -> LogEntryData {
diff --git a/src/eth/consensus/tests/test_simple_blocks.rs b/src/eth/consensus/tests/test_simple_blocks.rs
new file mode 100644
index 000000000..880943428
--- /dev/null
+++ b/src/eth/consensus/tests/test_simple_blocks.rs
@@ -0,0 +1,72 @@
+use std::sync::Arc;
+
+use tokio::sync::Mutex;
+use tonic::Request;
+
+use super::factories::create_follower_consensus_with_leader;
+use super::factories::create_mock_block_entry;
+use super::factories::create_mock_transaction_execution_entry;
+use crate::eth::consensus::append_entry::append_entry_service_server::AppendEntryService;
+use crate::eth::consensus::append_entry::AppendBlockCommitRequest;
+use crate::eth::consensus::append_entry::AppendTransactionExecutionsRequest;
+use crate::eth::consensus::AppendEntryServiceImpl;
+use crate::eth::consensus::Role;
+use crate::eth::consensus::TransactionExecutionEntry;
+
+#[tokio::test]
+async fn test_append_entries_transaction_executions_and_block() {
+    let consensus = create_follower_consensus_with_leader().await;
+    let service = AppendEntryServiceImpl {
+        consensus: Mutex::new(Arc::clone(&consensus)),
+    };
+
+    let leader_id = {
+        let peers = consensus.peers.read().await;
+        let (leader_address, _) = peers.iter().find(|&(_, (peer, _))| peer.role == Role::Leader).expect("Leader peer not found");
+        leader_address.to_string()
+    };
+    // Create 3 requests with 3 transactions each, totaling 9 transactions
+    let total_requests = 3;
+    let transactions_per_request = 3;
+    let mut all_executions = Vec::new();
+    let mut prev_log_index = 0;
+
+    for _ in 0..total_requests {
+        let executions: Vec<TransactionExecutionEntry> = (0..transactions_per_request).map(|_| create_mock_transaction_execution_entry()).collect();
+
+        // Store all executions for later verification
+        all_executions.extend(executions.clone());
+
+        // Convert to append entries request format
+        let request = Request::new(AppendTransactionExecutionsRequest {
+            term: 1,
+            leader_id: leader_id.clone(),
+            prev_log_index,
+            prev_log_term: 1, // assuming same term for simplicity
+            executions: executions.clone(),
+        });
+
+        // Append transactions
+        let response = service.append_transaction_executions(request).await;
+        assert!(response.is_ok());
+
+        // Update prev_log_index for the next request
+        prev_log_index += 1;
+    }
+
+    // Create and append block with transaction hashes
+    let transaction_hashes: Vec<String> = all_executions.iter().map(|e| e.hash.clone()).collect();
+
+    let block_entry = create_mock_block_entry(transaction_hashes.clone());
+
+    let block_request = Request::new(AppendBlockCommitRequest {
+        term: 1,
+        leader_id: leader_id.clone(),
+        prev_log_index,
+        prev_log_term: 1,
+        block_entry: Some(block_entry.clone()),
+    });
+
+    let block_response = service.append_block_commit(block_request).await;
+    assert!(block_response.is_ok());
+}