diff options
Diffstat (limited to 'test/test-suite/closure.test.js')
| -rw-r--r-- | test/test-suite/closure.test.js | 246 |
1 files changed, 246 insertions, 0 deletions
diff --git a/test/test-suite/closure.test.js b/test/test-suite/closure.test.js new file mode 100644 index 0000000..d44d908 --- /dev/null +++ b/test/test-suite/closure.test.js @@ -0,0 +1,246 @@ +"use strict"; + +const lua = require('../../src/lua.js'); +const lauxlib = require('../../src/lauxlib.js'); +const lualib = require('../../src/lualib.js'); +const {to_luastring} = require("../../src/fengaricore.js"); + +test("[test-suite] closure: testing equality", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + a = {} + for i = 1, 5 do a[i] = function (x) return x + a + _ENV end end + assert(a[3] == a[4] and a[4] == a[5]) + + for i = 1, 5 do a[i] = function (x) return i + a + _ENV end end + assert(a[3] ~= a[4] and a[4] ~= a[5]) + + local function f() + return function (x) return math.sin(_ENV[x]) end + end + assert(f() == f()) + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: testing closures with 'for' control variable", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + a = {} + for i=1,10 do + a[i] = {set = function(x) i=x end, get = function () return i end} + if i == 3 then break end + end + assert(a[4] == nil) + a[1].set(10) + assert(a[2].get() == 2) + a[2].set('a') + assert(a[3].get() == 3) + assert(a[2].get() == 'a') + + a = {} + local t = {"a", "b"} + for i = 1, #t do + local k = t[i] + a[i] = {set = function(x, y) i=x; k=y end, + get = function () return i, k end} + if i == 2 then break end + end + a[1].set(10, 20) + local r,s = a[2].get() + assert(r == 2 and s == 'b') + r,s = a[1].get() + assert(r == 10 and s == 20) + a[2].set('a', 'b') + r,s = a[2].get() + assert(r == "a" and s == "b") + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: testing closures with 'for' control variable x break", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + local t = {"a", "b"} + + for i=1,3 do + f = function () return i end + break + end + assert(f() == 1) + + for k = 1, #t do + local v = t[k] + f = function () return k, v end + break + end + assert(({f()})[1] == 1) + assert(({f()})[2] == "a") + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: testing closure x break x return x errors", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + local b + function f(x) + local first = 1 + while 1 do + if x == 3 and not first then return end + local a = 'xuxu' + b = function (op, y) + if op == 'set' then + a = x+y + else + return a + end + end + if x == 1 then do break end + elseif x == 2 then return + else if x ~= 3 then error() end + end + first = nil + end + end + + for i=1,3 do + f(i) + assert(b('get') == 'xuxu') + b('set', 10); assert(b('get') == 10+i) + b = nil + end + + pcall(f, 4); + assert(b('get') == 'xuxu') + b('set', 10); assert(b('get') == 14) + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: testing multi-level closure", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + local w + function f(x) + return function (y) + return function (z) return w+x+y+z end + end + end + + y = f(10) + w = 1.345 + assert(y(20)(30) == 60+w) + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: testing closures x repeat-until", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + local a = {} + local i = 1 + repeat + local x = i + a[i] = function () i = x+1; return x end + until i > 10 or a[i]() ~= x + assert(i == 11 and a[1]() == 1 and a[3]() == 3 and i == 4) + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: testing closures created in 'then' and 'else' parts of 'if's", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + a = {} + for i = 1, 10 do + if i % 3 == 0 then + local y = 0 + a[i] = function (x) local t = y; y = x; return t end + elseif i % 3 == 1 then + goto L1 + error'not here' + ::L1:: + local y = 1 + a[i] = function (x) local t = y; y = x; return t end + elseif i % 3 == 2 then + local t + goto l4 + ::l4a:: a[i] = t; goto l4b + error("should never be here!") + ::l4:: + local y = 2 + t = function (x) local t = y; y = x; return t end + goto l4a + error("should never be here!") + ::l4b:: + end + end + + for i = 1, 10 do + assert(a[i](i * 10) == i % 3 and a[i]() == i * 10) + end + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); + + +test("[test-suite] closure: test for correctly closing upvalues in tail calls of vararg functions", () => { + let L = lauxlib.luaL_newstate(); + if (!L) throw Error("failed to create lua state"); + + let luaCode = ` + local function t () + local function c(a,b) assert(a=="test" and b=="OK") end + local function v(f, ...) c("test", f() ~= 1 and "FAILED" or "OK") end + local x = 1 + return v(function() return x end) + end + t() + `; + lualib.luaL_openlibs(L); + if (lauxlib.luaL_loadstring(L, to_luastring(luaCode)) === lua.LUA_ERRSYNTAX) + throw new SyntaxError(lua.lua_tojsstring(L, -1)); + lua.lua_call(L, 0, 0); +}); |