summaryrefslogtreecommitdiffstats
path: root/third_party/rust/tokio-stream/tests/stream_iter.rs
diff options
context:
space:
mode:
Diffstat (limited to 'third_party/rust/tokio-stream/tests/stream_iter.rs')
-rw-r--r--third_party/rust/tokio-stream/tests/stream_iter.rs18
1 files changed, 18 insertions, 0 deletions
diff --git a/third_party/rust/tokio-stream/tests/stream_iter.rs b/third_party/rust/tokio-stream/tests/stream_iter.rs
new file mode 100644
index 0000000000..8b9ee3ce5b
--- /dev/null
+++ b/third_party/rust/tokio-stream/tests/stream_iter.rs
@@ -0,0 +1,18 @@
+use tokio_stream as stream;
+use tokio_test::task;
+
+use std::iter;
+
+#[tokio::test]
+async fn coop() {
+ let mut stream = task::spawn(stream::iter(iter::repeat(1)));
+
+ for _ in 0..10_000 {
+ if stream.poll_next().is_pending() {
+ assert!(stream.is_woken());
+ return;
+ }
+ }
+
+ panic!("did not yield");
+}