symfony/tests/Symfony/Tests/Components/Console/Output/OutputTest.php

<?php

/*
 * This file is part of the symfony package.
 * (c) Fabien Potencier <fabien.potencier@symfony-project.com>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

namespace Symfony\Tests\Components\Console\Output;

use Symfony\Components\Console\Output\Output;

class OutputTest extends \PHPUnit_Framework_TestCase
{
  public function testConstructor()
  {
    $output = new TestOutput(Output::VERBOSITY_QUIET, true);
    $this->assertEquals(Output::VERBOSITY_QUIET, $output->getVerbosity(), '__construct() takes the verbosity as its first argument');
    $this->assertTrue($output->isDecorated(), '__construct() takes the decorated flag as its second argument');
  }

  public function testSetIsDecorated()
  {
    $output = new TestOutput();
    $output->setDecorated(true);
    $this->assertTrue($output->isDecorated(), 'setDecorated() sets the decorated flag');
  }

  public function testSetGetVerbosity()
  {
    $output = new TestOutput();
    $output->setVerbosity(Output::VERBOSITY_QUIET);
    $this->assertEquals(Output::VERBOSITY_QUIET, $output->getVerbosity(), '->setVerbosity() sets the verbosity');
  }

  public function testSetStyle()
  {
    Output::setStyle('FOO', array('bg' => 'red', 'fg' => 'yellow', 'blink' => true));
    $this->assertEquals(array('bg' => 'red', 'fg' => 'yellow', 'blink' => true), TestOutput::getStyle('foo'), '::setStyle() sets a new style');
  }

  public function testWrite()
  {
    $output = new TestOutput(Output::VERBOSITY_QUIET);
    $output->writeln('foo');
    $this->assertEquals('', $output->output, '->writeln() outputs nothing if verbosity is set to VERBOSITY_QUIET');

    $output = new TestOutput();
    $output->writeln(array('foo', 'bar'));
    $this->assertEquals("foo\nbar\n", $output->output, '->writeln() can take an array of messages to output');

    $output = new TestOutput();
    $output->writeln('<info>foo</info>', Output::OUTPUT_RAW);
    $this->assertEquals("<info>foo</info>\n", $output->output, '->writeln() outputs the raw message if OUTPUT_RAW is specified');

    $output = new TestOutput();
    $output->writeln('<info>foo</info>', Output::OUTPUT_PLAIN);
    $this->assertEquals("foo\n", $output->output, '->writeln() strips decoration tags if OUTPUT_PLAIN is specified');

    $output = new TestOutput();
    $output->setDecorated(false);
    $output->writeln('<info>foo</info>');
    $this->assertEquals("foo\n", $output->output, '->writeln() strips decoration tags if decoration is set to false');

    $output = new TestOutput();
    $output->setDecorated(true);
    $output->writeln('<foo>foo</foo>');
    $this->assertEquals("\033[33;41;5mfoo\033[0m\n", $output->output, '->writeln() decorates the output');

    try
    {
      $output->writeln('<foo>foo</foo>', 24);
      $this->fail('->writeln() throws an \InvalidArgumentException when the type does not exist');
    }
    catch (\Exception $e)
    {
      $this->assertType('\InvalidArgumentException', $e, '->writeln() throws an \InvalidArgumentException when the type does not exist');
      $this->assertEquals('Unknown output type given (24)', $e->getMessage());
    }

    try
    {
      $output->writeln('<bar>foo</bar>');
      $this->fail('->writeln() throws an \InvalidArgumentException when a style does not exist');
    }
    catch (\Exception $e)
    {
      $this->assertType('\InvalidArgumentException', $e, '->writeln() throws an \InvalidArgumentException when a style does not exist');
      $this->assertEquals('Unknown style "bar".', $e->getMessage());
    }
  }
}

class TestOutput extends Output
{
  public $output = '';

  static public function getStyle($name)
  {
    return static::$styles[$name];
  }

  public function doWrite($message, $newline)
  {
    $this->output .= $message.($newline ? "\n" : '');
  }
}