ExportAdvancedPricingTest.php

Go to the documentation of this file.

<?php
namespace Magento\AdvancedPricingImportExport\Test\TestCase;

use Magento\Catalog\Test\Page\Adminhtml\CatalogProductIndex;
use Magento\ImportExport\Test\Page\Adminhtml\AdminExportIndex;
use Magento\Mtf\Fixture\FixtureFactory;
use Magento\Mtf\TestCase\Injectable;
use Magento\Mtf\TestStep\TestStepFactory;
use Magento\Store\Test\Fixture\Website;

class ExportAdvancedPricingTest extends Injectable
{
    private $stepFactory;

    private $fixtureFactory;

    private $adminExportIndex;

    private $configData;

    private $catalogProductIndex;

    public function __prepare(
        CatalogProductIndex $catalogProductIndex
    ) {
        $catalogProductIndex->open();
        $catalogProductIndex->getProductGrid()->massaction([], 'Delete', true, 'Select All');
    }

    public function __inject(
        TestStepFactory $stepFactory,
        FixtureFactory $fixtureFactory,
        AdminExportIndex $adminExportIndex,
        CatalogProductIndex $catalogProductIndexPage
    ) {
        $this->stepFactory = $stepFactory;
        $this->fixtureFactory = $fixtureFactory;
        $this->adminExportIndex = $adminExportIndex;
        $this->catalogProductIndex = $catalogProductIndexPage;
    }

    public function test(
        $exportData,
        array $products = [],
        $configData = null,
        Website $website = null,
        array $advancedPricingAttributes = [],
        $currencyCustomWebsite = null
    ) {
        $this->configData = $configData;

        if ($this->configData) {
            $this->stepFactory->create(
                \Magento\Config\Test\TestStep\SetupConfigurationStep::class,
                ['configData' => $configData]
            )->run();
        }

        if ($website) {
            $website->persist();
            $this->setupCurrencyForCustomWebsite($website, $currencyCustomWebsite);
        }
        $products = $this->prepareProducts($products, $website);
        $this->adminExportIndex->open();

        $exportData = $this->fixtureFactory->createByCode(
            'exportData',
            [
                'dataset' => $exportData,
                'data' => [
                    'data_export' => $products[0]
                ]
            ]
        );
        $exportData->persist();

        $this->adminExportIndex->getExportForm()->fill($exportData, null, $advancedPricingAttributes);
        $this->adminExportIndex->getFilterExport()->clickContinue();

        if (!empty($advancedPricingAttributes)) {
            $products = [$products[0]];
        }

        return [
            'products' => $products
        ];
    }

    private function setupCurrencyForCustomWebsite($website, $currencyDataset)
    {
        $configFixture = $this->fixtureFactory->createByCode(
            'configData',
            [
                'dataset' => $currencyDataset,
                'data' => [
                    'scope' => [
                        'fixture' => $website,
                        'scope_type' => 'website',
                        'website_id' => $website->getWebsiteId(),
                        'set_level' => 'website',
                    ]
                ]
            ]
        );
        $configFixture->persist();
    }

    public function prepareProducts(array $products, Website $website = null)
    {
        if (empty($products)) {
            return null;
        }
        $createdProducts = [];
        foreach ($products as $product) {
            $data = [
                'website_ids' => [
                    ['websites' => $website]
                ]
            ];
            if ($website) {
                $data['tier_price'] = [
                    'data' => [
                        'website' => $website
                    ]
                ];
            }

            if (isset($product['data'])) {
                $data = array_merge($data, $product['data']);
            }

            $product = $this->fixtureFactory->createByCode(
                $product['fixture'],
                [
                    'dataset' => $product['dataset'],
                    'data' => $data
                ]
            );
            $product->persist();
            $createdProducts[] = $product;
        }

        return $createdProducts;
    }

    public function tearDown()
    {
        if ($this->configData) {
            $this->stepFactory->create(
                \Magento\Config\Test\TestStep\SetupConfigurationStep::class,
                ['configData' => 'price_scope_website', 'rollback' => true]
            )->run();
        }
    }
}