summaryrefslogtreecommitdiffstats
path: root/src/test/ui/generator/issue-69039.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/test/ui/generator/issue-69039.rs')
-rw-r--r--src/test/ui/generator/issue-69039.rs34
1 files changed, 34 insertions, 0 deletions
diff --git a/src/test/ui/generator/issue-69039.rs b/src/test/ui/generator/issue-69039.rs
new file mode 100644
index 000000000..ccc141860
--- /dev/null
+++ b/src/test/ui/generator/issue-69039.rs
@@ -0,0 +1,34 @@
+// 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())
+ );
+}