shlink/module/CLI/test/Command/ListShortcodesCommandTest.php

<?php
namespace ShlinkioTest\Shlink\CLI\Command;

use PHPUnit_Framework_TestCase as TestCase;
use Prophecy\Argument;
use Prophecy\Prophecy\ObjectProphecy;
use Shlinkio\Shlink\CLI\Command\Shortcode\ListShortcodesCommand;
use Shlinkio\Shlink\Core\Entity\ShortUrl;
use Shlinkio\Shlink\Core\Service\ShortUrlServiceInterface;
use Symfony\Component\Console\Application;
use Symfony\Component\Console\Helper\QuestionHelper;
use Symfony\Component\Console\Tester\CommandTester;
use Zend\I18n\Translator\Translator;
use Zend\Paginator\Adapter\ArrayAdapter;
use Zend\Paginator\Paginator;

class ListShortcodesCommandTest extends TestCase
{
    /**
     * @var CommandTester
     */
    protected $commandTester;
    /**
     * @var QuestionHelper
     */
    protected $questionHelper;
    /**
     * @var ObjectProphecy
     */
    protected $shortUrlService;

    public function setUp()
    {
        $this->shortUrlService = $this->prophesize(ShortUrlServiceInterface::class);
        $app = new Application();
        $command = new ListShortcodesCommand($this->shortUrlService->reveal(), Translator::factory([]));
        $app->add($command);

        $this->questionHelper = $command->getHelper('question');
        $this->commandTester = new CommandTester($command);
    }

    /**
     * @test
     */
    public function noInputCallsListJustOnce()
    {
        $this->questionHelper->setInputStream($this->getInputStream('\n'));
        $this->shortUrlService->listShortUrls(1)->willReturn(new Paginator(new ArrayAdapter()))
                                                ->shouldBeCalledTimes(1);

        $this->commandTester->execute(['command' => 'shortcode:list']);
    }

    /**
     * @test
     */
    public function loadingMorePagesCallsListMoreTimes()
    {
        // The paginator will return more than one page for the first 3 times
        $data = [];
        for ($i = 0; $i < 30; $i++) {
            $data[] = new ShortUrl();
        }
        $data = array_chunk($data, 11);

        $questionHelper = $this->questionHelper;
        $that = $this;
        $this->shortUrlService->listShortUrls(Argument::any())->will(function () use (&$data, $questionHelper, $that) {
            $questionHelper->setInputStream($that->getInputStream('y'));
            return new Paginator(new ArrayAdapter(array_shift($data)));
        })->shouldBeCalledTimes(3);

        $this->commandTester->execute(['command' => 'shortcode:list']);
    }

    /**
     * @test
     */
    public function havingMorePagesButAnsweringNoCallsListJustOnce()
    {
        // The paginator will return more than one page
        $data = [];
        for ($i = 0; $i < 30; $i++) {
            $data[] = new ShortUrl();
        }

        $this->questionHelper->setInputStream($this->getInputStream('n'));
        $this->shortUrlService->listShortUrls(Argument::any())->willReturn(new Paginator(new ArrayAdapter($data)))
                                                              ->shouldBeCalledTimes(1);

        $this->commandTester->execute(['command' => 'shortcode:list']);
    }

    /**
     * @test
     */
    public function passingPageWillMakeListStartOnThatPage()
    {
        $page = 5;
        $this->questionHelper->setInputStream($this->getInputStream('\n'));
        $this->shortUrlService->listShortUrls($page)->willReturn(new Paginator(new ArrayAdapter()))
                                                    ->shouldBeCalledTimes(1);

        $this->commandTester->execute([
            'command' => 'shortcode:list',
            '--page' => $page,
        ]);
    }

    protected function getInputStream($inputData)
    {
        $stream = fopen('php://memory', 'r+', false);
        fputs($stream, $inputData);
        rewind($stream);

        return $stream;
    }
}