CommandTester.php 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162
  1. <?php
  2. /*
  3. * This file is part of the Symfony package.
  4. *
  5. * (c) Fabien Potencier <fabien@symfony.com>
  6. *
  7. * For the full copyright and license information, please view the LICENSE
  8. * file that was distributed with this source code.
  9. */
  10. namespace Symfony\Component\Console\Tester;
  11. use Symfony\Component\Console\Command\Command;
  12. use Symfony\Component\Console\Input\ArrayInput;
  13. use Symfony\Component\Console\Output\StreamOutput;
  14. use Symfony\Component\Console\Input\InputInterface;
  15. use Symfony\Component\Console\Output\OutputInterface;
  16. /**
  17. * Eases the testing of console commands.
  18. *
  19. * @author Fabien Potencier <fabien@symfony.com>
  20. * @author Robin Chalas <robin.chalas@gmail.com>
  21. */
  22. class CommandTester
  23. {
  24. private $command;
  25. private $input;
  26. private $output;
  27. private $inputs = array();
  28. private $statusCode;
  29. /**
  30. * Constructor.
  31. *
  32. * @param Command $command A Command instance to test
  33. */
  34. public function __construct(Command $command)
  35. {
  36. $this->command = $command;
  37. }
  38. /**
  39. * Executes the command.
  40. *
  41. * Available execution options:
  42. *
  43. * * interactive: Sets the input interactive flag
  44. * * decorated: Sets the output decorated flag
  45. * * verbosity: Sets the output verbosity flag
  46. *
  47. * @param array $input An array of command arguments and options
  48. * @param array $options An array of execution options
  49. *
  50. * @return int The command exit code
  51. */
  52. public function execute(array $input, array $options = array())
  53. {
  54. // set the command name automatically if the application requires
  55. // this argument and no command name was passed
  56. if (!isset($input['command'])
  57. && (null !== $application = $this->command->getApplication())
  58. && $application->getDefinition()->hasArgument('command')
  59. ) {
  60. $input = array_merge(array('command' => $this->command->getName()), $input);
  61. }
  62. $this->input = new ArrayInput($input);
  63. if ($this->inputs) {
  64. $this->input->setStream(self::createStream($this->inputs));
  65. }
  66. if (isset($options['interactive'])) {
  67. $this->input->setInteractive($options['interactive']);
  68. }
  69. $this->output = new StreamOutput(fopen('php://memory', 'w', false));
  70. $this->output->setDecorated(isset($options['decorated']) ? $options['decorated'] : false);
  71. if (isset($options['verbosity'])) {
  72. $this->output->setVerbosity($options['verbosity']);
  73. }
  74. return $this->statusCode = $this->command->run($this->input, $this->output);
  75. }
  76. /**
  77. * Gets the display returned by the last execution of the command.
  78. *
  79. * @param bool $normalize Whether to normalize end of lines to \n or not
  80. *
  81. * @return string The display
  82. */
  83. public function getDisplay($normalize = false)
  84. {
  85. rewind($this->output->getStream());
  86. $display = stream_get_contents($this->output->getStream());
  87. if ($normalize) {
  88. $display = str_replace(PHP_EOL, "\n", $display);
  89. }
  90. return $display;
  91. }
  92. /**
  93. * Gets the input instance used by the last execution of the command.
  94. *
  95. * @return InputInterface The current input instance
  96. */
  97. public function getInput()
  98. {
  99. return $this->input;
  100. }
  101. /**
  102. * Gets the output instance used by the last execution of the command.
  103. *
  104. * @return OutputInterface The current output instance
  105. */
  106. public function getOutput()
  107. {
  108. return $this->output;
  109. }
  110. /**
  111. * Gets the status code returned by the last execution of the application.
  112. *
  113. * @return int The status code
  114. */
  115. public function getStatusCode()
  116. {
  117. return $this->statusCode;
  118. }
  119. /**
  120. * Sets the user inputs.
  121. *
  122. * @param array An array of strings representing each input
  123. * passed to the command input stream.
  124. *
  125. * @return CommandTester
  126. */
  127. public function setInputs(array $inputs)
  128. {
  129. $this->inputs = $inputs;
  130. return $this;
  131. }
  132. private static function createStream(array $inputs)
  133. {
  134. $stream = fopen('php://memory', 'r+', false);
  135. fwrite($stream, implode(PHP_EOL, $inputs));
  136. rewind($stream);
  137. return $stream;
  138. }
  139. }