Update Twig extension for Twig 2. Add unit tests

Author: mheap

composer.json

@@ -17,6 +17,7 @@
         }
     },
     "require-dev": {
-        "phpunit/phpunit": "^4.8"
+        "phpunit/phpunit": "^4.8",
+        "twig/twig": "^2.4"
     }
 }

src/Gravatar/Extension/Twig/GravatarExtension.php

@@ -22,8 +22,8 @@ public function __construct(Service $service)
     public function getFunctions()
     {
         return array(
-            'gravatar'          => new \Twig_Function_Method($this, 'get'),
-            'gravatar_exist'    => new \Twig_Function_Method($this, 'exist'),
+            'gravatar'          => new \Twig_Function('gravatar', array($this, 'get'), array('is_safe' => array('html'))),
+            'gravatar_exist'    => new \Twig_Function('gravatar_exist', array($this, 'exist')),
         );
     }
 

tests/Gravatar/Tests/TwigExtensionTest.php

@@ -0,0 +1,51 @@
+<?php
+
+namespace Gravatar\Tests;
+
+use Gravatar\Service;
+use Gravatar\Extension\Twig\GravatarExtension;
+
+use Twig_Environment;
+use Twig_Loader_Array;
+
+class TwigExtensionTest extends \PHPUnit_Framework_TestCase
+{
+    public function testRegisterExtension()
+    {
+        $twig = new Twig_Environment(new Twig_Loader_Array(array()));
+        $gravatarService = new Service();
+        $twig->addExtension(new GravatarExtension($gravatarService));
+    }
+
+    public function testRenderGravatarUrl()
+    {
+        $twig = new Twig_Environment(new Twig_Loader_Array(array(
+            'index.html' => '{{gravatar(email, {"size": 50})}}',
+        )));
+
+        $gravatarService = new Service();
+
+        $twig->addExtension(new GravatarExtension($gravatarService));
+
+        $url = $twig->render('index.html', array(
+            'email' => '[email protected]'
+        ));
+
+        $this->assertEquals('http://www.gravatar.com/avatar/b58996c504c5638798eb6b511e6f49af?s=50&r=g', $url);
+    }
+
+    public function testRenderGravatarExists()
+    {
+        $twig = new Twig_Environment(new Twig_Loader_Array(array(
+            'exists' => '{{gravatar_exist(email)}}',
+        )));
+
+        $gravatarService = new Service();
+
+        $twig->addExtension(new GravatarExtension($gravatarService));
+
+        $this->assertEquals("", $twig->render('exists', array('email' => '[email protected]')));
+        $this->assertEquals("1", $twig->render('exists', array('email' => '[email protected]')));
+    }
+
+}

tests/bootstrap.php

@@ -1,3 +1,3 @@
 <?php
 
-require_once __DIR__ . '/../src/Gravatar/Service.php';
+require_once __DIR__ . '/../vendor/autoload.php';