doctrine2/tests/Doctrine/Tests/ORM/Tools/Console/Command/InfoCommandTest.php

<?php

namespace Doctrine\Tests\ORM\Tools\Console\Command;

use Doctrine\ORM\Tools\Console\Helper\EntityManagerHelper;
use Doctrine\Tests\Models\Cache\AttractionInfo;
use Doctrine\Tests\Models\Cache\City;
use Symfony\Component\Console\Tester\CommandTester;
use Symfony\Component\Console\Helper\HelperSet;
use Symfony\Component\Console\Application;
use Doctrine\Tests\OrmFunctionalTestCase;
use Doctrine\ORM\Tools\Console\Command\InfoCommand;

class InfoCommandTest extends OrmFunctionalTestCase
{
    /**
     * @var \Symfony\Component\Console\Application
     */
    private $application;

    /**
     * @var \Doctrine\ORM\Tools\Console\Command\InfoCommand
     */
    private $command;

    /**
     * @var \Symfony\Component\Console\Tester\CommandTester
     */
    private $tester;

    protected function setUp()
    {
        parent::setUp();

        $this->application = new Application();
        $command           = new InfoCommand();

        $this->application->setHelperSet(
            new HelperSet(['em' => new EntityManagerHelper($this->_em)])
        );

        $this->application->add($command);

        $this->command    = $this->application->find('orm:info');
        $this->tester     = new CommandTester($command);
    }

    public function testListAllClasses()
    {
        $this->tester->execute(['command' => $this->command->getName()]);

        $this->assertContains(AttractionInfo::class, $this->tester->getDisplay());
        $this->assertContains(City::class, $this->tester->getDisplay());
    }
}