[rustc.git] / tests / run-make / coverage / generator.rs

#![feature(generators, generator_trait)]

use std::ops::{Generator, GeneratorState};
use std::pin::Pin;

// The following implementation of a function called from a `yield` statement
// (apparently requiring the Result and the `String` type or constructor)
// creates conditions where the `generator::StateTransform` MIR transform will
// drop all `Counter` `Coverage` statements from a MIR. `simplify.rs` has logic
// to handle this condition, and still report dead block coverage.
fn get_u32(val: bool) -> Result<u32, String> {
    if val { Ok(1) } else { Err(String::from("some error")) }
}

fn main() {
    let is_true = std::env::args().len() == 1;
    let mut generator = || {
        yield get_u32(is_true);
        return "foo";
    };

    match Pin::new(&mut generator).resume(()) {
        GeneratorState::Yielded(Ok(1)) => {}
        _ => panic!("unexpected return from resume"),
    }
    match Pin::new(&mut generator).resume(()) {
        GeneratorState::Complete("foo") => {}
        _ => panic!("unexpected return from resume"),
    }
}
Commit	Line	Data
17df50a5 XL	1	#![feature(generators, generator_trait)]
	2
	3	use std::ops::{Generator, GeneratorState};
	4	use std::pin::Pin;
	5
	6	// The following implementation of a function called from a `yield` statement
	7	// (apparently requiring the Result and the `String` type or constructor)
	8	// creates conditions where the `generator::StateTransform` MIR transform will
	9	// drop all `Counter` `Coverage` statements from a MIR. `simplify.rs` has logic
	10	// to handle this condition, and still report dead block coverage.
	11	fn get_u32(val: bool) -> Result<u32, String> {
	12	if val { Ok(1) } else { Err(String::from("some error")) }
	13	}
	14
	15	fn main() {
	16	let is_true = std::env::args().len() == 1;
	17	let mut generator = \|\| {
	18	yield get_u32(is_true);
	19	return "foo";
	20	};
	21
	22	match Pin::new(&mut generator).resume(()) {
	23	GeneratorState::Yielded(Ok(1)) => {}
	24	_ => panic!("unexpected return from resume"),
	25	}
	26	match Pin::new(&mut generator).resume(()) {
	27	GeneratorState::Complete("foo") => {}
	28	_ => panic!("unexpected return from resume"),
	29	}
	30	}