71 lines
1.7 KiB
PHP
71 lines
1.7 KiB
PHP
|
<?php
|
||
|
|
||
|
/*
|
||
|
* This file is part of Twig.
|
||
|
*
|
||
|
* (c) Fabien Potencier
|
||
|
*
|
||
|
* For the full copyright and license information, please view the LICENSE
|
||
|
* file that was distributed with this source code.
|
||
|
*/
|
||
|
|
||
|
namespace Twig\Test;
|
||
|
|
||
|
use PHPUnit\Framework\TestCase;
|
||
|
use Twig\Compiler;
|
||
|
use Twig\Environment;
|
||
|
use Twig\Loader\ArrayLoader;
|
||
|
use Twig\Node\Node;
|
||
|
|
||
|
abstract class NodeTestCase extends TestCase
|
||
|
{
|
||
|
/**
|
||
|
* @var Environment
|
||
|
*/
|
||
|
private $currentEnv;
|
||
|
|
||
|
abstract public function getTests();
|
||
|
|
||
|
/**
|
||
|
* @dataProvider getTests
|
||
|
*/
|
||
|
public function testCompile($node, $source, $environment = null, $isPattern = false)
|
||
|
{
|
||
|
$this->assertNodeCompilation($source, $node, $environment, $isPattern);
|
||
|
}
|
||
|
|
||
|
public function assertNodeCompilation($source, Node $node, ?Environment $environment = null, $isPattern = false)
|
||
|
{
|
||
|
$compiler = $this->getCompiler($environment);
|
||
|
$compiler->compile($node);
|
||
|
|
||
|
if ($isPattern) {
|
||
|
$this->assertStringMatchesFormat($source, trim($compiler->getSource()));
|
||
|
} else {
|
||
|
$this->assertEquals($source, trim($compiler->getSource()));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
protected function getCompiler(?Environment $environment = null)
|
||
|
{
|
||
|
return new Compiler($environment ?? $this->getEnvironment());
|
||
|
}
|
||
|
|
||
|
protected function getEnvironment()
|
||
|
{
|
||
|
return $this->currentEnv = new Environment(new ArrayLoader([]));
|
||
|
}
|
||
|
|
||
|
protected function getVariableGetter($name, $line = false)
|
||
|
{
|
||
|
$line = $line > 0 ? "// line $line\n" : '';
|
||
|
|
||
|
return sprintf('%s($context["%s"] ?? null)', $line, $name);
|
||
|
}
|
||
|
|
||
|
protected function getAttributeGetter()
|
||
|
{
|
||
|
return 'CoreExtension::getAttribute($this->env, $this->source, ';
|
||
|
}
|
||
|
}
|