fix unit tests

Troopers · Jun 23, 2017 · e103470 · e103470
1 parent 2cd691a
commit e103470
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/Tests/Handler/AlertifySessionHandlerTest.php b/Tests/Handler/AlertifySessionHandlerTest.php
@@ -29,16 +29,16 @@ public function testDefault()
         /** @var AlertifyHelper $helper */
         $helper = new AlertifyHelper($this->session);
         $handler = new AlertifySessionHandler(
-            $this->getTwigEnvironmentMock(),
             $container->getParameter('troopers_alertify')
         );
+        $twig = $this->getTwigEnvironmentMock();
 
         $helper->congrat('Alert1');
-        $this->assertEquals(1, count(explode(' ', $handler->handle($this->session))));
+        $this->assertEquals(1, count(explode(' ', $handler->handle($this->session, $twig))));
         $helper->congrat('Alert2');
         $helper->congrat('Alert3', ['opt1' => 42]);
         $helper->congrat('Alert4');
-        $this->assertEquals(3, count(explode(' ', $handler->handle($this->session))));
+        $this->assertEquals(3, count(explode(' ', $handler->handle($this->session, $twig))));
     }
 
     /**
@@ -52,6 +52,7 @@ protected function mockSession()
     protected function getTwigEnvironmentMock()
     {
         $twigEnvironment = $this->getMockBuilder('Twig_Environment')
+            ->disableOriginalConstructor()
             ->setMethods(['render'])
             ->getMock();