From d71fa9371ca392cc2daadbc01d4e4c13d056a103 Mon Sep 17 00:00:00 2001
From: Niko Matsakis <niko@alum.mit.edu>
Date: Fri, 14 Jul 2017 19:11:21 -0400
Subject: [PATCH] fix yields-in-args test and add a reverse one

---
 .../generator/yield-in-args-rev.rs            | 24 +++++++++++++++++++
 .../compile-fail/generator/yield-in-args.rs   |  8 +++----
 2 files changed, 28 insertions(+), 4 deletions(-)
 create mode 100644 src/test/compile-fail/generator/yield-in-args-rev.rs

diff --git a/src/test/compile-fail/generator/yield-in-args-rev.rs b/src/test/compile-fail/generator/yield-in-args-rev.rs
new file mode 100644
index 000000000000..2fadbfeb9d88
--- /dev/null
+++ b/src/test/compile-fail/generator/yield-in-args-rev.rs
@@ -0,0 +1,24 @@
+// Copyright 2017 The Rust Project Developers. See the COPYRIGHT
+// file at the top-level directory of this distribution and at
+// http://rust-lang.org/COPYRIGHT.
+//
+// Licensed under the Apache License, Version 2.0 <LICENSE-APACHE or
+// http://www.apache.org/licenses/LICENSE-2.0> or the MIT license
+// <LICENSE-MIT or http://opensource.org/licenses/MIT>, at your
+// option. This file may not be copied, modified, or distributed
+// except according to those terms.
+
+#![feature(generators)]
+
+fn foo(_a: (), _b: &bool) {}
+
+// Some examples that probably *could* be accepted, but which we reject for now.
+
+fn bar() {
+	|| {
+		let b = true;
+		foo(yield, &b);
+	}; //~ ERROR `b` does not live long enough
+}
+
+fn main() { }
diff --git a/src/test/compile-fail/generator/yield-in-args.rs b/src/test/compile-fail/generator/yield-in-args.rs
index 02106662b18f..c719baa5e2a3 100644
--- a/src/test/compile-fail/generator/yield-in-args.rs
+++ b/src/test/compile-fail/generator/yield-in-args.rs
@@ -13,8 +13,8 @@
 fn foo(_b: &bool, _a: ()) {}
 
 fn main() {
-    || {
-        let b = true;
-        foo(&b, yield); // ERROR `b` does not live long enough
-    };
+	|| {
+		let b = true;
+		foo(&b, yield);
+	}; //~ ERROR `b` does not live long enough
 }