diff --git a/src/Symfony/Bridge/Twig/Extension/TranslationExtension.php b/src/Symfony/Bridge/Twig/Extension/TranslationExtension.php index 3718e9e2c4..2ab3832d89 100644 --- a/src/Symfony/Bridge/Twig/Extension/TranslationExtension.php +++ b/src/Symfony/Bridge/Twig/Extension/TranslationExtension.php @@ -28,10 +28,14 @@ class TranslationExtension extends \Twig_Extension private $translator; private $translationNodeVisitor; - public function __construct(TranslatorInterface $translator) + public function __construct(TranslatorInterface $translator, \Twig_NodeVisitorInterface $translationNodeVisitor = null) { + if (!$translationNodeVisitor) { + $translationNodeVisitor = new TranslationNodeVisitor(); + } + $this->translator = $translator; - $this->translationNodeVisitor = new TranslationNodeVisitor(); + $this->translationNodeVisitor = $translationNodeVisitor; } public function getTranslator()