/** * Test if correct sub selection is returned * * @test * @return void */ public function correctMediaSelectionIsReturned() { $imageElement1 = new Tx_News_Domain_Model_Media(); $imageElement1->setType(Tx_News_Domain_Model_Media::MEDIA_TYPE_IMAGE); $imageElement2 = new Tx_News_Domain_Model_Media(); $imageElement2->setType(Tx_News_Domain_Model_Media::MEDIA_TYPE_IMAGE); $imageElement3 = new Tx_News_Domain_Model_Media(); $imageElement3->setType(Tx_News_Domain_Model_Media::MEDIA_TYPE_IMAGE); $multimediaElement1 = new Tx_News_Domain_Model_Media(); $multimediaElement1->setType(Tx_News_Domain_Model_Media::MEDIA_TYPE_MULTIMEDIA); $news = new Tx_News_Domain_Model_News(); $news->addMedia($imageElement1); $news->addMedia($imageElement2); $news->addMedia($imageElement3); $news->addMedia($imageElement3); $news->addMedia($multimediaElement1); $this->assertEquals(3, count($news->getMediaTypeImage())); $this->assertEquals(1, count($news->getMediaTypeMultimedia())); $this->assertEquals(4, count($news->getMedia())); }