Transform the inner expression of yield expressions (#66)

Author: SabrinaJewson

async-stream-impl/src/lib.rs

@@ -114,6 +114,8 @@ impl VisitMut for Scrub<'_> {
             syn::Expr::Yield(yield_expr) => {
                 self.has_yielded = true;
 
+                syn::visit_mut::visit_expr_yield_mut(self, yield_expr);
+
                 let value_expr = yield_expr.expr.as_ref().unwrap_or(&self.unit);
 
                 // let ident = &self.yielder;

async-stream/tests/try_stream.rs

@@ -78,3 +78,10 @@ async fn multi_try() {
         values
     );
 }
+
+#[allow(unused)]
+fn issue_65() -> impl Stream<Item = Result<u32, ()>> {
+    try_stream! {
+        yield Err(())?;
+    }
+}