summaryrefslogtreecommitdiffstats
path: root/tests/ui/generator/drop-and-replace.rs
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-17 12:18:58 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2024-04-17 12:18:58 +0000
commita4b7ed7a42c716ab9f05e351f003d589124fd55d (patch)
treeb620cd3f223850b28716e474e80c58059dca5dd4 /tests/ui/generator/drop-and-replace.rs
parentAdding upstream version 1.67.1+dfsg1. (diff)
downloadrustc-a4b7ed7a42c716ab9f05e351f003d589124fd55d.tar.xz
rustc-a4b7ed7a42c716ab9f05e351f003d589124fd55d.zip
Adding upstream version 1.68.2+dfsg1.upstream/1.68.2+dfsg1
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'tests/ui/generator/drop-and-replace.rs')
-rw-r--r--tests/ui/generator/drop-and-replace.rs45
1 files changed, 45 insertions, 0 deletions
diff --git a/tests/ui/generator/drop-and-replace.rs b/tests/ui/generator/drop-and-replace.rs
new file mode 100644
index 000000000..a9a50a122
--- /dev/null
+++ b/tests/ui/generator/drop-and-replace.rs
@@ -0,0 +1,45 @@
+// run-pass
+// Regression test for incorrect DropAndReplace behavior introduced in #60840
+// and fixed in #61373. When combined with the optimization implemented in
+// #60187, this produced incorrect code for generators when a saved local was
+// re-assigned.
+
+#![feature(generators, generator_trait)]
+
+use std::ops::{Generator, GeneratorState};
+use std::pin::Pin;
+
+#[derive(Debug, PartialEq)]
+struct Foo(i32);
+
+impl Drop for Foo {
+ fn drop(&mut self) { }
+}
+
+fn main() {
+ let mut a = || {
+ let mut x = Foo(4);
+ yield;
+ assert_eq!(x.0, 4);
+
+ // At one point this tricked our dataflow analysis into thinking `x` was
+ // StorageDead after the assignment.
+ x = Foo(5);
+ assert_eq!(x.0, 5);
+
+ {
+ let y = Foo(6);
+ yield;
+ assert_eq!(y.0, 6);
+ }
+
+ assert_eq!(x.0, 5);
+ };
+
+ loop {
+ match Pin::new(&mut a).resume(()) {
+ GeneratorState::Complete(()) => break,
+ _ => (),
+ }
+ }
+}