[Console] Set user inputs from CommandTester

chalasr · chalasr · commit ce8658c83283 · 2016-05-24T22:24:10.000+02:00
Long array syntax

CS improving

Remove useless count check
diff --git a/src/Symfony/Component/Console/Tester/CommandTester.php b/src/Symfony/Component/Console/Tester/CommandTester.php
@@ -16,17 +16,21 @@
 use Symfony\Component\Console\Output\StreamOutput;
 use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Console\Helper\HelperSet;
+use Symfony\Component\Console\Helper\QuestionHelper;
 
 /**
  * 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 $userInputs = array();
     private $statusCode;
 
     /**
@@ -64,6 +68,10 @@ public function execute(array $input, array $options = array())
             $input = array_merge(array('command' => $this->command->getName()), $input);
         }
 
+        if ($this->userInputs) {
+            $this->createInputStream();
+        }
+
         $this->input = new ArrayInput($input);
         if (isset($options['interactive'])) {
             $this->input->setInteractive($options['interactive']);
@@ -129,4 +137,48 @@ public function getStatusCode()
     {
         return $this->statusCode;
     }
+
+    /**
+     * Sets the faked user inputs.
+     *
+     * @param array $inputs
+     *
+     * @return CommandTester
+     */
+    public function setUserInputs(array $inputs)
+    {
+        $this->userInputs = $inputs;
+
+        return $this;
+    }
+
+    /**
+     * Gets the faked user inputs.
+     *
+     * @return array $inputs
+     */
+    public function getUserInputs()
+    {
+        return $this->userInputs;
+    }
+
+    /**
+     * Creates the command input stream.
+     */
+    private function createInputStream()
+    {
+        $stream = fopen('php://memory', 'r+', false);
+
+        fputs($stream, implode("\n", $this->userInputs));
+        rewind($stream);
+
+        if (null === $this->command->getHelperSet()) {
+            $helper = new QuestionHelper();
+            $this->command->setHelperSet(new HelperSet(array($helper)));
+        } else {
+            $helper = $this->command->getHelper('question');
+        }
+
+        $helper->setInputStream($stream);
+    }
 }
diff --git a/src/Symfony/Component/Console/Tests/Tester/CommandTesterTest.php b/src/Symfony/Component/Console/Tests/Tester/CommandTesterTest.php
@@ -15,6 +15,7 @@
 use Symfony\Component\Console\Command\Command;
 use Symfony\Component\Console\Output\Output;
 use Symfony\Component\Console\Tester\CommandTester;
+use Symfony\Component\Console\Question\Question;
 
 class CommandTesterTest extends \PHPUnit_Framework_TestCase
 {
@@ -81,4 +82,21 @@ public function testCommandFromApplication()
         // check that there is no need to pass the command name here
         $this->assertEquals(0, $tester->execute(array()));
     }
+
+    public function testCommandWithUserInputs()
+    {
+        $question = 'What\'s your name?';
+
+        $command = new Command('foo');
+        $command->setCode(function ($input, $output) use ($question, $command) {
+            $command->getHelper('question')->ask($input, $output, new Question($question));
+        });
+
+        $tester = new CommandTester($command);
+        $tester->setUserInputs(array('Bobby'));
+
+        $this->assertEquals(0, $tester->execute(array()));
+        $this->assertEquals($question, $tester->getDisplay(true));
+        $this->assertInternalType('resource', $command->getHelper('question')->getInputStream());
+    }
 }
