[Console] Add SymfonyStyle::setInputStream for command testing

chalasr · chalasr · commit dfd22bb012b8 · 2016-05-29T17:51:13.000+02:00
Use Console InvalidArgumentException, test the exception

Use better phpdoc block

Fabbot fixes

Don't check stream type in SymfonyStyle::setInputStream

Test output of an interactive command with questions
diff --git a/src/Symfony/Component/Console/Style/SymfonyStyle.php b/src/Symfony/Component/Console/Style/SymfonyStyle.php
@@ -40,6 +40,7 @@ class SymfonyStyle extends OutputStyle
     private $progressBar;
     private $lineLength;
     private $bufferedOutput;
+    private $inputStream;
 
     /**
      * @param InputInterface  $input
@@ -348,6 +349,10 @@ public function askQuestion(Question $question)
 
         if (!$this->questionHelper) {
             $this->questionHelper = new SymfonyQuestionHelper();
+
+            if ($this->inputStream) {
+                $this->questionHelper->setInputStream($this->inputStream);
+            }
         }
 
         $answer = $this->questionHelper->ask($this->input, $this, $question);
@@ -387,6 +392,18 @@ public function newLine($count = 1)
         $this->bufferedOutput->write(str_repeat("\n", $count));
     }
 
+    /**
+     * Sets the input stream to read from when interacting with the user.
+     *
+     * This is mainly useful for testing purpose.
+     *
+     * @param resource $stream The input stream
+     */
+    public function setInputStream($stream)
+    {
+        $this->inputStream = $stream;
+    }
+
     /**
      * @return ProgressBar
      */
diff --git a/src/Symfony/Component/Console/Tests/Fixtures/Style/SymfonyStyle/command/interactive_command_1.php b/src/Symfony/Component/Console/Tests/Fixtures/Style/SymfonyStyle/command/interactive_command_1.php
@@ -0,0 +1,25 @@
+<?php
+
+use Symfony\Component\Console\Input\InputInterface;
+use Symfony\Component\Console\Output\OutputInterface;
+use Symfony\Component\Console\Tests\Style\SymfonyStyleWithForcedLineLength;
+
+//Ensure that questions have the expected outputs
+return function (InputInterface $input, OutputInterface $output) {
+    $output = new SymfonyStyleWithForcedLineLength($input, $output);
+    $questions = array(
+        'What\'s your name?',
+        'How are you?',
+        'Where do you come from?',
+    );
+    $inputs = ['Foo', 'Bar', 'Baz'];
+    $stream = fopen('php://memory', 'r+', false);
+
+    fputs($stream, implode(PHP_EOL, $inputs));
+    rewind($stream);
+
+    $output->setInputStream($stream);
+    $output->ask($questions[0]);
+    $output->ask($questions[1]);
+    $output->ask($questions[2]);
+};
diff --git a/src/Symfony/Component/Console/Tests/Fixtures/Style/SymfonyStyle/output/interactive_output_1.txt b/src/Symfony/Component/Console/Tests/Fixtures/Style/SymfonyStyle/output/interactive_output_1.txt
@@ -0,0 +1,7 @@
+
+ What's your name?:
+ > 
+ How are you?:
+ > 
+ Where do you come from?:
+ > 
diff --git a/src/Symfony/Component/Console/Tests/Style/SymfonyStyleTest.php b/src/Symfony/Component/Console/Tests/Style/SymfonyStyleTest.php
@@ -55,6 +55,24 @@ public function inputCommandToOutputFilesProvider()
         return array_map(null, glob($baseDir.'/command/command_*.php'), glob($baseDir.'/output/output_*.txt'));
     }
 
+    /**
+     * @dataProvider inputInteractiveCommandToOutputFilesProvider
+     */
+    public function testInteractiveOutputs($inputCommandFilepath, $outputFilepath)
+    {
+        $code = require $inputCommandFilepath;
+        $this->command->setCode($code);
+        $this->tester->execute(array());
+        $this->assertStringEqualsFile($outputFilepath, $this->tester->getDisplay(true));
+    }
+
+    public function inputInteractiveCommandToOutputFilesProvider()
+    {
+        $baseDir = __DIR__.'/../Fixtures/Style/SymfonyStyle';
+
+        return array_map(null, glob($baseDir.'/command/interactive_command_*.php'), glob($baseDir.'/output/interactive_output_*.txt'));
+    }
+
     public function testLongWordsBlockWrapping()
     {
         $word = 'Lopadotemachoselachogaleokranioleipsanodrimhypotrimmatosilphioparaomelitokatakechymenokichlepikossyphophattoperisteralektryonoptekephalliokigklopeleiolagoiosiraiobaphetraganopterygovgollhjvhvljfezefeqifzeiqgiqzhrsdgihqzridghqridghqirshdghdghieridgheirhsdgehrsdvhqrsidhqshdgihrsidvqhneriqsdvjzergetsrfhgrstsfhsetsfhesrhdgtesfhbzrtfbrztvetbsdfbrsdfbrn';
@@ -70,6 +88,28 @@ public function testLongWordsBlockWrapping()
         $expectedCount = (int) ceil($wordLength / ($maxLineLength)) + (int) ($wordLength > $maxLineLength - 5);
         $this->assertSame($expectedCount, substr_count($this->tester->getDisplay(true), ' § '));
     }
+
+    public function testSetInputStream()
+    {
+        $command = $this->command;
+        $inputs = ['Foo', 'Bar', 'Baz'];
+        $stream = fopen('php://memory', 'r+', false);
+
+        fputs($stream, implode(PHP_EOL, $inputs));
+        rewind($stream);
+
+        $command->setCode(function ($input, $output) use ($command, $stream) {
+            $sfStyle = new SymfonyStyle($input, $output);
+
+            $sfStyle->setInputStream($stream);
+            $sfStyle->ask('What\'s your name?');
+            $sfStyle->ask('How are you?');
+            $sfStyle->ask('Where do you come from?');
+        });
+
+        $this->tester->execute(array());
+        $this->assertSame(0, $this->tester->getStatusCode());
+    }
 }
 
 /**
