CreateFlatCatalogProduct.php

Go to the documentation of this file.

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

use Magento\Config\Test\TestStep\SetupConfigurationStep;
use Magento\Catalog\Test\Page\Category\CatalogCategoryView;
use Magento\Mtf\Fixture\FixtureFactory;
use Magento\Mtf\TestCase\Injectable;
use Magento\Catalog\Test\Fixture\Category;

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

    private $configData;

    private $fixtureFactory;

    private $category;

    private $catalogCategoryView;

    public function __prepare(Category $category)
    {
        $category->persist();
        return [
            'category' => $category
        ];
    }

    public function __inject(
        Category $category,
        FixtureFactory $fixtureFactory,
        CatalogCategoryView $catalogCategoryView
    ) {
        $this->category = $category;
        $this->fixtureFactory = $fixtureFactory;
        $this->catalogCategoryView = $catalogCategoryView;
    }

    public function test($configData, $productsCount)
    {
        $this->objectManager->create(SetupConfigurationStep::class, ['configData' => $this->configData])->run();
        $this->createBulkOfProducts($productsCount);
        $this->configData = $configData;
        return ['category' => $this->category, 'catalogCategoryView' => $this->catalogCategoryView];
    }

    public function tearDown()
    {
        $this->objectManager->create(
            SetupConfigurationStep::class,
            ['configData' => $this->configData, 'rollback' => true]
        )->run();
    }

    private function createBulkOfProducts($productsCount)
    {
        for ($counter = 1; $counter <= $productsCount; $counter++) {
            $product = $this->fixtureFactory->createByCode(
                'catalogProductSimple',
                [
                    'dataset' => 'default',
                    'data' => [
                        'category_ids' => [
                            'category' => $this->category
                        ]
                    ]
                ]
            );
            $product->persist();
        }
    }
}