123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186 |
- <?php
- /*
- * This file is part of the Symfony package.
- *
- * (c) Fabien Potencier <fabien@symfony.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Symfony\Component\Console\Tester;
- use Symfony\Component\Console\Input\InputInterface;
- use Symfony\Component\Console\Output\ConsoleOutput;
- use Symfony\Component\Console\Output\OutputInterface;
- use Symfony\Component\Console\Output\StreamOutput;
- /**
- * @author Amrouche Hamza <hamza.simperfit@gmail.com>
- */
- trait TesterTrait
- {
- /** @var StreamOutput */
- private $output;
- private $inputs = [];
- private $captureStreamsIndependently = false;
- /**
- * Gets the display returned by the last execution of the command or application.
- *
- * @throws \RuntimeException If it's called before the execute method
- *
- * @return string The display
- */
- public function getDisplay(bool $normalize = false)
- {
- if (null === $this->output) {
- throw new \RuntimeException('Output not initialized, did you execute the command before requesting the display?');
- }
- rewind($this->output->getStream());
- $display = stream_get_contents($this->output->getStream());
- if ($normalize) {
- $display = str_replace(\PHP_EOL, "\n", $display);
- }
- return $display;
- }
- /**
- * Gets the output written to STDERR by the application.
- *
- * @param bool $normalize Whether to normalize end of lines to \n or not
- *
- * @return string
- */
- public function getErrorOutput(bool $normalize = false)
- {
- if (!$this->captureStreamsIndependently) {
- throw new \LogicException('The error output is not available when the tester is run without "capture_stderr_separately" option set.');
- }
- rewind($this->output->getErrorOutput()->getStream());
- $display = stream_get_contents($this->output->getErrorOutput()->getStream());
- if ($normalize) {
- $display = str_replace(\PHP_EOL, "\n", $display);
- }
- return $display;
- }
- /**
- * Gets the input instance used by the last execution of the command or application.
- *
- * @return InputInterface The current input instance
- */
- public function getInput()
- {
- return $this->input;
- }
- /**
- * Gets the output instance used by the last execution of the command or application.
- *
- * @return OutputInterface The current output instance
- */
- public function getOutput()
- {
- return $this->output;
- }
- /**
- * Gets the status code returned by the last execution of the command or application.
- *
- * @throws \RuntimeException If it's called before the execute method
- *
- * @return int The status code
- */
- public function getStatusCode()
- {
- if (null === $this->statusCode) {
- throw new \RuntimeException('Status code not initialized, did you execute the command before requesting the status code?');
- }
- return $this->statusCode;
- }
- /**
- * Sets the user inputs.
- *
- * @param array $inputs An array of strings representing each input
- * passed to the command input stream
- *
- * @return $this
- */
- public function setInputs(array $inputs)
- {
- $this->inputs = $inputs;
- return $this;
- }
- /**
- * Initializes the output property.
- *
- * Available options:
- *
- * * decorated: Sets the output decorated flag
- * * verbosity: Sets the output verbosity flag
- * * capture_stderr_separately: Make output of stdOut and stdErr separately available
- */
- private function initOutput(array $options)
- {
- $this->captureStreamsIndependently = \array_key_exists('capture_stderr_separately', $options) && $options['capture_stderr_separately'];
- if (!$this->captureStreamsIndependently) {
- $this->output = new StreamOutput(fopen('php://memory', 'w', false));
- if (isset($options['decorated'])) {
- $this->output->setDecorated($options['decorated']);
- }
- if (isset($options['verbosity'])) {
- $this->output->setVerbosity($options['verbosity']);
- }
- } else {
- $this->output = new ConsoleOutput(
- $options['verbosity'] ?? ConsoleOutput::VERBOSITY_NORMAL,
- $options['decorated'] ?? null
- );
- $errorOutput = new StreamOutput(fopen('php://memory', 'w', false));
- $errorOutput->setFormatter($this->output->getFormatter());
- $errorOutput->setVerbosity($this->output->getVerbosity());
- $errorOutput->setDecorated($this->output->isDecorated());
- $reflectedOutput = new \ReflectionObject($this->output);
- $strErrProperty = $reflectedOutput->getProperty('stderr');
- $strErrProperty->setAccessible(true);
- $strErrProperty->setValue($this->output, $errorOutput);
- $reflectedParent = $reflectedOutput->getParentClass();
- $streamProperty = $reflectedParent->getProperty('stream');
- $streamProperty->setAccessible(true);
- $streamProperty->setValue($this->output, fopen('php://memory', 'w', false));
- }
- }
- /**
- * @return resource
- */
- private static function createStream(array $inputs)
- {
- $stream = fopen('php://memory', 'r+', false);
- foreach ($inputs as $input) {
- fwrite($stream, $input.\PHP_EOL);
- }
- rewind($stream);
- return $stream;
- }
- }
|