From 698f8c2f01ea549d77d7dc3338a12e04c11057b9 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Wed, 17 Apr 2024 14:02:58 +0200
Subject: Adding upstream version 1.64.0+dfsg1.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 src/test/ui/generator/iterator-count.rs | 44 +++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 src/test/ui/generator/iterator-count.rs

(limited to 'src/test/ui/generator/iterator-count.rs')

diff --git a/src/test/ui/generator/iterator-count.rs b/src/test/ui/generator/iterator-count.rs
new file mode 100644
index 000000000..90eefe02f
--- /dev/null
+++ b/src/test/ui/generator/iterator-count.rs
@@ -0,0 +1,44 @@
+// run-pass
+
+#![feature(generators, generator_trait)]
+
+use std::marker::Unpin;
+use std::ops::{GeneratorState, Generator};
+use std::pin::Pin;
+
+struct W<T>(T);
+
+// This impl isn't safe in general, but the generator used in this test is movable
+// so it won't cause problems.
+impl<T: Generator<(), Return = ()> + Unpin> Iterator for W<T> {
+    type Item = T::Yield;
+
+    fn next(&mut self) -> Option<Self::Item> {
+        match Pin::new(&mut self.0).resume(()) {
+            GeneratorState::Complete(..) => None,
+            GeneratorState::Yielded(v) => Some(v),
+        }
+    }
+}
+
+fn test() -> impl Generator<(), Return=(), Yield=u8> + Unpin {
+    || {
+        for i in 1..6 {
+            yield i
+        }
+    }
+}
+
+fn main() {
+    let end = 11;
+
+    let closure_test = |start| {
+        move || {
+            for i in start..end {
+                yield i
+            }
+        }
+    };
+
+    assert!(W(test()).chain(W(closure_test(6))).eq(1..11));
+}
-- 
cgit v1.2.3