AssertUrlRewriteProductInGrid.php

Go to the documentation of this file.

<?php
namespace Magento\UrlRewrite\Test\Constraint;

use Magento\Catalog\Test\Fixture\Category;
use Magento\Mtf\Constraint\AbstractConstraint;
use Magento\Mtf\Fixture\FixtureInterface;
use Magento\Mtf\Util\Protocol\CurlTransport\WebapiDecorator;
use Magento\UrlRewrite\Test\Page\Adminhtml\UrlRewriteIndex;

class AssertUrlRewriteProductInGrid extends AbstractConstraint
{
    private $parentCategoryIndex;

    private $webApi;

    private $targetPathTemplate = 'catalog/product/view/id/%s/category/%s';

    public function processAssert(
        UrlRewriteIndex $urlRewriteIndex,
        WebapiDecorator $webApi,
        FixtureInterface $product,
        Category $category = null
    ) {
        $this->webApi = $webApi;
        $urlRewriteIndex->open();
        $categories = $product->getDataFieldConfig('category_ids')['source']->getCategories();
        $rootCategoryArray = [];
        foreach ($categories as $index => $category) {
            $parentName = $category->getDataFieldConfig('parent_id')['source']->getParentCategory()->getName();
            $rootCategoryArray[$parentName]['name'] = !empty($category->getUrlKey())
                ? strtolower($category->getUrlKey())
                : strtolower($category->getName());
            $rootCategoryArray[$parentName]['index'] = $index;
        }

        $stores = $product->getDataFieldConfig('website_ids')['source']->getStores();
        foreach ($stores as $store) {
            $rootCategoryName = $store->getDataFieldConfig('group_id')['source']
                ->getStoreGroup()
                ->getDataFieldConfig('root_category_id')['source']
                ->getCategory()
                ->getName();

            $this->parentCategoryIndex = $rootCategoryArray[$rootCategoryName]['index'];

            $storeName = $store->getName();
            $filters = [
                [
                    'request_path' => $product->getUrlKey() . '.html',
                    'store_id' => $storeName
                ],
                [
                    'request_path' => $rootCategoryArray[$rootCategoryName]['name'] . '.html',
                    'store_id' => $storeName
                ],
                [
                    'request_path' =>
                        $rootCategoryArray[$rootCategoryName]['name'] . '/' . $product->getUrlKey() . '.html',
                    'target_path' => $this->getTargetPath($product, $category),
                    'store_id' => $storeName
                ],
            ];
            foreach ($filters as $filter) {
                \PHPUnit\Framework\Assert::assertTrue(
                    $urlRewriteIndex->getUrlRedirectGrid()->isRowVisible($filter, true, false),
                    'URL Rewrite with request path \'' . $filter['request_path'] . '\' is absent in grid.'
                );
            }
        }
    }

    private function getTargetPath(FixtureInterface $product, FixtureInterface $category = null)
    {
        $productId = $product->getId()
            ? $product->getId()
            : $this->retrieveProductBySku($product->getSku())['id'];
        $categoryId = $product->hasData('category_ids')
            ? $this->getCategoryId($product)
            : ($category ? $category->getId() : '');
        return sprintf($this->targetPathTemplate, $productId, $categoryId);
    }

    private function getCategoryId(FixtureInterface $product)
    {
        $productSku = $product->getSku();
        $categoryId = $product->getDataFieldConfig('category_ids')['source']
            ->getCategories()[$this->parentCategoryIndex]->getId();
        $categoryId = $categoryId
            ? $categoryId
            : $this->retrieveProductBySku($productSku)
                ['extension_attributes']['category_links'][$this->parentCategoryIndex]['category_id'];
        return $categoryId;
    }

    public function retrieveProductBySku($sku)
    {
        $url = $_ENV['app_frontend_url'] . 'rest/all/V1/products/' . $sku;
        $this->webApi->write($url, [], WebapiDecorator::GET);
        $response = json_decode($this->webApi->read(), true);
        $this->webApi->close();
        return $response;
    }

    public function toString()
    {
        return 'URL Rewrite is present in grid.';
    }
}