From 9fb8eb19a1dc5371adcc398f2899a84de96dfc48 Mon Sep 17 00:00:00 2001
From: Rostyslav Tyshko <tyshko.rostyslav@gmail.com>
Date: Wed, 16 Oct 2024 04:32:17 +0200
Subject: [PATCH] add test_pop_size

---
 mempool/src/lib.rs | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/mempool/src/lib.rs b/mempool/src/lib.rs
index bbc48ec..ce9f7e2 100644
--- a/mempool/src/lib.rs
+++ b/mempool/src/lib.rs
@@ -125,6 +125,19 @@ mod tests {
         assert_eq!(item, Some(&test_item_with_id(1)));
     }
 
+    #[test]
+    fn test_pop_size() {
+        let mut pool = MemPool::new();
+        pool.push_item(test_item_with_id(1));
+        pool.push_item(test_item_with_id(2));
+        pool.push_item(test_item_with_id(3));
+
+        let items = pool.pop_size(2);
+        assert_eq!(items, vec![test_item_with_id(1), test_item_with_id(2)]);
+        assert_eq!(pool.len(), 1);
+    }
+
+
     #[test]
     fn test_push_pop() {
         let mut mempool: MemPool<TestItem> = MemPool::new();