public function getByIds(array $ids, $locale)
{
$media = [];
$mediaEntities = $this->mediaRepository->findMedia(['pagination' => false, 'ids' => $ids]);
$this->count = count($mediaEntities);
foreach ($mediaEntities as $mediaEntity) {
$media[array_search($mediaEntity->getId(), $ids)] = $this->addFormatsAndUrl(new Media($mediaEntity, $locale, null));
}
ksort($media);
return array_values($media);
}
/** * @dataProvider provideGetByIds */ public function testGetByIds($ids, $media, $result) { $this->mediaRepository->findMedia(Argument::any())->willReturn($media); $this->formatManager->getFormats(Argument::cetera())->willReturn(null); $medias = $this->mediaManager->getByIds($ids, 'en'); for ($i = 0; $i < count($medias); ++$i) { $this->assertEquals($result[$i]->getId(), $medias[$i]->getId()); } }