CHIEFSOFT\ameye
141 lines
3.4 KiB
PHP
141 lines
3.4 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
/**
|
|
* This file is part of CodeIgniter 4 framework.
|
|
*
|
|
* (c) CodeIgniter Foundation <admin@codeigniter.com>
|
|
*
|
|
* For the full copyright and license information, please view
|
|
* the LICENSE file that was distributed with this source code.
|
|
*/
|
|
|
|
namespace CodeIgniter\Test\Mock;
|
|
|
|
use CodeIgniter\CLI\InputOutput;
|
|
use CodeIgniter\Test\Filters\CITestStreamFilter;
|
|
use CodeIgniter\Test\PhpStreamWrapper;
|
|
use InvalidArgumentException;
|
|
use LogicException;
|
|
|
|
final class MockInputOutput extends InputOutput
|
|
{
|
|
/**
|
|
* String to be entered by the user.
|
|
*
|
|
* @var list<string>
|
|
*/
|
|
private array $inputs = [];
|
|
|
|
/**
|
|
* Output lines.
|
|
*
|
|
* @var array<int, string>
|
|
* @phpstan-var list<string>
|
|
*/
|
|
private array $outputs = [];
|
|
|
|
/**
|
|
* Sets user inputs.
|
|
*
|
|
* @param array<int, string> $inputs
|
|
* @phpstan-param list<string> $inputs
|
|
*/
|
|
public function setInputs(array $inputs): void
|
|
{
|
|
$this->inputs = $inputs;
|
|
}
|
|
|
|
/**
|
|
* Gets the item from the output array.
|
|
*
|
|
* @param int|null $index The output array index. If null, returns all output
|
|
* string. If negative int, returns the last $index-th
|
|
* item.
|
|
*/
|
|
public function getOutput(?int $index = null): string
|
|
{
|
|
if ($index === null) {
|
|
return implode('', $this->outputs);
|
|
}
|
|
|
|
if (array_key_exists($index, $this->outputs)) {
|
|
return $this->outputs[$index];
|
|
}
|
|
|
|
if ($index < 0) {
|
|
$i = count($this->outputs) + $index;
|
|
|
|
if (array_key_exists($i, $this->outputs)) {
|
|
return $this->outputs[$i];
|
|
}
|
|
}
|
|
|
|
throw new InvalidArgumentException(
|
|
'No such index in output: ' . $index . ', the last index is: '
|
|
. (count($this->outputs) - 1)
|
|
);
|
|
}
|
|
|
|
/**
|
|
* Returns the outputs array.
|
|
*/
|
|
public function getOutputs(): array
|
|
{
|
|
return $this->outputs;
|
|
}
|
|
|
|
private function addStreamFilters(): void
|
|
{
|
|
CITestStreamFilter::registration();
|
|
CITestStreamFilter::addOutputFilter();
|
|
CITestStreamFilter::addErrorFilter();
|
|
}
|
|
|
|
private function removeStreamFilters(): void
|
|
{
|
|
CITestStreamFilter::removeOutputFilter();
|
|
CITestStreamFilter::removeErrorFilter();
|
|
}
|
|
|
|
public function input(?string $prefix = null): string
|
|
{
|
|
if ($this->inputs === []) {
|
|
throw new LogicException(
|
|
'No input data. Specifiy input data with `MockInputOutput::setInputs()`.'
|
|
);
|
|
}
|
|
|
|
$input = array_shift($this->inputs);
|
|
|
|
$this->addStreamFilters();
|
|
|
|
PhpStreamWrapper::register();
|
|
PhpStreamWrapper::setContent($input);
|
|
|
|
$userInput = parent::input($prefix);
|
|
$this->outputs[] = CITestStreamFilter::$buffer . $input . PHP_EOL;
|
|
|
|
PhpStreamWrapper::restore();
|
|
|
|
$this->removeStreamFilters();
|
|
|
|
if ($input !== $userInput) {
|
|
throw new LogicException($input . '!==' . $userInput);
|
|
}
|
|
|
|
return $input;
|
|
}
|
|
|
|
public function fwrite($handle, string $string): void
|
|
{
|
|
$this->addStreamFilters();
|
|
|
|
parent::fwrite($handle, $string);
|
|
$this->outputs[] = CITestStreamFilter::$buffer;
|
|
|
|
$this->removeStreamFilters();
|
|
}
|
|
}
|