UpdateTopCategoryEntityTest.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Test\TestCase\Category;

use Magento\Catalog\Test\Fixture\Category;
use Magento\Catalog\Test\Page\Adminhtml\CatalogCategoryEdit;
use Magento\Catalog\Test\Page\Adminhtml\CatalogCategoryIndex;
use Magento\Mtf\TestCase\Injectable;
use Magento\Mtf\Fixture\FixtureFactory;

class UpdateTopCategoryEntityTest extends Injectable
{
    /* tags */
    const MVP = 'yes';
    /* end tags */

    protected $catalogCategoryIndex;

    protected $catalogCategoryEdit;

    protected $fixtureFactory;

    public function __inject(
        CatalogCategoryIndex $catalogCategoryIndex,
        CatalogCategoryEdit $catalogCategoryEdit,
        FixtureFactory $fixtureFactory
    ) {
        $this->fixtureFactory = $fixtureFactory;
        $this->catalogCategoryIndex = $catalogCategoryIndex;
        $this->catalogCategoryEdit = $catalogCategoryEdit;
    }

    public function test(
        Category $category,
        Category $initialCategory,
        $nestingLevel
    ) {
        $initialCategory->persist();
        $topCategory =  $this->getParentCategoryByNestingLevel($initialCategory, $nestingLevel);
        $this->catalogCategoryIndex->open();
        $this->catalogCategoryIndex->getTreeCategories()->selectCategory($topCategory);
        $this->catalogCategoryEdit->getEditForm()->fill($category);
        $this->catalogCategoryEdit->getFormPageActions()->save();

        $categories = [];
        $categoriesBeforeSave = [];
        $this->getCategoryFixture($categories, $initialCategory, $category->getData(), $nestingLevel);
        $this->getCategory($initialCategory, $categoriesBeforeSave, $nestingLevel);

        return [
            'categories' => $categories,
            'categoriesBeforeSave' => $categoriesBeforeSave
        ];
    }

    private function getCategoryFixture(array &$categories, Category $currentCategory, array $data, int $nestingLevel)
    {
        if (--$nestingLevel) {
            $parentCategory = $this->getCategoryFixture(
                $categories,
                $currentCategory->getDataFieldConfig('parent_id')['source']->getParentCategory(),
                $data,
                $nestingLevel
            );
            $category = $this->fixtureFactory->createByCode(
                'category',
                ['data' => array_merge($currentCategory->getData(), ['parent_id' => ['source' => $parentCategory]])]
            );
        } else {
            $category = $this->fixtureFactory->createByCode(
                'category',
                ['data' => array_merge($currentCategory->getData(), $data)]
            );
        }
        $categories[$nestingLevel + 1] = $category;
        return $category;
    }

    private function getCategory(Category $initialCategory, &$categoriesBeforeSave, $nestingLevel)
    {
        if (--$nestingLevel) {
            $parentCategory = $this->getCategory(
                $initialCategory->getDataFieldConfig('parent_id')['source']->getParentCategory(),
                $categoriesBeforeSave,
                $nestingLevel
            );
            $category = $this->fixtureFactory->createByCode(
                'category',
                ['data' => array_merge($initialCategory->getData(), ['parent_id' => ['source' => $parentCategory]])]
            );
        } else {
            $category = $initialCategory;
        }
        $categoriesBeforeSave[$nestingLevel + 1] = $category;
        return $category;
    }

    private function getParentCategoryByNestingLevel(Category $category, $nestingLevel)
    {
        for ($nestingIterator = 1; $nestingIterator < $nestingLevel; $nestingIterator++) {
            $category = $category->getDataFieldConfig('parent_id')['source']->getParentCategory();
        }

        return $category;
    }
}