272 lines
7.6 KiB
Java
272 lines
7.6 KiB
Java
/*******************************************************************************
|
|
* Copyright (c) 2009 Luaj.org. All rights reserved.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
******************************************************************************/
|
|
package org.luaj.vm2;
|
|
|
|
import java.io.ByteArrayInputStream;
|
|
import java.io.InputStream;
|
|
|
|
import junit.framework.TestCase;
|
|
|
|
import org.luaj.vm2.compiler.LuaC;
|
|
import org.luaj.vm2.luajc.JavaBytecodeCompiler;
|
|
|
|
/**
|
|
* Test compilation of various fragments that have
|
|
* caused problems for jit compiling during development.
|
|
*
|
|
*/
|
|
public class FragmentsTest extends TestCase {
|
|
|
|
public void runFragment( Varargs expected, String script ) {
|
|
try {
|
|
String name = getName();
|
|
LuaTable _G = org.luaj.vm2.lib.JsePlatform.standardGlobals();
|
|
InputStream is = new ByteArrayInputStream(script.getBytes("UTF-8"));
|
|
LuaValue chunk ;
|
|
if ( true ) {
|
|
chunk = JavaBytecodeCompiler.load(is,name,_G);
|
|
} else {
|
|
chunk = (new LuaC()).load( is, name, _G );
|
|
}
|
|
Varargs actual = chunk.invoke();
|
|
assertEquals( expected.narg(), actual.narg() );
|
|
for ( int i=1; i<=actual.narg(); i++ )
|
|
assertEquals( expected.arg(i), actual.arg(i) );
|
|
} catch (Exception e) {
|
|
// TODO Auto-generated catch block
|
|
e.printStackTrace();
|
|
fail(e.toString());
|
|
}
|
|
}
|
|
|
|
public void testVarVarargsUseArg() {
|
|
runFragment( LuaValue.varargsOf( new LuaValue[] {
|
|
LuaValue.valueOf("a"),
|
|
LuaValue.valueOf(2),
|
|
LuaValue.valueOf("b"),
|
|
LuaValue.valueOf("c"),
|
|
LuaValue.NIL }),
|
|
"function q(a,...)\n" +
|
|
" return a,arg.n,arg[1],arg[2],arg[3]\n" +
|
|
"end\n" +
|
|
"return q('a','b','c')\n" );
|
|
}
|
|
|
|
public void testVarVarargsUseBoth() {
|
|
runFragment( LuaValue.varargsOf( new LuaValue[] {
|
|
LuaValue.valueOf("a"),
|
|
LuaValue.valueOf("nil"),
|
|
LuaValue.valueOf("b"),
|
|
LuaValue.valueOf("c")}),
|
|
"function r(a,...)\n" +
|
|
" return a,type(arg),...\n" +
|
|
"end\n" +
|
|
"return r('a','b','c')\n" );
|
|
}
|
|
|
|
public void testArgVarargsUseBoth() {
|
|
runFragment( LuaValue.varargsOf( new LuaValue[] {
|
|
LuaValue.NIL,
|
|
LuaValue.valueOf("b"),
|
|
LuaValue.valueOf("c")}),
|
|
"function v(arg,...)\n" +
|
|
" return arg,...\n" +
|
|
"end\n" +
|
|
"return v('a','b','c')\n" );
|
|
}
|
|
|
|
public void testArgParamUseNone() {
|
|
// the name "arg" is treated specially, and ends up masking the argument value in 5.1
|
|
runFragment( LuaValue.valueOf("table"),
|
|
"function v(arg,...)\n" +
|
|
" return type(arg)\n" +
|
|
"end\n" +
|
|
"return v('abc')\n" );
|
|
}
|
|
|
|
public void testSetlistVarargs() {
|
|
runFragment( LuaValue.valueOf("abc"),
|
|
"local f = function() return 'abc' end\n" +
|
|
"local g = { f() }\n" +
|
|
"return g[1]\n" );
|
|
}
|
|
|
|
public void testSelfOp() {
|
|
runFragment( LuaValue.valueOf("bcd"),
|
|
"local s = 'abcde'\n"+
|
|
"return s:sub(2,4)\n" );
|
|
}
|
|
|
|
public void testSetListWithOffsetAndVarargs() {
|
|
runFragment( LuaValue.valueOf(1003),
|
|
"local bar = {1000, math.sqrt(9)}\n"+
|
|
"return bar[1]+bar[2]\n" );
|
|
}
|
|
|
|
|
|
public void testMultiAssign() {
|
|
// arargs evaluations are all done before assignments
|
|
runFragment( LuaValue.varargsOf(new LuaValue[]{
|
|
LuaValue.valueOf(111),
|
|
LuaValue.valueOf(111),
|
|
LuaValue.valueOf(111)}),
|
|
"a,b,c = 1,10,100\n" +
|
|
"a,b,c = a+b+c, a+b+c, a+b+c\n" +
|
|
"return a,b,c\n" );
|
|
}
|
|
|
|
public void testUpvalues() {
|
|
runFragment( LuaValue.valueOf(999),
|
|
"local a = function(x)\n" +
|
|
" return function(y)\n" +
|
|
" return x + y\n" +
|
|
" end\n" +
|
|
"end\n" +
|
|
"local b = a(222)\n" +
|
|
"local c = b(777)\n" +
|
|
"print( 'c=', c )\n" +
|
|
"return c\n" );
|
|
}
|
|
|
|
public void testNeedsArgAndHasArg() {
|
|
runFragment( LuaValue.varargsOf(LuaValue.valueOf(333),LuaValue.NIL,LuaValue.valueOf(222)),
|
|
"function r(q,...)\n"+
|
|
" local a=arg\n"+
|
|
" return a and a[2]\n"+
|
|
"end\n" +
|
|
"function s(q,...)\n"+
|
|
" local a=arg\n"+
|
|
" local b=...\n"+
|
|
" return a and a[2],b\n"+
|
|
"end\n" +
|
|
"return r(111,222,333),s(111,222,333)" );
|
|
|
|
}
|
|
|
|
public void testNonAsciiStringLiterals() {
|
|
runFragment( LuaValue.valueOf("7,8,12,10,9,11,133,222"),
|
|
"local a='\\a\\b\\f\\n\\t\\v\\133\\222'\n"+
|
|
"local t={string.byte(a,1,#a)}\n"+
|
|
"return table.concat(t,',')\n" );
|
|
}
|
|
|
|
public void testControlCharStringLiterals() {
|
|
runFragment( LuaValue.valueOf("97,0,98,18,99,18,100,18,48,101"),
|
|
"local a='a\\0b\\18c\\018d\\0180e'\n"+
|
|
"local t={string.byte(a,1,#a)}\n"+
|
|
"return table.concat(t,',')\n" );
|
|
}
|
|
|
|
public void testLoopVarNames() {
|
|
runFragment( LuaValue.valueOf(" 234,1,aa 234,2,bb"),
|
|
"local w = ''\n"+
|
|
"function t()\n"+
|
|
" for f,var in ipairs({'aa','bb'}) do\n"+
|
|
" local s = 234\n"+
|
|
" w = w..' '..s..','..f..','..var\n"+
|
|
" end\n"+
|
|
"end\n" +
|
|
"t()\n" +
|
|
"return w\n" );
|
|
|
|
}
|
|
|
|
public void testForLoops() {
|
|
runFragment( LuaValue.valueOf("12345 357 963"),
|
|
"local s,t,u = '','',''\n"+
|
|
"for m=1,5 do\n"+
|
|
" s = s..m\n"+
|
|
"end\n"+
|
|
"for m=3,7,2 do\n"+
|
|
" t = t..m\n"+
|
|
"end\n"+
|
|
"for m=9,3,-3 do\n"+
|
|
" u = u..m\n"+
|
|
"end\n"+
|
|
"return s..' '..t..' '..u\n" );
|
|
}
|
|
|
|
public void testLocalFunctionDeclarations() {
|
|
runFragment( LuaValue.varargsOf(LuaValue.valueOf("function"),LuaValue.valueOf("nil")),
|
|
"local function aaa()\n"+
|
|
" return type(aaa)\n"+
|
|
"end\n"+
|
|
"local bbb = function()\n"+
|
|
" return type(bbb)\n"+
|
|
"end\n"+
|
|
"return aaa(),bbb()\n" );
|
|
}
|
|
|
|
public void testNilsInTableConstructor() {
|
|
runFragment( LuaValue.valueOf("1=111 2=222 3=333 "),
|
|
"local t = { 111, 222, 333, nil, nil }\n"+
|
|
"local s = ''\n"+
|
|
"for i,v in ipairs(t) do \n" +
|
|
" s=s..tostring(i)..'='..tostring(v)..' '\n" +
|
|
"end\n"+
|
|
"return s\n" );
|
|
|
|
}
|
|
|
|
public void testUnreachableCode() {
|
|
runFragment( LuaValue.valueOf(66),
|
|
"local function foo(x) return x * 2 end\n" +
|
|
"local function bar(x, y)\n" +
|
|
" if x==y then\n" +
|
|
" return y\n" +
|
|
" else\n" +
|
|
" return foo(x)\n" +
|
|
" end\n" +
|
|
"end\n" +
|
|
"return bar(33,44)\n" );
|
|
|
|
}
|
|
public void testVarargsWithParameters() {
|
|
runFragment( LuaValue.valueOf(222),
|
|
"local func = function(t,...)\n"+
|
|
" return (...)\n"+
|
|
"end\n"+
|
|
"return func(111,222,333)\n" );
|
|
}
|
|
|
|
public void testNoReturnValuesPlainCall() {
|
|
runFragment( LuaValue.TRUE,
|
|
"local testtable = {}\n"+
|
|
"return pcall( function() testtable[1]=2 end )\n" );
|
|
}
|
|
|
|
public void testVarargsInTableConstructor() {
|
|
runFragment( LuaValue.valueOf(222),
|
|
"local function foo() return 111,222,333 end\n"+
|
|
"local t = {'a','b',c='c',foo()}\n"+
|
|
"return t[4]\n" );
|
|
}
|
|
|
|
public void testVarargsInFirstArg() {
|
|
runFragment( LuaValue.valueOf(123),
|
|
"function aaa(x) return x end\n" +
|
|
"function bbb(y) return y end\n" +
|
|
"function ccc(z) return z end\n" +
|
|
"return ccc( aaa(bbb(123)), aaa(456) )\n" );
|
|
}
|
|
}
|