private function getSession($flashBag = null) { $session = $this->createMock('Symfony\\Component\\HttpFoundation\\Session\\Session'); $session->expects($this->any())->method('getFlashBag')->willReturn($flashBag ? $flashBag : $this->getFlashBag()); return $session; }