public function testAddFlash() { $flashBag = new FlashBag(); $session = $this->getMock('Symfony\\Component\\HttpFoundation\\Session\\Session'); $session->expects($this->once())->method('getFlashBag')->willReturn($flashBag); $container = $this->getMock('Symfony\\Component\\DependencyInjection\\ContainerInterface'); $container->expects($this->at(0))->method('has')->will($this->returnValue(true)); $container->expects($this->at(1))->method('get')->will($this->returnValue($session)); $controller = new TestController(); $controller->setContainer($container); $controller->addFlash('foo', 'bar'); $this->assertSame(array('bar'), $flashBag->get('foo')); }