From 00fb0bd66f87dbe46d419cb068b730129e0f9ba4 Mon Sep 17 00:00:00 2001
From: Tyler Mandry <tmandry@gmail.com>
Date: Fri, 31 May 2019 18:49:22 -0700
Subject: [PATCH] Test StorageDead statements explicitly
---
.../mir-opt/generator-storage-dead-unwind.rs | 104 ++++++++++++++++++
1 file changed, 104 insertions(+)
create mode 100644 src/test/mir-opt/generator-storage-dead-unwind.rs
diff --git a/src/test/mir-opt/generator-storage-dead-unwind.rs b/src/test/mir-opt/generator-storage-dead-unwind.rs
new file mode 100644
index 0000000000000..883748a73e626
--- /dev/null
+++ b/src/test/mir-opt/generator-storage-dead-unwind.rs
@@ -0,0 +1,104 @@
+// Test that we generate StorageDead on unwind paths for generators.
+//
+// Basic block and local names can safely change, but the StorageDead statements
+// should not go away.
+
+#![feature(generators, generator_trait)]
+
+struct Foo(i32);
+
+impl Drop for Foo {
+ fn drop(&mut self) {}
+}
+
+struct Bar(i32);
+
+fn take<T>(_x: T) {}
+
+fn main() {
+ let _gen = || {
+ let a = Foo(5);
+ let b = Bar(6);
+ yield;
+ take(a);
+ take(b);
+ };
+}
+
+// END RUST SOURCE
+
+// START rustc.main-{{closure}}.StateTransform.before.mir
+// ...
+// let _2: Foo;
+// ...
+// let mut _7: Foo;
+// ...
+// let mut _9: Bar;
+// scope 1 {
+// let _3: Bar;
+// scope 2 {
+// }
+// }
+// bb0: {
+// StorageLive(_2);
+// _2 = Foo(const 5i32,);
+// StorageLive(_3);
+// _3 = Bar(const 6i32,);
+// ...
+// _1 = suspend(move _5) -> [resume: bb2, drop: bb4];
+// }
+// bb1 (cleanup): {
+// resume;
+// }
+// bb2: {
+// ...
+// StorageLive(_7);
+// _7 = move _2;
+// _6 = const take::<Foo>(move _7) -> [return: bb9, unwind: bb8];
+// }
+// bb3 (cleanup): {
+// StorageDead(_2);
+// drop(_1) -> bb1;
+// }
+// bb4: {
+// ...
+// StorageDead(_3);
+// drop(_2) -> [return: bb5, unwind: bb3];
+// }
+// bb5: {
+// StorageDead(_2);
+// drop(_1) -> [return: bb6, unwind: bb1];
+// }
+// bb6: {
+// generator_drop;
+// }
+// bb7 (cleanup): {
+// StorageDead(_3);
+// StorageDead(_2);
+// drop(_1) -> bb1;
+// }
+// bb8 (cleanup): {
+// StorageDead(_7);
+// goto -> bb7;
+// }
+// bb9: {
+// StorageDead(_7);
+// StorageLive(_9);
+// _9 = move _3;
+// _8 = const take::<Bar>(move _9) -> [return: bb10, unwind: bb11];
+// }
+// bb10: {
+// StorageDead(_9);
+// ...
+// StorageDead(_3);
+// StorageDead(_2);
+// drop(_1) -> [return: bb12, unwind: bb1];
+// }
+// bb11 (cleanup): {
+// StorageDead(_9);
+// goto -> bb7;
+// }
+// bb12: {
+// return;
+// }
+// END rustc.main-{{closure}}.StateTransform.before.mir