[TwigBundle] fixed usage of getSource in tests

fabpot · fabpot · commit 317d46f24989 · 2016-10-22T08:04:15.000-07:00
diff --git a/src/Symfony/Bundle/TwigBundle/Tests/Loader/FilesystemLoaderTest.php b/src/Symfony/Bundle/TwigBundle/Tests/Loader/FilesystemLoaderTest.php
@@ -17,7 +17,7 @@
 
 class FilesystemLoaderTest extends TestCase
 {
-    public function testGetSource()
+    public function testGetSourceContext()
     {
         $parser = $this->getMock('Symfony\Component\Templating\TemplateNameParserInterface');
         $locator = $this->getMock('Symfony\Component\Config\FileLocatorInterface');
@@ -30,10 +30,10 @@ public function testGetSource()
         $loader->addPath(__DIR__.'/../DependencyInjection/Fixtures/Resources/views', 'namespace');
 
         // Twig-style
-        $this->assertEquals("This is a layout\n", $loader->getSource('@namespace/layout.html.twig'));
+        $this->assertEquals("This is a layout\n", $loader->getSourceContext('@namespace/layout.html.twig')->getCode());
 
         // Symfony-style
-        $this->assertEquals("This is a layout\n", $loader->getSource('TwigBundle::layout.html.twig'));
+        $this->assertEquals("This is a layout\n", $loader->getSourceContext('TwigBundle::layout.html.twig')->getCode());
     }
 
     public function testExists()
diff --git a/src/Symfony/Bundle/TwigBundle/Tests/TokenParser/LegacyRenderTokenParserTest.php b/src/Symfony/Bundle/TwigBundle/Tests/TokenParser/LegacyRenderTokenParserTest.php
@@ -27,7 +27,7 @@ public function testCompile($source, $expected)
     {
         $env = new \Twig_Environment($this->getMock('Twig_LoaderInterface'), array('cache' => false, 'autoescape' => false, 'optimizations' => 0));
         $env->addTokenParser(new RenderTokenParser());
-        $stream = $env->tokenize(new \Twig_Source($source));
+        $stream = $env->tokenize(new \Twig_Source($source, ''));
         $parser = new \Twig_Parser($env);
 
         $this->assertEquals($expected, $parser->parse($stream)->getNode('body')->getNode(0));
diff --git a/src/Symfony/Bundle/WebProfilerBundle/Profiler/TemplateManager.php b/src/Symfony/Bundle/WebProfilerBundle/Profiler/TemplateManager.php
@@ -126,7 +126,11 @@ protected function templateExists($template)
         }
 
         try {
-            $loader->getSource($template);
+            if ($loader instanceof \Twig_SourceContextLoaderInterface) {
+                $loader->getSourceContext($template);
+            } else {
+                $loader->getSource($template);
+            }
 
             return true;
         } catch (\Twig_Error_Loader $e) {
diff --git a/src/Symfony/Bundle/WebProfilerBundle/Tests/Profiler/TemplateManagerTest.php b/src/Symfony/Bundle/WebProfilerBundle/Tests/Profiler/TemplateManagerTest.php
@@ -31,11 +31,6 @@ class TemplateManagerTest extends TestCase
      */
     protected $profiler;
 
-    /**
-     * @var \PHPUnit_Framework_MockObject_MockObject
-     */
-    protected $profile;
-
     /**
      * @var \Symfony\Bundle\WebProfilerBundle\Profiler\TemplateManager
      */
@@ -129,11 +124,7 @@ public function profileHasCollectorCallback($panel)
 
     protected function mockProfile()
     {
-        $this->profile = $this->getMockBuilder('Symfony\Component\HttpKernel\Profiler\Profile')
-            ->disableOriginalConstructor()
-            ->getMock();
-
-        return  $this->profile;
+        return $this->getMockBuilder('Symfony\Component\HttpKernel\Profiler\Profile')->disableOriginalConstructor()->getMock();
     }
 
     protected function mockTwigEnvironment()
@@ -144,9 +135,12 @@ protected function mockTwigEnvironment()
             ->method('loadTemplate')
             ->will($this->returnValue('loadedTemplate'));
 
-        $this->twigEnvironment->expects($this->any())
-            ->method('getLoader')
-            ->will($this->returnValue($this->getMock('\Twig_LoaderInterface')));
+        if (interface_exists('\Twig_SourceContextLoaderInterface')) {
+            $loader = $this->getMock('\Twig_SourceContextLoaderInterface');
+        } else {
+            $loader = $this->getMock('\Twig_LoaderInterface');
+        }
+        $this->twigEnvironment->expects($this->any())->method('getLoader')->will($this->returnValue($loader));
 
         return $this->twigEnvironment;
     }

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@`
`17`	`17`
`18`	`18`	`class FilesystemLoaderTest extends TestCase`
`19`	`19`	`{`
`20`		`- public function testGetSource()`
	`20`	`+ public function testGetSourceContext()`
`21`	`21`	`{`
`22`	`22`	`$parser = $this->getMock('Symfony\Component\Templating\TemplateNameParserInterface');`
`23`	`23`	`$locator = $this->getMock('Symfony\Component\Config\FileLocatorInterface');`
`@@ -30,10 +30,10 @@ public function testGetSource()`
`30`	`30`	`$loader->addPath(__DIR__.'/../DependencyInjection/Fixtures/Resources/views', 'namespace');`
`31`	`31`
`32`	`32`	`// Twig-style`
`33`		`- $this->assertEquals("This is a layout\n", $loader->getSource('@namespace/layout.html.twig'));`
	`33`	`+ $this->assertEquals("This is a layout\n", $loader->getSourceContext('@namespace/layout.html.twig')->getCode());`
`34`	`34`
`35`	`35`	`// Symfony-style`
`36`		`- $this->assertEquals("This is a layout\n", $loader->getSource('TwigBundle::layout.html.twig'));`
	`36`	`+ $this->assertEquals("This is a layout\n", $loader->getSourceContext('TwigBundle::layout.html.twig')->getCode());`
`37`	`37`	`}`
`38`	`38`
`39`	`39`	`public function testExists()`
Original file line number	Diff line number	Diff line change
`@@ -27,7 +27,7 @@ public function testCompile($source, $expected)`
`27`	`27`	`{`
`28`	`28`	`$env = new \Twig_Environment($this->getMock('Twig_LoaderInterface'), array('cache' => false, 'autoescape' => false, 'optimizations' => 0));`
`29`	`29`	`$env->addTokenParser(new RenderTokenParser());`
`30`		`- $stream = $env->tokenize(new \Twig_Source($source));`
	`30`	`+ $stream = $env->tokenize(new \Twig_Source($source, ''));`
`31`	`31`	`$parser = new \Twig_Parser($env);`
`32`	`32`
`33`	`33`	`$this->assertEquals($expected, $parser->parse($stream)->getNode('body')->getNode(0));`