fix extension and add test

marcw · Apr 10, 2021 · 34166ec · 34166ec
1 parent bb87a02
commit 34166ec
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 1 deletion.
diff --git a/src/Twig/HeroiconsExtension.php b/src/Twig/HeroiconsExtension.php
@@ -3,6 +3,7 @@
 namespace MarcW\Heroicons\Twig;
 
 use MarcW\Heroicons\Heroicons;
+use Twig\TwigFunction;
 use Twig\Extension\AbstractExtension;
 
 class HeroiconsExtension extends AbstractExtension

diff --git a/tests/Twig/HeroiconsExtensionTest.php b/tests/Twig/HeroiconsExtensionTest.php
@@ -2,8 +2,10 @@
 
 namespace MarcW\Heroicons\Tests\Twig;
 
-use PHPUnit\Framework\TestCase;
 use MarcW\Heroicons\Twig\HeroiconsExtension;
+use PHPUnit\Framework\TestCase;
+use Twig\Environment;
+use Twig\Loader\ArrayLoader;
 
 class HeroiconsExtensionTest extends TestCase
 {
@@ -22,4 +24,15 @@ public function testGetHeroiconWithCustomClass()
 
         $this->assertSame(0, strpos($svg, '<svg class="foo bar" xmlns'));
     }
+
+    public function testExtensionIsRegistered()
+    {
+        $twig = new Environment(new ArrayLoader());
+        $twig->addExtension(new HeroiconsExtension());
+
+        $template = $twig->createTemplate('{{ heroicon("academic-cap") }}', 'test');
+        $output = $template->render();
+
+        $this->assertSame(0, strpos($output, '<svg xmlns'));
+    }
 }