/** * Test if correct sub selection is returned * * @test * @return void */ public function correctMediaSelectionIsReturned() { $imageElement1 = new Media(); $imageElement1->setType(Media::MEDIA_TYPE_IMAGE); $imageElement2 = new Media(); $imageElement2->setType(Media::MEDIA_TYPE_IMAGE); $imageElement3 = new Media(); $imageElement3->setType(Media::MEDIA_TYPE_IMAGE); $multimediaElement1 = new Media(); $multimediaElement1->setType(Media::MEDIA_TYPE_MULTIMEDIA); $news = new 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())); }