--- /dev/null
+<?php
+
+/*
+ * This file is part of Psy Shell.
+ *
+ * (c) 2012-2018 Justin Hileman
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Psy\Test;
+
+use Psy\Configuration;
+use Psy\Exception\ErrorException;
+use Psy\Exception\ParseErrorException;
+use Psy\Shell;
+use Psy\TabCompletion\Matcher\ClassMethodsMatcher;
+use Symfony\Component\Console\Output\StreamOutput;
+
+class ShellTest extends \PHPUnit\Framework\TestCase
+{
+ private $streams = [];
+
+ public function tearDown()
+ {
+ foreach ($this->streams as $stream) {
+ fclose($stream);
+ }
+ }
+
+ public function testScopeVariables()
+ {
+ $one = 'banana';
+ $two = 123;
+ $three = new \StdClass();
+ $__psysh__ = 'ignore this';
+ $_ = 'ignore this';
+ $_e = 'ignore this';
+
+ $shell = new Shell($this->getConfig());
+ $shell->setScopeVariables(compact('one', 'two', 'three', '__psysh__', '_', '_e', 'this'));
+
+ $this->assertNotContains('__psysh__', $shell->getScopeVariableNames());
+ $this->assertSame(['one', 'two', 'three', '_'], $shell->getScopeVariableNames());
+ $this->assertSame('banana', $shell->getScopeVariable('one'));
+ $this->assertSame(123, $shell->getScopeVariable('two'));
+ $this->assertSame($three, $shell->getScopeVariable('three'));
+ $this->assertNull($shell->getScopeVariable('_'));
+
+ $shell->setScopeVariables([]);
+ $this->assertSame(['_'], $shell->getScopeVariableNames());
+
+ $shell->setBoundObject($this);
+ $this->assertSame(['_', 'this'], $shell->getScopeVariableNames());
+ $this->assertSame($this, $shell->getScopeVariable('this'));
+ $this->assertSame(['_' => null], $shell->getScopeVariables(false));
+ $this->assertSame(['_' => null, 'this' => $this], $shell->getScopeVariables());
+ }
+
+ /**
+ * @expectedException \InvalidArgumentException
+ */
+ public function testUnknownScopeVariablesThrowExceptions()
+ {
+ $shell = new Shell($this->getConfig());
+ $shell->setScopeVariables(['foo' => 'FOO', 'bar' => 1]);
+ $shell->getScopeVariable('baz');
+ }
+
+ public function testIncludesWithScopeVariables()
+ {
+ $one = 'banana';
+ $two = 123;
+ $three = new \StdClass();
+ $__psysh__ = 'ignore this';
+ $_ = 'ignore this';
+ $_e = 'ignore this';
+
+ $config = $this->getConfig(['usePcntl' => false]);
+
+ $shell = new Shell($config);
+ $shell->setScopeVariables(compact('one', 'two', 'three', '__psysh__', '_', '_e', 'this'));
+ $shell->addInput('exit', true);
+
+ // This is super slow and we shouldn't do this :(
+ $shell->run(null, $this->getOutput());
+
+ $this->assertNotContains('__psysh__', $shell->getScopeVariableNames());
+ $this->assertSame(['one', 'two', 'three', '_', '_e'], $shell->getScopeVariableNames());
+ $this->assertSame('banana', $shell->getScopeVariable('one'));
+ $this->assertSame(123, $shell->getScopeVariable('two'));
+ $this->assertSame($three, $shell->getScopeVariable('three'));
+ $this->assertNull($shell->getScopeVariable('_'));
+ }
+
+ public function testIncludes()
+ {
+ $config = $this->getConfig(['configFile' => __DIR__ . '/fixtures/empty.php']);
+
+ $shell = new Shell($config);
+ $this->assertEmpty($shell->getIncludes());
+ $shell->setIncludes(['foo', 'bar', 'baz']);
+ $this->assertSame(['foo', 'bar', 'baz'], $shell->getIncludes());
+ }
+
+ public function testIncludesConfig()
+ {
+ $config = $this->getConfig([
+ 'defaultIncludes' => ['/file.php'],
+ 'configFile' => __DIR__ . '/fixtures/empty.php',
+ ]);
+
+ $shell = new Shell($config);
+
+ $includes = $shell->getIncludes();
+ $this->assertSame('/file.php', $includes[0]);
+ }
+
+ public function testAddMatchersViaConfig()
+ {
+ $shell = new FakeShell();
+ $matcher = new ClassMethodsMatcher();
+
+ $config = $this->getConfig([
+ 'matchers' => [$matcher],
+ ]);
+ $config->setShell($shell);
+
+ $this->assertSame([$matcher], $shell->matchers);
+ }
+
+ public function testAddMatchersViaConfigAfterShell()
+ {
+ $shell = new FakeShell();
+ $matcher = new ClassMethodsMatcher();
+
+ $config = $this->getConfig([]);
+ $config->setShell($shell);
+ $config->addMatchers([$matcher]);
+
+ $this->assertSame([$matcher], $shell->matchers);
+ }
+
+ public function testRenderingExceptions()
+ {
+ $shell = new Shell($this->getConfig());
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $e = new ParseErrorException('message', 13);
+
+ $shell->setOutput($output);
+ $shell->addCode('code');
+ $this->assertTrue($shell->hasCode());
+ $this->assertNotEmpty($shell->getCodeBuffer());
+
+ $shell->writeException($e);
+
+ $this->assertSame($e, $shell->getScopeVariable('_e'));
+ $this->assertFalse($shell->hasCode());
+ $this->assertEmpty($shell->getCodeBuffer());
+
+ rewind($stream);
+ $streamContents = stream_get_contents($stream);
+
+ $this->assertContains('PHP Parse error', $streamContents);
+ $this->assertContains('message', $streamContents);
+ $this->assertContains('line 13', $streamContents);
+ }
+
+ public function testHandlingErrors()
+ {
+ $shell = new Shell($this->getConfig());
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $shell->setOutput($output);
+
+ $oldLevel = error_reporting();
+ error_reporting($oldLevel & ~E_USER_NOTICE);
+
+ try {
+ $shell->handleError(E_USER_NOTICE, 'wheee', null, 13);
+ } catch (ErrorException $e) {
+ error_reporting($oldLevel);
+ $this->fail('Unexpected error exception');
+ }
+ error_reporting($oldLevel);
+
+ rewind($stream);
+ $streamContents = stream_get_contents($stream);
+
+ $this->assertContains('PHP Notice:', $streamContents);
+ $this->assertContains('wheee', $streamContents);
+ $this->assertContains('line 13', $streamContents);
+ }
+
+ /**
+ * @expectedException \Psy\Exception\ErrorException
+ */
+ public function testNotHandlingErrors()
+ {
+ $shell = new Shell($this->getConfig());
+ $oldLevel = error_reporting();
+ error_reporting($oldLevel | E_USER_NOTICE);
+
+ try {
+ $shell->handleError(E_USER_NOTICE, 'wheee', null, 13);
+ } catch (ErrorException $e) {
+ error_reporting($oldLevel);
+ throw $e;
+ }
+ }
+
+ public function testVersion()
+ {
+ $shell = new Shell($this->getConfig());
+
+ $this->assertInstanceOf('Symfony\Component\Console\Application', $shell);
+ $this->assertContains(Shell::VERSION, $shell->getVersion());
+ $this->assertContains(phpversion(), $shell->getVersion());
+ $this->assertContains(php_sapi_name(), $shell->getVersion());
+ }
+
+ public function testCodeBuffer()
+ {
+ $shell = new Shell($this->getConfig());
+
+ $shell->addCode('class');
+ $this->assertNull($shell->flushCode());
+ $this->assertTrue($shell->hasCode());
+
+ $shell->addCode('a');
+ $this->assertNull($shell->flushCode());
+ $this->assertTrue($shell->hasCode());
+
+ $shell->addCode('{}');
+ $code = $shell->flushCode();
+ $this->assertFalse($shell->hasCode());
+ $code = preg_replace('/\s+/', ' ', $code);
+ $this->assertNotNull($code);
+ $this->assertSame('class a { } return new \\Psy\\CodeCleaner\\NoReturnValue();', $code);
+ }
+
+ public function testKeepCodeBufferOpen()
+ {
+ $shell = new Shell($this->getConfig());
+
+ $shell->addCode('1 \\');
+ $this->assertNull($shell->flushCode());
+ $this->assertTrue($shell->hasCode());
+
+ $shell->addCode('+ 1 \\');
+ $this->assertNull($shell->flushCode());
+ $this->assertTrue($shell->hasCode());
+
+ $shell->addCode('+ 1');
+ $code = $shell->flushCode();
+ $this->assertFalse($shell->hasCode());
+ $code = preg_replace('/\s+/', ' ', $code);
+ $this->assertNotNull($code);
+ $this->assertSame('return 1 + 1 + 1;', $code);
+ }
+
+ /**
+ * @expectedException \Psy\Exception\ParseErrorException
+ */
+ public function testCodeBufferThrowsParseExceptions()
+ {
+ $shell = new Shell($this->getConfig());
+ $shell->addCode('this is not valid');
+ $shell->flushCode();
+ }
+
+ public function testClosuresSupport()
+ {
+ $shell = new Shell($this->getConfig());
+ $code = '$test = function () {}';
+ $shell->addCode($code);
+ $shell->flushCode();
+ $code = '$test()';
+ $shell->addCode($code);
+ $this->assertSame($shell->flushCode(), 'return $test();');
+ }
+
+ public function testWriteStdout()
+ {
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $shell = new Shell($this->getConfig());
+ $shell->setOutput($output);
+
+ $shell->writeStdout("{{stdout}}\n");
+
+ rewind($stream);
+ $streamContents = stream_get_contents($stream);
+
+ $this->assertSame('{{stdout}}' . PHP_EOL, $streamContents);
+ }
+
+ public function testWriteStdoutWithoutNewline()
+ {
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $shell = new Shell($this->getConfig());
+ $shell->setOutput($output);
+
+ $shell->writeStdout('{{stdout}}');
+
+ rewind($stream);
+ $streamContents = stream_get_contents($stream);
+
+ $this->assertSame('{{stdout}}<aside>⏎</aside>' . PHP_EOL, $streamContents);
+ }
+
+ /**
+ * @dataProvider getReturnValues
+ */
+ public function testWriteReturnValue($input, $expected)
+ {
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $shell = new Shell($this->getConfig());
+ $shell->setOutput($output);
+
+ $shell->writeReturnValue($input);
+ rewind($stream);
+ $this->assertEquals($expected, stream_get_contents($stream));
+ }
+
+ public function getReturnValues()
+ {
+ return [
+ ['{{return value}}', "=> \"\033[32m{{return value}}\033[39m\"" . PHP_EOL],
+ [1, "=> \033[35m1\033[39m" . PHP_EOL],
+ ];
+ }
+
+ /**
+ * @dataProvider getRenderedExceptions
+ */
+ public function testWriteException($exception, $expected)
+ {
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $shell = new Shell($this->getConfig());
+ $shell->setOutput($output);
+
+ $shell->writeException($exception);
+ rewind($stream);
+ $this->assertSame($expected, stream_get_contents($stream));
+ }
+
+ public function getRenderedExceptions()
+ {
+ return [
+ [new \Exception('{{message}}'), "Exception with message '{{message}}'" . PHP_EOL],
+ ];
+ }
+
+ /**
+ * @dataProvider getExecuteValues
+ */
+ public function testShellExecute($input, $expected)
+ {
+ $output = $this->getOutput();
+ $stream = $output->getStream();
+ $shell = new Shell($this->getConfig());
+ $shell->setOutput($output);
+ $this->assertEquals($expected, $shell->execute($input));
+ rewind($stream);
+ $this->assertSame('', stream_get_contents($stream));
+ }
+
+ public function getExecuteValues()
+ {
+ return [
+ ['return 12', 12],
+ ['"{{return value}}"', '{{return value}}'],
+ ['1', '1'],
+ ];
+ }
+
+ private function getOutput()
+ {
+ $stream = fopen('php://memory', 'w+');
+ $this->streams[] = $stream;
+
+ $output = new StreamOutput($stream, StreamOutput::VERBOSITY_NORMAL, false);
+
+ return $output;
+ }
+
+ private function getConfig(array $config = [])
+ {
+ // Mebbe there's a better way than this?
+ $dir = tempnam(sys_get_temp_dir(), 'psysh_shell_test_');
+ unlink($dir);
+
+ $defaults = [
+ 'configDir' => $dir,
+ 'dataDir' => $dir,
+ 'runtimeDir' => $dir,
+ ];
+
+ return new Configuration(array_merge($defaults, $config));
+ }
+}