test(mempool): refactor test_add_tx_with_duplicate_tx to use mempool state

Author: ayeletstarkware

crates/mempool/src/mempool_test.rs

@@ -223,17 +223,23 @@ fn test_new_with_duplicate_tx() {
 
 #[rstest]
 fn test_add_tx_with_duplicate_tx(mut mempool: Mempool) {
+    // Setup
     let input = add_tx_input!(tip: 50, tx_hash: Felt::ONE);
-    let same_input = input.clone();
+    let duplicate_input = input.clone();
 
+    // Test: assert that the duplicate tx is not added to the mempool.
     add_tx(&mut mempool, &input);
-
     assert_matches!(
-        mempool.add_tx(same_input.clone()),
+        mempool.add_tx(duplicate_input),
         Err(MempoolError::DuplicateTransaction { .. })
     );
-    // Assert that the original tx remains in the pool after the failed attempt.
-    assert_eq_mempool_queue(&mempool, &[same_input.tx])
+
+    // Assert that the original tx remains in Mempool after the failed attempt.
+    let expected_queue_txs = [TransactionReference::new(&input.tx)];
+    let expected_pool_txs = [input.tx];
+    let expected_mempool_state = MempoolState::new(expected_pool_txs, expected_queue_txs);
+
+    expected_mempool_state.assert_eq_mempool_state(&mempool);
 }
 
 #[rstest]