Skip to content

Commit ded3266

Browse files
committed
Add tests of panicking collect shortfalls, like panic_fuse()
1 parent 7d99416 commit ded3266

File tree

1 file changed

+53
-0
lines changed

1 file changed

+53
-0
lines changed

src/iter/collect/test.rs

Lines changed: 53 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77

88
use super::Collect;
99
use iter::plumbing::*;
10+
use rayon_core::join;
1011

1112
/// Promises to produce 2 items, but then produces 3. Does not do any
1213
/// splits at all.
@@ -168,3 +169,55 @@ fn right_produces_fewer_items() {
168169
}
169170
collect.complete();
170171
}
172+
173+
// The left consumer panics and the right stops short, like `panic_fuse()`.
174+
// We should get the left panic without ever reaching `Collect::complete()`.
175+
#[test]
176+
#[should_panic(expected = "left consumer panic")]
177+
fn left_panics() {
178+
let mut v = vec![];
179+
let mut collect = Collect::new(&mut v, 4);
180+
{
181+
let consumer = collect.as_consumer();
182+
let (left_consumer, right_consumer, _) = consumer.split_at(2);
183+
join(
184+
|| {
185+
let mut left_folder = left_consumer.into_folder();
186+
left_folder = left_folder.consume(0);
187+
panic!("left consumer panic");
188+
},
189+
|| {
190+
let mut right_folder = right_consumer.into_folder();
191+
right_folder = right_folder.consume(2);
192+
right_folder.complete() // early return
193+
},
194+
);
195+
}
196+
collect.complete();
197+
}
198+
199+
// The right consumer panics and the left stops short, like `panic_fuse()`.
200+
// We should get the right panic without ever reaching `Collect::complete()`.
201+
#[test]
202+
#[should_panic(expected = "right consumer panic")]
203+
fn right_panics() {
204+
let mut v = vec![];
205+
let mut collect = Collect::new(&mut v, 4);
206+
{
207+
let consumer = collect.as_consumer();
208+
let (left_consumer, right_consumer, _) = consumer.split_at(2);
209+
join(
210+
|| {
211+
let mut left_folder = left_consumer.into_folder();
212+
left_folder = left_folder.consume(0);
213+
left_folder.complete() // early return
214+
},
215+
|| {
216+
let mut right_folder = right_consumer.into_folder();
217+
right_folder = right_folder.consume(2);
218+
panic!("right consumer panic");
219+
},
220+
);
221+
}
222+
collect.complete();
223+
}

0 commit comments

Comments
 (0)