github/reth
1
1
Fork 0
mirror of https://github.com/paradigmxyz/reth.git synced 2026-01-26 15:48:13 -05:00
Code Issues Packages Projects Releases Wiki Activity

add remove logic in integration test of txpool listener (#15285)

Browse Source
This commit is contained in:
int88
2025-03-26 16:55:05 +08:00
committed by GitHub
parent 790b49e00a
commit abd4470971
1 changed files with 13 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
crates/transaction-pool/tests/it/listeners.rs
View File

@@ -2,8 +2,8 @@ use assert_matches::assert_matches;
use reth_transaction_pool::{
noop::MockTransactionValidator,
test_utils::{MockTransactionFactory, TestPoolBuilder},
FullTransactionEvent, TransactionEvent, TransactionListenerKind, TransactionOrigin,
TransactionPool,
FullTransactionEvent, PoolTransaction, TransactionEvent, TransactionListenerKind,
TransactionOrigin, TransactionPool,
};
use std::{future::poll_fn, task::Poll};
use tokio_stream::StreamExt;
@@ -21,6 +21,11 @@ async fn txpool_listener_by_hash() {
let mut events = result.unwrap();
assert_matches!(events.next().await, Some(TransactionEvent::Pending));
let removed_txs = txpool.remove_transactions(vec![*transaction.transaction.hash()]);
assert_eq!(transaction.transaction.hash(), removed_txs[0].transaction.hash());
assert_matches!(events.next().await, Some(TransactionEvent::Discarded));
}
#[tokio::test(flavor = "multi_thread")]
@@ -39,6 +44,12 @@ async fn txpool_listener_all() {
all_tx_events.next().await,
Some(FullTransactionEvent::Pending(hash)) if hash == *transaction.transaction.get_hash()
);
let removed_txs = txpool.remove_transactions(vec![*transaction.transaction.hash()]);
assert_eq!(transaction.transaction.hash(), removed_txs[0].transaction.hash());
assert_matches!(all_tx_events.next().await, Some(FullTransactionEvent::Discarded(hash)) if hash == *transaction.transaction.get_hash());
}
#[tokio::test(flavor = "multi_thread")]