test(mempool): test get_txs with holes, single account (#474)

starkware-libs · Jul 16, 2024 · 60a248b · 60a248b
1 parent 15f5564
commit 60a248b
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/crates/mempool/src/mempool_test.rs b/crates/mempool/src/mempool_test.rs
@@ -316,3 +316,22 @@ fn test_add_tx_sequential_nonces(mut mempool: Mempool) {
 
     expected_mempool_state.assert_eq_mempool_state(&mempool);
 }
+
+#[rstest]
+fn test_get_txs_with_holes_single_account() {
+    // Setup.
+    let input_nonce_1 = add_tx_input!(tx_hash: 0, tx_nonce: 1_u8, account_nonce: 0_u8);
+
+    let pool_txs = [input_nonce_1.tx];
+    let queue_txs = [];
+    let mut mempool: Mempool = MempoolState::new(pool_txs.clone(), queue_txs).into();
+
+    // Test.
+    let txs = mempool.get_txs(1).unwrap();
+
+    // Assert.
+    assert_eq!(txs, &[]);
+
+    let mempool_state = MempoolState::new(pool_txs, queue_txs);
+    mempool_state.assert_eq_mempool_state(&mempool);
+}