public function testSaveExistingWithNewMediaGalleryEntries()
{
$newEntriesData = [["label" => "label_text", 'position' => 10, 'disabled' => false, 'types' => ['image', 'small_image'], 'content' => ['name' => 'filename', 'mime_type' => 'image/jpeg', 'entry_data' => 'encoded_content']]];
$this->setupProductMocksForSave();
//media gallery data
$this->productData['media_gallery_entries'] = $newEntriesData;
$this->extensibleDataObjectConverterMock->expects($this->once())->method('toNestedArray')->will($this->returnValue($this->productData));
$this->initializedProductMock->setData('media_gallery', []);
$this->initializedProductMock->expects($this->any())->method('getMediaAttributes')->willReturn(["image" => "imageAttribute", "small_image" => "small_image_attribute"]);
//setup media attribute backend
$mediaTmpPath = '/tmp';
$relativePath = $mediaTmpPath . DIRECTORY_SEPARATOR . 'filename.jpg';
$absolutePath = '/a/b/filename.jpg';
$galleryAttributeBackendMock = $this->getMockBuilder('\\Magento\\Catalog\\Model\\Product\\Attribute\\Backend\\Media')->disableOriginalConstructor()->getMock();
$galleryAttributeBackendMock->expects($this->once())->method('clearMediaAttribute')->with($this->initializedProductMock, ['image', 'small_image']);
$mediaConfigMock = $this->getMockBuilder('Magento\\Catalog\\Model\\Product\\Media\\Config')->disableOriginalConstructor()->getMock();
$mediaConfigMock->expects($this->once())->method('getBaseTmpMediaPath')->willReturn($mediaTmpPath);
$directoryWriteMock = $this->getMockBuilder('\\Magento\\Framework\\Filesystem\\Directory\\WriteInterface')->getMockForAbstractClass();
$this->fileSystemMock->expects($this->once())->method('getDirectoryWrite')->willReturn($directoryWriteMock);
$directoryWriteMock->expects($this->once())->method('create')->with($mediaTmpPath);
$this->mimeTypeExtensionMapMock->expects($this->once())->method('getMimeTypeExtension')->with('image/jpeg')->willReturn("jpg");
$directoryWriteMock->expects($this->once())->method('writeFile')->with($relativePath, false);
//decoded value is false as it contains '_'
$directoryWriteMock->expects($this->once())->method('getAbsolutePath')->willReturn($absolutePath);
$this->initializedProductMock->expects($this->any())->method('getGalleryAttributeBackend')->willReturn($galleryAttributeBackendMock);
$this->initializedProductMock->expects($this->once())->method('getMediaConfig')->willReturn($mediaConfigMock);
//verify new entries
$contentDataObject = $this->getMockBuilder('Magento\\Catalog\\Model\\Product\\Media\\GalleryEntryContent')->disableOriginalConstructor()->setMethods(null)->getMock();
$this->contentFactoryMock->expects($this->once())->method('create')->willReturn($contentDataObject);
$this->contentValidatorMock->expects($this->once())->method('isValid')->willReturn(true);
$imageFileUri = "imageFileUri";
$galleryAttributeBackendMock->expects($this->once())->method('addImage')->with($this->initializedProductMock, $absolutePath, ['image', 'small_image'], true, false)->willReturn($imageFileUri);
$galleryAttributeBackendMock->expects($this->once())->method('updateImage')->with($this->initializedProductMock, $imageFileUri, ['label' => 'label_text', 'position' => 10, 'disabled' => false]);
$this->model->save($this->productMock);
}
public function testGetMimeTypeExtension()
{
$this->assertEquals("jpg", $this->model->getMimeTypeExtension("image/jpeg"));
$this->assertEquals("jpg", $this->model->getMimeTypeExtension("image/jpg"));
$this->assertEquals("png", $this->model->getMimeTypeExtension("image/png"));
$this->assertEquals("gif", $this->model->getMimeTypeExtension("image/gif"));
$this->assertEquals("", $this->model->getMimeTypeExtension("unknown"));
$this->assertEquals("", $this->model->getMimeTypeExtension(null));
}
/**
* @param ProductInterface $product
* @param array $newEntry
* @return $this
* @throws InputException
* @throws StateException
* @throws \Magento\Framework\Exception\LocalizedException
*/
protected function processNewMediaGalleryEntry(ProductInterface $product, array $newEntry)
{
/** @var ProductAttributeMediaGalleryEntryContentInterface $contentDataObject */
$contentDataObject = $newEntry['content'];
if (!$this->contentValidator->isValid($contentDataObject)) {
throw new InputException(__('The image content is not valid.'));
}
$fileContent = @base64_decode($contentDataObject->getEntryData(), true);
$fileName = $contentDataObject->getName();
$mimeType = $contentDataObject->getMimeType();
/** @var \Magento\Catalog\Model\Product\Media\Config $mediaConfig */
$mediaConfig = $product->getMediaConfig();
$mediaTmpPath = $mediaConfig->getBaseTmpMediaPath();
$mediaDirectory = $this->fileSystem->getDirectoryWrite(DirectoryList::MEDIA);
$mediaDirectory->create($mediaTmpPath);
$fileName = $fileName . '.' . $this->mimeTypeExtensionMap->getMimeTypeExtension($mimeType);
$relativeFilePath = $mediaTmpPath . DIRECTORY_SEPARATOR . $fileName;
$absoluteFilePath = $mediaDirectory->getAbsolutePath($relativeFilePath);
$mediaDirectory->writeFile($relativeFilePath, $fileContent);
/** @var \Magento\Catalog\Model\Product\Attribute\Backend\Media $galleryAttributeBackend */
$galleryAttributeBackend = $product->getGalleryAttributeBackend();
if ($galleryAttributeBackend == null) {
throw new StateException(__('Requested product does not support images.'));
}
$imageFileUri = $galleryAttributeBackend->addImage($product, $absoluteFilePath, isset($newEntry['types']) ? $newEntry['types'] : [], true, isset($newEntry['disabled']) ? $newEntry['disabled'] : true);
// Update additional fields that are still empty after addImage call
$galleryAttributeBackend->updateImage($product, $imageFileUri, ['label' => $newEntry['label'], 'position' => $newEntry['position'], 'disabled' => $newEntry['disabled']]);
return $this;
}
