75 lines
1.8 KiB
PHP
75 lines
1.8 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()
|
|
{
|
|
if (!$this->currentEnv) {
|
|
$this->currentEnv = new Environment(new ArrayLoader());
|
|
}
|
|
|
|
return $this->currentEnv;
|
|
}
|
|
|
|
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, ';
|
|
}
|
|
}
|