feat(mempool): add from_iter func to TransactionQueue

starkware-libs · Jul 10, 2024 · 282dcf3 · 282dcf3
1 parent c025838
commit 282dcf3
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/crates/mempool/src/mempool_test.rs b/crates/mempool/src/mempool_test.rs
@@ -12,6 +12,7 @@ use starknet_types_core::felt::Felt;
 
 use crate::mempool::{Mempool, MempoolInput, TransactionReference};
 use crate::transaction_pool::TransactionPool;
+use crate::transaction_queue::TransactionQueue;
 
 impl FromIterator<ThinTransaction> for TransactionPool {
     fn from_iter<T: IntoIterator<Item = ThinTransaction>>(txs: T) -> Self {
@@ -23,6 +24,16 @@ impl FromIterator<ThinTransaction> for TransactionPool {
     }
 }
 
+impl FromIterator<TransactionReference> for TransactionQueue {
+    fn from_iter<T: IntoIterator<Item = TransactionReference>>(txs: T) -> Self {
+        let mut queue = Self::default();
+        for tx in txs {
+            queue.insert(tx);
+        }
+        queue
+    }
+}
+
 #[track_caller]
 fn add_tx(mempool: &mut Mempool, input: &MempoolInput) {
     assert_eq!(mempool.add_tx(input.clone()), Ok(()));