|
7 | 7 |
|
8 | 8 | use super::Collect;
|
9 | 9 | use iter::plumbing::*;
|
| 10 | +use rayon_core::join; |
10 | 11 |
|
11 | 12 | /// Promises to produce 2 items, but then produces 3. Does not do any
|
12 | 13 | /// splits at all.
|
@@ -168,3 +169,55 @@ fn right_produces_fewer_items() {
|
168 | 169 | }
|
169 | 170 | collect.complete();
|
170 | 171 | }
|
| 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