diff options
Diffstat (limited to 'tests/View/TwigTest.php')
-rw-r--r-- | tests/View/TwigTest.php | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/tests/View/TwigTest.php b/tests/View/TwigTest.php index 2792a03..662681d 100644 --- a/tests/View/TwigTest.php +++ b/tests/View/TwigTest.php @@ -1,5 +1,6 @@ <?php +use Jasny\Flash; use Jasny\Controller\View\Twig; use Psr\Http\Message\ServerRequestInterface; use Psr\Http\Message\ResponseInterface; @@ -16,7 +17,7 @@ class TwigTest extends PHPUnit_Framework_TestCase */ public function testGetTwig() { - $view = $this->getView(['getTwigLoader', 'getTwigEnvironment', 'createTwigFunction', 'createTwigFilter', 'setViewExtension']); + $view = $this->getView(['getTwigLoader', 'getTwigEnvironment', 'createTwigFunction', 'createTwigFilter', 'setViewExtension', 'setViewVariable']); list($request, $response) = $this->getRequests(); list($loader, $env) = $this->getTwigObjects(); @@ -32,7 +33,12 @@ class TwigTest extends PHPUnit_Framework_TestCase $view->expects($this->once())->method('getRequest')->will($this->returnValue($request)); $request->expects($this->once())->method('getUri')->will($this->returnValue($uri)); $uri->expects($this->once())->method('getPath')->will($this->returnValue($path)); - $env->expects($this->once())->method('addGlobal')->with($this->equalTo('current_url'), $this->equalTo($path)); + $view->expects($this->exactly(2))->method('setViewVariable')->withConsecutive( + [$this->equalTo('current_url'), $this->equalTo($path)], + [$this->equalTo('flash'), $this->callback(function($flash) { + return $flash instanceof Flash && empty($flash->get()); + })] + ); $result = $view->getTwig(); $resultSaved = $view->getTwig(); |