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 = MemPool::new();