vendor/twig/twig/test/Twig/Tests/NodeVisitor/OptimizerTest.php

   1 <?php
   2
   3 /*
   4  * This file is part of Twig.
   5  *
   6  * (c) Fabien Potencier
   7  *
   8  * For the full copyright and license information, please view the LICENSE
   9  * file that was distributed with this source code.
  10  */
  11 class Twig_Tests_NodeVisitor_OptimizerTest extends PHPUnit_Framework_TestCase
  12 {
  13     public function testRenderBlockOptimizer()
  14     {
  15         $env = new Twig_Environment(new Twig_Loader_String(), array('cache' => false, 'autoescape' => false));
  16
  17         $stream = $env->parse($env->tokenize('{{ block("foo") }}', 'index'));
  18
  19         $node = $stream->getNode('body')->getNode(0);
  20
  21         $this->assertEquals('Twig_Node_Expression_BlockReference', get_class($node));
  22         $this->assertTrue($node->getAttribute('output'));
  23     }
  24
  25     public function testRenderParentBlockOptimizer()
  26     {
  27         $env = new Twig_Environment(new Twig_Loader_String(), array('cache' => false, 'autoescape' => false));
  28
  29         $stream = $env->parse($env->tokenize('{% extends "foo" %}{% block content %}{{ parent() }}{% endblock %}', 'index'));
  30
  31         $node = $stream->getNode('blocks')->getNode('content')->getNode(0)->getNode('body');
  32
  33         $this->assertEquals('Twig_Node_Expression_Parent', get_class($node));
  34         $this->assertTrue($node->getAttribute('output'));
  35     }
  36
  37     public function testRenderVariableBlockOptimizer()
  38     {
  39         if (version_compare(phpversion(), '5.4.0RC1', '>=')) {
  40             return;
  41         }
  42
  43         $env = new Twig_Environment(new Twig_Loader_String(), array('cache' => false, 'autoescape' => false));
  44         $stream = $env->parse($env->tokenize('{{ block(name|lower) }}', 'index'));
  45
  46         $node = $stream->getNode('body')->getNode(0)->getNode(1);
  47
  48         $this->assertEquals('Twig_Node_Expression_BlockReference', get_class($node));
  49         $this->assertTrue($node->getAttribute('output'));
  50     }
  51
  52     /**
  53      * @dataProvider getTestsForForOptimizer
  54      */
  55     public function testForOptimizer($template, $expected)
  56     {
  57         $env = new Twig_Environment(new Twig_Loader_String(), array('cache' => false));
  58
  59         $stream = $env->parse($env->tokenize($template, 'index'));
  60
  61         foreach ($expected as $target => $withLoop) {
  62             $this->assertTrue($this->checkForConfiguration($stream, $target, $withLoop), sprintf('variable %s is %soptimized', $target, $withLoop ? 'not ' : ''));
  63         }
  64     }
  65
  66     public function getTestsForForOptimizer()
  67     {
  68         return array(
  69             array('{% for i in foo %}{% endfor %}', array('i' => false)),
  70
  71             array('{% for i in foo %}{{ loop.index }}{% endfor %}', array('i' => true)),
  72
  73             array('{% for i in foo %}{% for j in foo %}{% endfor %}{% endfor %}', array('i' => false, 'j' => false)),
  74
  75             array('{% for i in foo %}{% include "foo" %}{% endfor %}', array('i' => true)),
  76
  77             array('{% for i in foo %}{% include "foo" only %}{% endfor %}', array('i' => false)),
  78
  79             array('{% for i in foo %}{% include "foo" with { "foo": "bar" } only %}{% endfor %}', array('i' => false)),
  80
  81             array('{% for i in foo %}{% include "foo" with { "foo": loop.index } only %}{% endfor %}', array('i' => true)),
  82
  83             array('{% for i in foo %}{% for j in foo %}{{ loop.index }}{% endfor %}{% endfor %}', array('i' => false, 'j' => true)),
  84
  85             array('{% for i in foo %}{% for j in foo %}{{ loop.parent.loop.index }}{% endfor %}{% endfor %}', array('i' => true, 'j' => true)),
  86
  87             array('{% for i in foo %}{% set l = loop %}{% for j in foo %}{{ l.index }}{% endfor %}{% endfor %}', array('i' => true, 'j' => false)),
  88
  89             array('{% for i in foo %}{% for j in foo %}{{ foo.parent.loop.index }}{% endfor %}{% endfor %}', array('i' => false, 'j' => false)),
  90
  91             array('{% for i in foo %}{% for j in foo %}{{ loop["parent"].loop.index }}{% endfor %}{% endfor %}', array('i' => true, 'j' => true)),
  92         );
  93     }
  94
  95     public function checkForConfiguration(Twig_NodeInterface $node = null, $target, $withLoop)
  96     {
  97         if (null === $node) {
  98             return;
  99         }
 100
 101         foreach ($node as $n) {
 102             if ($n instanceof Twig_Node_For) {
 103                 if ($target === $n->getNode('value_target')->getAttribute('name')) {
 104                     return $withLoop == $n->getAttribute('with_loop');
 105                 }
 106             }
 107
 108             $ret = $this->checkForConfiguration($n, $target, $withLoop);
 109             if (null !== $ret) {
 110                 return $ret;
 111             }
 112         }
 113     }
 114 }