fix corner case in `reduce_vars` (#4636)

2 files changed, 67 insertions, 1 deletions

diff --git a/lib/compress.js b/lib/compress.js
index b35fc6a5..a0e018e3 100644
--- a/lib/compress.js
+++ b/lib/compress.js
@@ -9836,7 +9836,8 @@ merge(Compressor.prototype, {
                                 }
                             } while (scope = scope.parent_scope);
                         }
-                    } else if (fixed.name && fixed.name.name == "await" && is_async(fixed)) {
+                    } else if (fixed.name && (fixed.name.name == "await" && is_async(fixed)
+                        || fixed.name.name == "yield" && is_generator(fixed))) {
                         single_use = false;
                     }
                     if (single_use) fixed.parent_scope = self.scope;
diff --git a/test/compress/yields.js b/test/compress/yields.js
index 5a6948ed..cd2518d6 100644
--- a/test/compress/yields.js
+++ b/test/compress/yields.js
@@ -255,6 +255,71 @@ collapse_property_lambda: {
     node_version: ">=4"
 }
 
+defun_name: {
+    input: {
+        function* yield() {
+            console.log("PASS");
+        }
+        yield().next();
+    }
+    expect: {
+        function* yield() {
+            console.log("PASS");
+        }
+        yield().next();
+    }
+    expect_stdout: "PASS"
+    node_version: ">=4"
+}
+
+drop_fname: {
+    rename = true
+    options = {
+        reduce_vars: true,
+        toplevel: true,
+        unused: true,
+    }
+    mangle = {
+        toplevel: true,
+    }
+    input: {
+        function* yield() {
+            console.log("PASS");
+        }
+        yield().next();
+    }
+    expect: {
+        (function*() {
+            console.log("PASS");
+        })().next();
+    }
+    expect_stdout: "PASS"
+    node_version: ">=4"
+}
+
+keep_fname: {
+    options = {
+        keep_fnames: true,
+        reduce_vars: true,
+        toplevel: true,
+        unused: true,
+    }
+    input: {
+        function* yield() {
+            console.log("PASS");
+        }
+        yield().next();
+    }
+    expect: {
+        function* yield() {
+            console.log("PASS");
+        }
+        yield().next();
+    }
+    expect_stdout: "PASS"
+    node_version: ">=4"
+}
+
 evaluate: {
     options = {
         evaluate: true,