rust/tests/ui/coroutine/issue-69039.rs
Oli Scherer aef0f4024a Error on using yield without also using #[coroutine] on the closure
And suggest adding the `#[coroutine]` to the closure
2024-04-24 08:05:29 +00:00

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())
);
}