about summary refs log tree commit diff
path: root/tests/run-coverage/yield.rs
diff options
context:
space:
mode:
authorZalathar <Zalathar@users.noreply.github.com>2023-06-12 18:07:04 +1000
committerZalathar <Zalathar@users.noreply.github.com>2023-06-28 11:09:19 +1000
commite0625b4586c5a0f855a1157b09ae384f5de0ecf7 (patch)
treed6f6d80e1d6b9aaec8b6340986a3e22475775651 /tests/run-coverage/yield.rs
parent22e119bbacd3aa11db04d84947b4fb2c5ccb0435 (diff)
downloadrust-e0625b4586c5a0f855a1157b09ae384f5de0ecf7.tar.gz
rust-e0625b4586c5a0f855a1157b09ae384f5de0ecf7.zip
Migrate most of the existing coverage tests over to `run-coverage`
Diffstat (limited to 'tests/run-coverage/yield.rs')
-rw-r--r--tests/run-coverage/yield.rs37
1 files changed, 37 insertions, 0 deletions
diff --git a/tests/run-coverage/yield.rs b/tests/run-coverage/yield.rs
new file mode 100644
index 00000000000..ff7616656ff
--- /dev/null
+++ b/tests/run-coverage/yield.rs
@@ -0,0 +1,37 @@
+#![feature(generators, generator_trait)]
+#![allow(unused_assignments)]
+
+use std::ops::{Generator, GeneratorState};
+use std::pin::Pin;
+
+fn main() {
+    let mut generator = || {
+        yield 1;
+        return "foo"
+    };
+
+    match Pin::new(&mut generator).resume(()) {
+        GeneratorState::Yielded(1) => {}
+        _ => panic!("unexpected value from resume"),
+    }
+    match Pin::new(&mut generator).resume(()) {
+        GeneratorState::Complete("foo") => {}
+        _ => panic!("unexpected value from resume"),
+    }
+
+    let mut generator = || {
+        yield 1;
+        yield 2;
+        yield 3;
+        return "foo"
+    };
+
+    match Pin::new(&mut generator).resume(()) {
+        GeneratorState::Yielded(1) => {}
+        _ => panic!("unexpected value from resume"),
+    }
+    match Pin::new(&mut generator).resume(()) {
+        GeneratorState::Yielded(2) => {}
+        _ => panic!("unexpected value from resume"),
+    }
+}