1 files changed, 164 insertions, 0 deletions
diff --git a/www/wiki/extensions/Scribunto/tests/phpunit/engines/LuaCommon/LuaInterpreterTest.php b/www/wiki/extensions/Scribunto/tests/phpunit/engines/LuaCommon/LuaInterpreterTest.php
new file mode 100644
index 00000000..7ee38144
--- /dev/null
+++ b/www/wiki/extensions/Scribunto/tests/phpunit/engines/LuaCommon/LuaInterpreterTest.php
@@ -0,0 +1,164 @@
+<?php
+
+// @codingStandardsIgnoreLine Squiz.Classes.ValidClassName.NotCamelCaps
+abstract class Scribunto_LuaInterpreterTest extends MediaWikiTestCase {
+	abstract protected function newInterpreter( $opts = [] );
+
+	protected function setUp() {
+		parent::setUp();
+		try {
+			$this->newInterpreter();
+		} catch ( Scribunto_LuaInterpreterNotFoundError $e ) {
+			$this->markTestSkipped( "interpreter not available" );
+		}
+	}
+
+	protected function getBusyLoop( $interpreter ) {
+		$chunk = $interpreter->loadString( '
+			local args = {...}
+			local x, i
+			local s = string.rep("x", 1000000)
+			local n = args[1]
+			for i = 1, n do
+				x = x or string.find(s, "y", 1, true)
+			end',
+			'busy' );
+		return $chunk;
+	}
+
+	/** @dataProvider provideRoundtrip */
+	public function testRoundtrip( /*...*/ ) {
+		$args = func_get_args();
+		$args = $this->normalizeOrder( $args );
+		$interpreter = $this->newInterpreter();
+		$passthru = $interpreter->loadString( 'return ...', 'passthru' );
+		$finalArgs = $args;
+		array_unshift( $finalArgs, $passthru );
+		$ret = call_user_func_array( [ $interpreter, 'callFunction' ], $finalArgs );
+		$ret = $this->normalizeOrder( $ret );
+		$this->assertSame( $args, $ret );
+	}
+
+	/** @dataProvider provideRoundtrip */
+	public function testDoubleRoundtrip( /* ... */ ) {
+		$args = func_get_args();
+		$args = $this->normalizeOrder( $args );
+
+		$interpreter = $this->newInterpreter();
+		$interpreter->registerLibrary( 'test',
+			[ 'passthru' => [ $this, 'passthru' ] ] );
+		$doublePassthru = $interpreter->loadString(
+			'return test.passthru(...)', 'doublePassthru' );
+
+		$finalArgs = $args;
+		array_unshift( $finalArgs, $doublePassthru );
+		$ret = call_user_func_array( [ $interpreter, 'callFunction' ], $finalArgs );
+		$ret = $this->normalizeOrder( $ret );
+		$this->assertSame( $args, $ret );
+	}
+
+	/**
+	 * This cannot be done in testRoundtrip and testDoubleRoundtrip, because
+	 * assertSame( NAN, NAN ) returns false.
+	 */
+	public function testRoundtripNAN() {
+		$interpreter = $this->newInterpreter();
+
+		$passthru = $interpreter->loadString( 'return ...', 'passthru' );
+		$ret = $interpreter->callFunction( $passthru, NAN );
+		$this->assertTrue( is_nan( $ret[0] ), 'NaN was not passed through' );
+
+		$interpreter->registerLibrary( 'test',
+			[ 'passthru' => [ $this, 'passthru' ] ] );
+		$doublePassthru = $interpreter->loadString(
+			'return test.passthru(...)', 'doublePassthru' );
+		$ret = $interpreter->callFunction( $doublePassthru, NAN );
+		$this->assertTrue( is_nan( $ret[0] ), 'NaN was not double passed through' );
+	}
+
+	private function normalizeOrder( $a ) {
+		ksort( $a );
+		foreach ( $a as &$value ) {
+			if ( is_array( $value ) ) {
+				$value = $this->normalizeOrder( $value );
+			}
+		}
+		return $a;
+	}
+
+	public function passthru( /* ... */ ) {
+		$args = func_get_args();
+		return $args;
+	}
+
+	public function provideRoundtrip() {
+		return [
+			[ 1 ],
+			[ true ],
+			[ false ],
+			[ 'hello' ],
+			[ implode( '', array_map( 'chr', range( 0, 255 ) ) ) ],
+			[ 1, 2, 3 ],
+			[ [] ],
+			[ [ 0 => 'foo', 1 => 'bar' ] ],
+			[ [ 1 => 'foo', 2 => 'bar' ] ],
+			[ [ 'x' => 'foo', 'y' => 'bar', 'z' => [] ] ],
+			[ INF ],
+			[ -INF ],
+			[ 'ok', null, 'ok' ],
+			[ null, 'ok' ],
+			[ 'ok', null ],
+			[ null ],
+		];
+	}
+
+	public function testTimeLimit() {
+		if ( php_uname( 's' ) === 'Darwin' ) {
+			$this->markTestSkipped( "Darwin is lacking POSIX timer, skipping CPU time limiting test." );
+		}
+
+		$interpreter = $this->newInterpreter( [ 'cpuLimit' => 2 ] );
+		$chunk = $this->getBusyLoop( $interpreter );
+		try {
+			$interpreter->callFunction( $chunk, 1e9 );
+			$this->fail( "Expected ScribuntoException was not thrown" );
+		} catch ( ScribuntoException $ex ) {
+			$this->assertSame( 'scribunto-common-timeout', $ex->messageName );
+		}
+	}
+
+	public function testTestMemoryLimit() {
+		$interpreter = $this->newInterpreter( [ 'memoryLimit' => 20 * 1e6 ] );
+		$chunk = $interpreter->loadString( '
+			t = {}
+			for i = 1, 10 do
+				t[#t + 1] = string.rep("x" .. i, 1000000)
+			end
+			',
+			'memoryLimit' );
+		try {
+			$interpreter->callFunction( $chunk );
+			$this->fail( "Expected ScribuntoException was not thrown" );
+		} catch ( ScribuntoException $ex ) {
+			$this->assertSame( 'scribunto-lua-error', $ex->messageName );
+			$this->assertSame( 'not enough memory', $ex->messageArgs[1] );
+		}
+	}
+
+	public function testWrapPHPFunction() {
+		$interpreter = $this->newInterpreter();
+		$func = $interpreter->wrapPhpFunction( function ( $n ) {
+			return [ 42, $n ];
+		} );
+		$res = $interpreter->callFunction( $func, 'From PHP' );
+		$this->assertEquals( [ 42, 'From PHP' ], $res );
+
+		$chunk = $interpreter->loadString( '
+			f = ...
+			return f( "From Lua" )
+			',
+			'wrappedPhpFunction' );
+		$res = $interpreter->callFunction( $chunk, $func );
+		$this->assertEquals( [ 42, 'From Lua' ], $res );
+	}
+}