From 9ddb75c28b5b0a356afb22ef123c4eff9836afb9 Mon Sep 17 00:00:00 2001
From: alexlamsl <alexlamsl@gmail.com>
Date: Fri, 24 Nov 2017 13:17:43 +0800
Subject: [PATCH] extend `function`ality add tests

---
 lib/compress.js              | 12 ++---
 test/compress/hoist_props.js | 85 ++++++++++++++++++++++++++++++++++++
 2 files changed, 91 insertions(+), 6 deletions(-)

diff --git a/lib/compress.js b/lib/compress.js
index 15a2f656..744a1ea5 100644
--- a/lib/compress.js
+++ b/lib/compress.js
@@ -333,7 +333,7 @@ merge(Compressor.prototype, {
                             }
                         }
                     }
-                    mark_escaped(d, node, value, 0);
+                    mark_escaped(d, node.scope, node, value, 0);
                 }
                 if (node instanceof AST_SymbolCatch) {
                     node.definition().fixed = false;
@@ -623,11 +623,11 @@ merge(Compressor.prototype, {
             }
         }
 
-        function mark_escaped(d, node, value, level) {
+        function mark_escaped(d, scope, node, value, level) {
             var parent = tw.parent(level);
             if (value) {
                 if (value.is_constant()) return;
-                if (level > 0 && value.is_constant_expression()) return;
+                if (level > 0 && value.is_constant_expression(scope)) return;
             }
             if (parent instanceof AST_Assign && parent.operator == "=" && node === parent.right
                 || parent instanceof AST_Call && node !== parent.expression
@@ -636,13 +636,13 @@ merge(Compressor.prototype, {
                 d.escaped = true;
                 return;
             } else if (parent instanceof AST_Array) {
-                mark_escaped(d, parent, parent, level + 1);
+                mark_escaped(d, scope, parent, parent, level + 1);
             } else if (parent instanceof AST_ObjectKeyVal && node === parent.value) {
                 var obj = tw.parent(level + 1);
-                mark_escaped(d, obj, obj, level + 2);
+                mark_escaped(d, scope, obj, obj, level + 2);
             } else if (parent instanceof AST_PropAccess && node === parent.expression) {
                 value = read_property(value, parent.property);
-                mark_escaped(d, parent, value, level + 1);
+                mark_escaped(d, scope, parent, value, level + 1);
                 if (value) return;
             }
             if (level == 0) d.direct_access = true;
diff --git a/test/compress/hoist_props.js b/test/compress/hoist_props.js
index 176cd9a8..b2dd0270 100644
--- a/test/compress/hoist_props.js
+++ b/test/compress/hoist_props.js
@@ -523,6 +523,7 @@ issue_2508_1: {
             console.log(x);
         })([ 1 ]);
     }
+    expect_stdout: true
 }
 
 issue_2508_2: {
@@ -547,4 +548,88 @@ issue_2508_2: {
             console.log(x);
         })({ b: 2 });
     }
+    expect_stdout: true
+}
+
+issue_2508_3: {
+    options = {
+        collapse_vars: true,
+        hoist_props: true,
+        reduce_vars: true,
+        toplevel: true,
+        unused: true,
+    }
+    input: {
+        var o = {
+            a: [ o ],
+            f: function(x) {
+                console.log(x);
+            }
+        };
+        o.f(o.a);
+    }
+    expect: {
+        var o = {
+            a: [ o ],
+            f: function(x) {
+                console.log(x);
+            }
+        };
+        o.f(o.a);
+    }
+    expect_stdout: true
+}
+
+issue_2508_4: {
+    options = {
+        collapse_vars: true,
+        hoist_props: true,
+        reduce_vars: true,
+        toplevel: true,
+        unused: true,
+    }
+    input: {
+        var o = {
+            a: { b: o },
+            f: function(x) {
+                console.log(x);
+            }
+        };
+        o.f(o.a);
+    }
+    expect: {
+        var o = {
+            a: { b: o },
+            f: function(x) {
+                console.log(x);
+            }
+        };
+        o.f(o.a);
+    }
+    expect_stdout: true
+}
+
+issue_2508_5: {
+    options = {
+        collapse_vars: true,
+        hoist_props: true,
+        reduce_vars: true,
+        toplevel: true,
+        unused: true,
+    }
+    input: {
+        var o = {
+            f: function(x) {
+                console.log(x);
+            }
+        };
+        o.f(o.f);
+    }
+    expect: {
+        var o_f = function(x) {
+            console.log(x);
+        };
+        o_f(o_f);
+    }
+    expect_stdout: true
 }