diff --git a/src/tests.rs b/src/tests.rs index 5006538..49b9be3 100644 --- a/src/tests.rs +++ b/src/tests.rs @@ -912,16 +912,18 @@ fn queue() { let n = rng.gen_range(1..20); for _ in 0..n { - if rng.gen_bool(0.6) || root.length(transaction) == 0 { - let how_many = rng.gen_range(1..20); + let how_many = rng.gen_range(1..20); + if rng.gen_bool(0.6) || root.length(transaction) < how_many { let elements = (i..i + how_many).map(U64::from).collect::>(); root = root.enqueue_many(transaction, &elements); i += how_many; } else { let res; - (root, res) = root.dequeue(transaction).unwrap(); - assert_eq!(res.get(), j); - j += 1; + (root, res) = root.dequeue_many(transaction, how_many).unwrap(); + for x in res { + assert_eq!(x.get(), j); + j += 1; + } } }