Spaces:
Sleeping
Sleeping
| /* | |
| * 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\Command\Command; | |
| use Symfony\Component\Console\Input\ArrayInput; | |
| use Symfony\Component\Console\Output\StreamOutput; | |
| use Symfony\Component\Console\Input\InputInterface; | |
| use Symfony\Component\Console\Output\OutputInterface; | |
| /** | |
| * Eases the testing of console commands. | |
| * | |
| * @author Fabien Potencier <fabien@symfony.com> | |
| * @author Robin Chalas <robin.chalas@gmail.com> | |
| */ | |
| class CommandTester | |
| { | |
| private $command; | |
| private $input; | |
| private $output; | |
| private $inputs = array(); | |
| private $statusCode; | |
| /** | |
| * Constructor. | |
| * | |
| * @param Command $command A Command instance to test | |
| */ | |
| public function __construct(Command $command) | |
| { | |
| $this->command = $command; | |
| } | |
| /** | |
| * Executes the command. | |
| * | |
| * Available execution options: | |
| * | |
| * * interactive: Sets the input interactive flag | |
| * * decorated: Sets the output decorated flag | |
| * * verbosity: Sets the output verbosity flag | |
| * | |
| * @param array $input An array of command arguments and options | |
| * @param array $options An array of execution options | |
| * | |
| * @return int The command exit code | |
| */ | |
| public function execute(array $input, array $options = array()) | |
| { | |
| // set the command name automatically if the application requires | |
| // this argument and no command name was passed | |
| if (!isset($input['command']) | |
| && (null !== $application = $this->command->getApplication()) | |
| && $application->getDefinition()->hasArgument('command') | |
| ) { | |
| $input = array_merge(array('command' => $this->command->getName()), $input); | |
| } | |
| $this->input = new ArrayInput($input); | |
| if ($this->inputs) { | |
| $this->input->setStream(self::createStream($this->inputs)); | |
| } | |
| if (isset($options['interactive'])) { | |
| $this->input->setInteractive($options['interactive']); | |
| } | |
| $this->output = new StreamOutput(fopen('php://memory', 'w', false)); | |
| $this->output->setDecorated(isset($options['decorated']) ? $options['decorated'] : false); | |
| if (isset($options['verbosity'])) { | |
| $this->output->setVerbosity($options['verbosity']); | |
| } | |
| return $this->statusCode = $this->command->run($this->input, $this->output); | |
| } | |
| /** | |
| * Gets the display returned by the last execution of the command. | |
| * | |
| * @param bool $normalize Whether to normalize end of lines to \n or not | |
| * | |
| * @return string The display | |
| */ | |
| public function getDisplay($normalize = false) | |
| { | |
| rewind($this->output->getStream()); | |
| $display = stream_get_contents($this->output->getStream()); | |
| if ($normalize) { | |
| $display = str_replace(PHP_EOL, "\n", $display); | |
| } | |
| return $display; | |
| } | |
| /** | |
| * Gets the input instance used by the last execution of the command. | |
| * | |
| * @return InputInterface The current input instance | |
| */ | |
| public function getInput() | |
| { | |
| return $this->input; | |
| } | |
| /** | |
| * Gets the output instance used by the last execution of the command. | |
| * | |
| * @return OutputInterface The current output instance | |
| */ | |
| public function getOutput() | |
| { | |
| return $this->output; | |
| } | |
| /** | |
| * Gets the status code returned by the last execution of the application. | |
| * | |
| * @return int The status code | |
| */ | |
| public function getStatusCode() | |
| { | |
| return $this->statusCode; | |
| } | |
| /** | |
| * Sets the user inputs. | |
| * | |
| * @param array An array of strings representing each input | |
| * passed to the command input stream. | |
| * | |
| * @return CommandTester | |
| */ | |
| public function setInputs(array $inputs) | |
| { | |
| $this->inputs = $inputs; | |
| return $this; | |
| } | |
| private static function createStream(array $inputs) | |
| { | |
| $stream = fopen('php://memory', 'r+', false); | |
| fwrite($stream, implode(PHP_EOL, $inputs)); | |
| rewind($stream); | |
| return $stream; | |
| } | |
| } | |