luaj/src/test/java/lua/LuaJTest.java

package lua;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;

import junit.framework.TestCase;
import lua.StackState;
import lua.addon.luajava.LuaJava;
import lua.io.Closure;
import lua.io.LoadState;
import lua.io.Proto;
import lua.value.LValue;


public class LuaJTest extends TestCase {

	public void testTest1() throws IOException, InterruptedException {
		runTest( "test1" );
	}

	public void testTest2() throws IOException, InterruptedException {
		runTest( "test2" );
	}

	public void testTest3() throws IOException, InterruptedException {
		runTest( "test3" );
	}

	public void testTest4() throws IOException, InterruptedException {
		runTest( "test4" );
	}

	public void testTest5() throws IOException, InterruptedException {
		runTest( "test5" );
	}

	public void testTest6() throws IOException, InterruptedException {
		runTest( "test6" );
	}

	public void testTest7() throws IOException, InterruptedException {
		runTest( "test7" );
	}
	
	public void testUpvalues() throws IOException, InterruptedException {
		runTest( "upvalues" );
	}
	
	public void testUpvalues2() throws IOException, InterruptedException {
		runTest( "upvalues2" );
	}
	
	private void runTest( String testName ) throws IOException, InterruptedException {
		// add LuaJava bindings
		LuaJava.install();
		
		// new lua state 
		StackState state = new StackState();
		
		// load the file
		Proto p = loadScriptResource( state, testName, "/" + testName + ".luac" );
		
		// Replace System.out with a ByteArrayOutputStream
		ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
		Builtin.redirectOutput( outputStream );
		try {
			// create closure and execute
			Closure c = new Closure( state, p );
			state.doCall(c, new LValue[0], 0);
			
			final String actualOutput = new String( outputStream.toByteArray() );
			final String expectedOutput = getExpectedOutput( testName );
			
			assertEquals( expectedOutput, actualOutput );
		} finally {
			Builtin.restoreStandardOutput();
			outputStream.close();
		}
	}
	
	private Proto loadScriptResource( StackState state, String name, String path ) throws IOException {
		InputStream compiledScript = getClass().getResourceAsStream( path );
		try {
			return LoadState.undump(state, compiledScript, name);
		} finally {
			compiledScript.close();
		}
	}
	
	private String getExpectedOutput( final String testName ) throws IOException, InterruptedException {
		String expectedOutputName = "/" + testName + "-expected.out";
		InputStream is = getClass().getResourceAsStream( expectedOutputName );
		if ( is != null ) {
			try {
				return readString( is );
			} finally {
				is.close();
			}
		} else {
			InputStream script = getClass().getResourceAsStream( "/" + testName + ".luac" );
			try {
				return collectProcessOutput( new String[] { "lua", "-" }, script );
			} finally {
				script.close();
			}
		}
	}
	
	private String collectProcessOutput( String[] cmd, final InputStream input ) throws IOException, InterruptedException {
		Runtime r = Runtime.getRuntime();
		final ByteArrayOutputStream baos = new ByteArrayOutputStream();
		final Process p = r.exec( cmd );
		try {
			// start a thread to write the given input to the subprocess.
			Thread inputCopier = (new Thread() {
				public void run() {
					try {
						OutputStream processStdIn = p.getOutputStream();
						try {
							copy( input, processStdIn );
						} finally {
							processStdIn.close();
						}
					} catch ( IOException e ) {
						e.printStackTrace();
					}
				}
			});
			inputCopier.start();
			
			// start another thread to read output from the subprocess.
			Thread outputCopier = (new Thread() {
				public void run() {
					try {
						InputStream processStdOut = p.getInputStream();
						try {
							copy( processStdOut, baos );
						} finally {
							processStdOut.close();
						}
					} catch ( IOException ioe ) {
						ioe.printStackTrace();
					}
				}
			});
			outputCopier.start();
			
			p.waitFor();
			inputCopier.join();
			outputCopier.join();
			
			return new String( baos.toByteArray() );
			
		} finally {
			p.destroy();
		}
	}
	
	private String readString( InputStream is ) throws IOException {
		ByteArrayOutputStream baos = new ByteArrayOutputStream();
		copy( is, baos );
		return new String( baos.toByteArray() );
	}
	
	private void copy( InputStream is, OutputStream os ) throws IOException {
		byte[] buf = new byte[ 1024 ];
		int r;
		while ( ( r = is.read( buf ) ) >= 0 ) {
			os.write( buf, 0, r );
		}
	}

}