about summary refs log tree commit diff
path: root/tests/coverage/coroutine.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tests/coverage/coroutine.rs')
-rw-r--r--tests/coverage/coroutine.rs30
1 files changed, 30 insertions, 0 deletions
diff --git a/tests/coverage/coroutine.rs b/tests/coverage/coroutine.rs
new file mode 100644
index 00000000000..86d19af6f4f
--- /dev/null
+++ b/tests/coverage/coroutine.rs
@@ -0,0 +1,30 @@
+#![feature(coroutines, coroutine_trait)]
+
+use std::ops::{Coroutine, CoroutineState};
+use std::pin::Pin;
+
+// The following implementation of a function called from a `yield` statement
+// (apparently requiring the Result and the `String` type or constructor)
+// creates conditions where the `coroutine::StateTransform` MIR transform will
+// drop all `Counter` `Coverage` statements from a MIR. `simplify.rs` has logic
+// to handle this condition, and still report dead block coverage.
+fn get_u32(val: bool) -> Result<u32, String> {
+    if val { Ok(1) } else { Err(String::from("some error")) }
+}
+
+fn main() {
+    let is_true = std::env::args().len() == 1;
+    let mut coroutine = || {
+        yield get_u32(is_true);
+        return "foo";
+    };
+
+    match Pin::new(&mut coroutine).resume(()) {
+        CoroutineState::Yielded(Ok(1)) => {}
+        _ => panic!("unexpected return from resume"),
+    }
+    match Pin::new(&mut coroutine).resume(()) {
+        CoroutineState::Complete("foo") => {}
+        _ => panic!("unexpected return from resume"),
+    }
+}