blob: ccc141860aa5f908683fb4d3ada63de8a96c1ff6 [file] [log] [blame]
// run-pass
#![feature(generators, generator_trait)]
use std::ops::{Generator, GeneratorState};
fn mkstr(my_name: String, my_mood: String) -> String {
format!("{} is {}", my_name.trim(), my_mood.trim())
}
fn my_scenario() -> impl Generator<String, Yield = &'static str, Return = String> {
|_arg: String| {
let my_name = yield "What is your name?";
let my_mood = yield "How are you feeling?";
mkstr(my_name, my_mood)
}
}
fn main() {
let mut my_session = Box::pin(my_scenario());
assert_eq!(
my_session.as_mut().resume("_arg".to_string()),
GeneratorState::Yielded("What is your name?")
);
assert_eq!(
my_session.as_mut().resume("Your Name".to_string()),
GeneratorState::Yielded("How are you feeling?")
);
assert_eq!(
my_session.as_mut().resume("Sensory Organs".to_string()),
GeneratorState::Complete("Your Name is Sensory Organs".to_string())
);
}