Skip to content

Commit

Permalink
add executed_transaction_traces() test
Browse files Browse the repository at this point in the history
  • Loading branch information
YaroShkvorets committed Mar 19, 2024
1 parent 6edcffc commit 6234c4a
Showing 1 changed file with 33 additions and 1 deletion.
34 changes: 33 additions & 1 deletion core/src/block.rs
Original file line number Diff line number Diff line change
Expand Up @@ -114,7 +114,7 @@ impl pb::Block {
#[cfg(test)]
mod tests {
use super::*;
use crate::block::pb::TransactionStatus::TransactionstatusExecuted;
use crate::block::pb::TransactionStatus::{TransactionstatusExecuted, TransactionstatusSoftfail};

// Helper function to create a test pb::Block instance
fn create_test_block(
Expand Down Expand Up @@ -206,6 +206,38 @@ mod tests {
assert_eq!(executed_traces, produced_traces);
}

#[test]
fn test_executed_transaction_traces_no_receipt() {
let executed_traces = vec![
pb::TransactionTrace {
id: String::from("trx1"),
receipt: Some(pb::TransactionReceiptHeader {
status: TransactionstatusExecuted as i32,
..Default::default()
}),
..Default::default()
},
pb::TransactionTrace {
id: String::from("trx2"),
receipt: Some(pb::TransactionReceiptHeader {
status: TransactionstatusSoftfail as i32,
..Default::default()
}),
..Default::default()
},
pb::TransactionTrace {
id: String::from("trx3"),
receipt: None,
..Default::default()
},
];
let block = create_test_block(false, executed_traces.clone(), vec![], 2, 0, 5, 0, 7, 0);

let produced_traces: Vec<_> = block.executed_transaction_traces().cloned().collect();
assert_eq!(executed_traces[0], produced_traces[0]);
assert_eq!(produced_traces.len(), 1);
}

#[test]
fn test_transaction_traces_count() {
let block = create_test_block(false, vec![], vec![], 8, 0, 5, 0, 7, 0);
Expand Down

0 comments on commit 6234c4a

Please sign in to comment.