public function testGetBlockByType() { $page = new Page(); $block1 = $this->getMockBuilder('Sonata\\PageBundle\\Model\\Block')->getMock(); $block1->expects($this->once())->method('getType')->will($this->returnValue('sonata.page.block.action')); $block2 = $this->getMockBuilder('Sonata\\PageBundle\\Model\\Block')->getMock(); $block2->expects($this->once())->method('getType')->will($this->returnValue('sonata.page.block.container')); $block3 = $this->getMockBuilder('Sonata\\PageBundle\\Model\\Block')->getMock(); $block3->expects($this->once())->method('getType')->will($this->returnValue('sonata.page.block.action')); $page->addBlocks($block1); $page->addBlocks($block2); $page->addBlocks($block3); $types = $page->getBlocksByType('sonata.page.block.action'); $this->assertEquals(2, count($types)); }