leilukin-tumbleblog/includes/lib/Twig/Test/NodeTestCase.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, ';
}
}