ImagesAndVideos.php

Go to the documentation of this file.

<?php
namespace Magento\ProductVideo\Test\Block\Adminhtml\Product\Edit\Tab;

use Magento\Backend\Test\Block\Widget\Tab;
use Magento\Mtf\Client\Element\SimpleElement;

class ImagesAndVideos extends Tab
{
    private $firstVideoId = 0;

    protected $addVideoButton = '[data-role="add-video-button"]';

    protected $newVideoDialog = '.mage-new-video-dialog';

    protected $imageItem = '.image.item';

    public function getVideoDialog()
    {
        $this->waitForElementVisible($this->newVideoDialog);
        return $this->blockFactory->create(
            \Magento\ProductVideo\Test\Block\Adminhtml\Product\Edit\Tab\Images\VideoDialog::class,
            ['element' => $this->browser->find($this->newVideoDialog)]
        );
    }

    protected function clickAddVideo()
    {
        $this->_rootElement->find($this->addVideoButton)->click();
    }

    public function setFieldsData(array $fields, SimpleElement $element = null)
    {
        if (!array_key_exists('images', $fields['media_gallery']['value'])) {
            return $this;
        }
        if ($fields['media_gallery']['value']['images'] == '') {
            $fields['media_gallery']['value']['images'] = [];
        }
        $currentImages = $this->getImageIds();
        $newImages = array_keys($fields['media_gallery']['value']['images']);
        $updateIds = array_intersect($currentImages, $newImages);
        $addIds = array_diff($newImages, $currentImages);
        $deleteIds = array_diff($currentImages, $newImages);

        foreach ($updateIds as $id) {
            $this->updateVideo($id, $fields['media_gallery']['value']['images'][$id]);
        }

        foreach ($deleteIds as $id) {
            $this->deleteVideo($id);
        }

        foreach ($addIds as $id) {
            $this->addVideo($fields['media_gallery']['value']['images'][$id]);
        }

        return $this;
    }

    protected function addVideo($data)
    {
        $this->clickAddVideo();
        $this->getVideoDialog()->fillForm($data)->clickSaveButton();
    }

    protected function deleteVideo($id)
    {
        $this->clickVideo($id);
        $this->getVideoDialog()->clickDeleteButton();
    }

    protected function updateVideo($id, $data)
    {
        $this->clickVideo($id);
        $this->getVideoDialog()->fillForm($data)->clickEditButton();
    }

    public function getFieldsData($tabFields = null, SimpleElement $element = null)
    {
        $fields = reset($tabFields);
        $name = key($tabFields);
        $formData = [];
        if (empty($fields['value']) || !array_key_exists('images', $fields['value'])) {
            return '';
        }
        if ($fields['value']['images'] == '') {
            $fields['value']['images'] = [];
        }

        $formData[$name]['images'] = [];

        $imageArray = $fields['value']['images'];
        $resetImages = array_flip(array_keys($imageArray));

        foreach ($imageArray as $keyRoot => $fieldSet) {
            $image = $this->_rootElement->find($this->getImageSelector($resetImages[$keyRoot]));
            if ($image) {
                $image->click();
                $videoDialog = $this->getVideoDialog();
                $data = $videoDialog->getVideoInfo();
                foreach (array_keys($fieldSet) as $field) {
                    if (isset($data[$field])) {
                        $formData[$name]['images'][$keyRoot][$field] = $data[$field];
                    }
                }
                $videoDialog->clickCloseButton();
            }
        }

        if (count($formData[$name]['images']) == 0) {
            $formData[$name]['images'] = '';
        }

        return $formData;
    }

    protected function getImageSelector($id)
    {
        ++$id;
        return $this->imageItem . ':nth-child(' . $id . ') .product-image-wrapper';
    }

    protected function getImageIds()
    {
        $images = $this->_rootElement->getElements($this->imageItem);
        return array_keys($images);
    }

    protected function clickVideo($id)
    {
        $this->_rootElement->find($this->getImageSelector($id))->click();
    }

    public function clickFirstVideo()
    {
        $this->_rootElement->find($this->getImageSelector($this->firstVideoId))->click();
        return $this;
    }

    public function deleteFirstVideo()
    {
        $this->deleteVideo($this->firstVideoId);
    }
}