[6.x] Allow commands to overwrite their input and output handlers

src/Illuminate/Console/Command.php

@@ -377,7 +377,7 @@ public function options()
      * Confirm a question with the user.
      *
      * @param  string  $question
-     * @param  bool    $default
+     * @param  bool  $default
      * @return bool
      */
     public function confirm($question, $default = false)
@@ -433,7 +433,7 @@ public function askWithCompletion($question, $choices, $default = null)
      * Prompt the user for input but hide the answer from the console.
      *
      * @param  string  $question
-     * @param  bool    $fallback
+     * @param  bool  $fallback
      * @return mixed
      */
     public function secret($question, $fallback = true)
@@ -449,10 +449,10 @@ public function secret($question, $fallback = true)
      * Give the user a single choice from an array of answers.
      *
      * @param  string  $question
-     * @param  array   $choices
+     * @param  array  $choices
      * @param  string|null  $default
-     * @param  mixed|null   $attempts
-     * @param  bool|null    $multiple
+     * @param  mixed|null  $attempts
+     * @param  bool|null  $multiple
      * @return string
      */
     public function choice($question, array $choices, $default = null, $attempts = null, $multiple = null)
@@ -467,10 +467,10 @@ public function choice($question, array $choices, $default = null, $attempts = n
     /**
      * Format input to textual table.
      *
-     * @param  array   $headers
+     * @param  array  $headers
      * @param  \Illuminate\Contracts\Support\Arrayable|array  $rows
      * @param  string  $tableStyle
-     * @param  array   $columnStyles
+     * @param  array  $columnStyles
      * @return void
      */
     public function table($headers, $rows, $tableStyle = 'default', array $columnStyles = [])
@@ -588,6 +588,28 @@ public function alert($string)
         $this->output->newLine();
     }
 
+    /**
+     * Set the input interface implementation.
+     *
+     * @param  \Symfony\Component\Console\Input\InputInterface  $input
+     * @return void
+     */
+    public function setInput(InputInterface $input)
+    {
+        $this->input = $input;
+    }
+
+    /**
+     * Set the output interface implementation.
+     *
+     * @param  \Illuminate\Console\OutputStyle  $output
+     * @return void
+     */
+    public function setOutput(OutputStyle $output)
+    {
+        $this->output = $output;
+    }
+
     /**
      * Set the verbosity level.
      *

tests/Console/CommandTest.php

@@ -9,6 +9,7 @@
 use PHPUnit\Framework\TestCase;
 use Symfony\Component\Console\Input\ArrayInput;
 use Symfony\Component\Console\Input\InputArgument;
+use Symfony\Component\Console\Input\InputInterface;
 use Symfony\Component\Console\Input\InputOption;
 use Symfony\Component\Console\Output\NullOutput;
 
@@ -87,4 +88,28 @@ protected function getOptions()
         $this->assertEquals('test-first-option', $command->option('option-one'));
         $this->assertEquals('test-second-option', $command->option('option-two'));
     }
+
+    public function testTheInputSetterOverwrite()
+    {
+        $input = m::mock(InputInterface::class);
+        $input->shouldReceive('hasArgument')->once()->with('foo')->andReturn(false);
+
+        $command = new Command;
+        $command->setInput($input);
+
+        $this->assertFalse($command->hasArgument('foo'));
+    }
+
+    public function testTheOutputSetterOverwrite()
+    {
+        $output = m::mock(OutputStyle::class);
+        $output->shouldReceive('writeln')->once()->withArgs(function (...$args) {
+            return $args[0] === '<info>foo</info>';
+        });
+
+        $command = new Command;
+        $command->setOutput($output);
+
+        $command->info('foo');
+    }
 }