mirror of https://github.com/rust-lang/rust
36 lines
982 B
Rust
36 lines
982 B
Rust
//@ run-pass
|
|
|
|
#![feature(coroutines, coroutine_trait)]
|
|
|
|
use std::ops::{Coroutine, CoroutineState};
|
|
|
|
fn mkstr(my_name: String, my_mood: String) -> String {
|
|
format!("{} is {}", my_name.trim(), my_mood.trim())
|
|
}
|
|
|
|
fn my_scenario() -> impl Coroutine<String, Yield = &'static str, Return = String> {
|
|
#[coroutine]
|
|
|_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()),
|
|
CoroutineState::Yielded("What is your name?")
|
|
);
|
|
assert_eq!(
|
|
my_session.as_mut().resume("Your Name".to_string()),
|
|
CoroutineState::Yielded("How are you feeling?")
|
|
);
|
|
assert_eq!(
|
|
my_session.as_mut().resume("Sensory Organs".to_string()),
|
|
CoroutineState::Complete("Your Name is Sensory Organs".to_string())
|
|
);
|
|
}
|