fixing issue #143 and #144.

1 files changed, 34 insertions, 11 deletions

diff --git a/spec/outputs/attrib.lua b/spec/outputs/attrib.lua
index aaff747..919b7e0 100644
--- a/spec/outputs/attrib.lua
+++ b/spec/outputs/attrib.lua
@@ -1,14 +1,35 @@
+local a <const>, b <const>, c <const>, d <const> = 1, 2, 3, 4
 do
-        local a <close>, b <close> = setmetatable({ }, {
+        local a, b = setmetatable({ }, {
                 __close = function(self)
                         return print("closed")
                 end
         })
+        local _close_0 <close> = a
+        local _close_1 <close> = b
         local c <const>, d <const> = 123, 'abc'
         close(a, b)
         const(c, d)
 end
 do
+        local a <const> = f()
+        local b, c, d
+        do
+                local _obj_0, _obj_1 = f1()
+                b, c = _obj_0[1], _obj_0[2]
+                d = _obj_1[1]
+        end
+end
+do
+        local a, b, c, d
+        do
+                local _obj_0, _obj_1, _obj_2 = f()
+                a = _obj_0
+                b, c = _obj_1[1], _obj_1[2]
+                d = _obj_2[1]
+        end
+end
+do
         local v <close> = (function()
                 if flag then
                         return func()
@@ -66,7 +87,7 @@ do
                 return _with_0
         end)()
         local _ <close> = setmetatable({ }, {
-                __close = function(self)
+                __close = function()
                         return print("second")
                 end
         })
@@ -82,17 +103,19 @@ local _defers = setmetatable({ }, {
                 self[#self] = nil
         end
 })
+local def
+def = function(item)
+        _defers[#_defers + 1] = item
+        return _defers
+end
 do
-        _defers[#_defers + 1] = function()
+        local _ <close> = def(function()
                 return print(3)
-        end
-        local _ <close> = _defers
-        _defers[#_defers + 1] = function()
+        end)
+        local _ <close> = def(function()
                 return print(2)
-        end
-        local _ <close> = _defers
-        _defers[#_defers + 1] = function()
+        end)
+        local _ <close> = def(function()
                 return print(1)
-        end
-        local _ <close> = _defers
+        end)
 end