319 lines
11 KiB
PHP
319 lines
11 KiB
PHP
|
<?php
|
||
|
|
||
|
/*
|
||
|
* This file is part of the Symfony package.
|
||
|
*
|
||
|
* (c) Fabien Potencier <fabien@symfony.com>
|
||
|
*
|
||
|
* For the full copyright and license information, please view the LICENSE
|
||
|
* file that was distributed with this source code.
|
||
|
*/
|
||
|
|
||
|
namespace Symfony\Component\Console\Tests\Helper;
|
||
|
|
||
|
use Symfony\Component\Console\Helper\ProgressBar;
|
||
|
use Symfony\Component\Console\Output\StreamOutput;
|
||
|
|
||
|
class ProgressBarTest extends \PHPUnit_Framework_TestCase
|
||
|
{
|
||
|
protected $lastMessagesLength;
|
||
|
|
||
|
public function testAdvance()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream());
|
||
|
$bar->start();
|
||
|
$bar->advance();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0 [>---------------------------]').
|
||
|
$this->generateOutput(' 1 [->--------------------------]'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testAdvanceWithStep()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream());
|
||
|
$bar->start();
|
||
|
$bar->advance(5);
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0 [>---------------------------]').
|
||
|
$this->generateOutput(' 5 [----->----------------------]'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testAdvanceMultipleTimes()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream());
|
||
|
$bar->start();
|
||
|
$bar->advance(3);
|
||
|
$bar->advance(2);
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0 [>---------------------------]').
|
||
|
$this->generateOutput(' 3 [--->------------------------]').
|
||
|
$this->generateOutput(' 5 [----->----------------------]'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testCustomizations()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 10);
|
||
|
$bar->setBarWidth(10);
|
||
|
$bar->setBarCharacter('_');
|
||
|
$bar->setEmptyBarCharacter(' ');
|
||
|
$bar->setProgressCharacter('/');
|
||
|
$bar->setFormat(' %current%/%max% [%bar%] %percent%%');
|
||
|
$bar->start();
|
||
|
$bar->advance();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/10 [/ ] 0%').
|
||
|
$this->generateOutput(' 1/10 [_/ ] 10%'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testPercent()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 50);
|
||
|
$bar->start();
|
||
|
$bar->display();
|
||
|
$bar->advance();
|
||
|
$bar->advance();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 1/50 [>---------------------------] 2%').
|
||
|
$this->generateOutput(' 2/50 [=>--------------------------] 4%'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testOverwriteWithShorterLine()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 50);
|
||
|
$bar->setFormat(' %current%/%max% [%bar%] %percent%%');
|
||
|
$bar->start();
|
||
|
$bar->display();
|
||
|
$bar->advance();
|
||
|
|
||
|
// set shorter format
|
||
|
$bar->setFormat(' %current%/%max% [%bar%]');
|
||
|
$bar->advance();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 1/50 [>---------------------------] 2%').
|
||
|
$this->generateOutput(' 2/50 [=>--------------------------] '),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testSetCurrentProgress()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 50);
|
||
|
$bar->start();
|
||
|
$bar->display();
|
||
|
$bar->advance();
|
||
|
$bar->setCurrent(15);
|
||
|
$bar->setCurrent(25);
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 1/50 [>---------------------------] 2%').
|
||
|
$this->generateOutput(' 15/50 [========>-------------------] 30%').
|
||
|
$this->generateOutput(' 25/50 [==============>-------------] 50%'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @expectedException \LogicException
|
||
|
* @expectedExceptionMessage You must start the progress bar
|
||
|
*/
|
||
|
public function testSetCurrentBeforeStarting()
|
||
|
{
|
||
|
$bar = new ProgressBar($this->getOutputStream());
|
||
|
$bar->setCurrent(15);
|
||
|
}
|
||
|
|
||
|
/**
|
||
|
* @expectedException \LogicException
|
||
|
* @expectedExceptionMessage You can't regress the progress bar
|
||
|
*/
|
||
|
public function testRegressProgress()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 50);
|
||
|
$bar->start();
|
||
|
$bar->setCurrent(15);
|
||
|
$bar->setCurrent(10);
|
||
|
}
|
||
|
|
||
|
public function testRedrawFrequency()
|
||
|
{
|
||
|
$bar = $this->getMock('Symfony\Component\Console\Helper\ProgressBar', array('display'), array($output = $this->getOutputStream(), 6));
|
||
|
$bar->expects($this->exactly(4))->method('display');
|
||
|
|
||
|
$bar->setRedrawFrequency(2);
|
||
|
$bar->start();
|
||
|
$bar->setCurrent(1);
|
||
|
$bar->advance(2);
|
||
|
$bar->advance(2);
|
||
|
$bar->advance(1);
|
||
|
}
|
||
|
|
||
|
public function testMultiByteSupport()
|
||
|
{
|
||
|
if (!function_exists('mb_strlen') || (false === $encoding = mb_detect_encoding('■'))) {
|
||
|
$this->markTestSkipped('The mbstring extension is needed for multi-byte support');
|
||
|
}
|
||
|
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream());
|
||
|
$bar->start();
|
||
|
$bar->setBarCharacter('■');
|
||
|
$bar->advance(3);
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0 [>---------------------------]').
|
||
|
$this->generateOutput(' 3 [■■■>------------------------]'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testClear()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 50);
|
||
|
$bar->start();
|
||
|
$bar->setCurrent(25);
|
||
|
$bar->clear();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/50 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 25/50 [==============>-------------] 50%').
|
||
|
$this->generateOutput(''),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testPercentNotHundredBeforeComplete()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(), 200);
|
||
|
$bar->start();
|
||
|
$bar->display();
|
||
|
$bar->advance(199);
|
||
|
$bar->advance();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/200 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 0/200 [>---------------------------] 0%').
|
||
|
$this->generateOutput(' 199/200 [===========================>] 99%').
|
||
|
$this->generateOutput(' 200/200 [============================] 100%'),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
public function testNonDecoratedOutput()
|
||
|
{
|
||
|
$bar = new ProgressBar($output = $this->getOutputStream(false));
|
||
|
$bar->start();
|
||
|
$bar->advance();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals('', stream_get_contents($output->getStream()));
|
||
|
}
|
||
|
|
||
|
public function testParallelBars()
|
||
|
{
|
||
|
$output = $this->getOutputStream();
|
||
|
$bar1 = new ProgressBar($output, 2);
|
||
|
$bar2 = new ProgressBar($output, 3);
|
||
|
$bar2->setProgressCharacter('#');
|
||
|
$bar3 = new ProgressBar($output);
|
||
|
|
||
|
$bar1->start();
|
||
|
$output->write("\n");
|
||
|
$bar2->start();
|
||
|
$output->write("\n");
|
||
|
$bar3->start();
|
||
|
|
||
|
for ($i = 1; $i <= 3; $i++) {
|
||
|
// up two lines
|
||
|
$output->write("\033[2A");
|
||
|
if ($i <= 2) {
|
||
|
$bar1->advance();
|
||
|
}
|
||
|
$output->write("\n");
|
||
|
$bar2->advance();
|
||
|
$output->write("\n");
|
||
|
$bar3->advance();
|
||
|
}
|
||
|
$output->write("\033[2A");
|
||
|
$output->write("\n");
|
||
|
$output->write("\n");
|
||
|
$bar3->finish();
|
||
|
|
||
|
rewind($output->getStream());
|
||
|
$this->assertEquals(
|
||
|
$this->generateOutput(' 0/2 [>---------------------------] 0%')."\n".
|
||
|
$this->generateOutput(' 0/3 [#---------------------------] 0%')."\n".
|
||
|
rtrim($this->generateOutput(' 0 [>---------------------------]')).
|
||
|
|
||
|
"\033[2A".
|
||
|
$this->generateOutput(' 1/2 [==============>-------------] 50%')."\n".
|
||
|
$this->generateOutput(' 1/3 [=========#------------------] 33%')."\n".
|
||
|
rtrim($this->generateOutput(' 1 [->--------------------------]')).
|
||
|
|
||
|
"\033[2A".
|
||
|
$this->generateOutput(' 2/2 [============================] 100%')."\n".
|
||
|
$this->generateOutput(' 2/3 [==================#---------] 66%')."\n".
|
||
|
rtrim($this->generateOutput(' 2 [-->-------------------------]')).
|
||
|
|
||
|
"\033[2A".
|
||
|
"\n".
|
||
|
$this->generateOutput(' 3/3 [============================] 100%')."\n".
|
||
|
rtrim($this->generateOutput(' 3 [--->------------------------]')).
|
||
|
|
||
|
"\033[2A".
|
||
|
"\n".
|
||
|
"\n".
|
||
|
rtrim($this->generateOutput(' 3 [============================]')),
|
||
|
stream_get_contents($output->getStream())
|
||
|
);
|
||
|
}
|
||
|
|
||
|
protected function getOutputStream($decorated = true)
|
||
|
{
|
||
|
return new StreamOutput(fopen('php://memory', 'r+', false), StreamOutput::VERBOSITY_NORMAL, $decorated);
|
||
|
}
|
||
|
|
||
|
protected function generateOutput($expected)
|
||
|
{
|
||
|
$expectedout = $expected;
|
||
|
|
||
|
if (null !== $this->lastMessagesLength) {
|
||
|
$expectedout = str_pad($expected, $this->lastMessagesLength, "\x20", STR_PAD_RIGHT);
|
||
|
}
|
||
|
|
||
|
$this->lastMessagesLength = strlen($expectedout);
|
||
|
|
||
|
return "\x0D".$expectedout;
|
||
|
}
|
||
|
}
|