about summary refs log tree commit diff
path: root/tests/coverage/yield.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tests/coverage/yield.rs')
-rw-r--r--tests/coverage/yield.rs37
1 files changed, 37 insertions, 0 deletions
diff --git a/tests/coverage/yield.rs b/tests/coverage/yield.rs
new file mode 100644
index 00000000000..b7e2ba31b59
--- /dev/null
+++ b/tests/coverage/yield.rs
@@ -0,0 +1,37 @@
+#![feature(coroutines, coroutine_trait)]
+#![allow(unused_assignments)]
+
+use std::ops::{Coroutine, CoroutineState};
+use std::pin::Pin;
+
+fn main() {
+    let mut coroutine = || {
+        yield 1;
+        return "foo";
+    };
+
+    match Pin::new(&mut coroutine).resume(()) {
+        CoroutineState::Yielded(1) => {}
+        _ => panic!("unexpected value from resume"),
+    }
+    match Pin::new(&mut coroutine).resume(()) {
+        CoroutineState::Complete("foo") => {}
+        _ => panic!("unexpected value from resume"),
+    }
+
+    let mut coroutine = || {
+        yield 1;
+        yield 2;
+        yield 3;
+        return "foo";
+    };
+
+    match Pin::new(&mut coroutine).resume(()) {
+        CoroutineState::Yielded(1) => {}
+        _ => panic!("unexpected value from resume"),
+    }
+    match Pin::new(&mut coroutine).resume(()) {
+        CoroutineState::Yielded(2) => {}
+        _ => panic!("unexpected value from resume"),
+    }
+}