ApplyCatalogPriceRulesTest.php

Go to the documentation of this file.

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

use Magento\Catalog\Test\Fixture\CatalogProductSimple;
use Magento\CatalogRule\Test\Fixture\CatalogRule;
use Magento\Customer\Test\Fixture\Customer;
use Magento\Mtf\Util\Command\Cli\Cron;
use Magento\CatalogRule\Test\Page\Adminhtml\CatalogRuleEdit;
use Magento\Mtf\TestStep\TestStepFactory;
use Magento\Mtf\Fixture\FixtureInterface;

class ApplyCatalogPriceRulesTest extends AbstractCatalogRuleEntityTest
{
    /* tags */
    const TEST_TYPE = 'acceptance_test, extended_acceptance_test';
    const MVP = 'yes';
    /* end tags */

    protected $promo;

    public function test(
        array $catalogRules,
        CatalogRuleEdit $catalogRuleEdit,
        TestStepFactory $stepFactory,
        Cron $cron,
        $isCronEnabled = false,
        Customer $customer = null,
        array $products = [],
        $promo = 0
    ) {
        $this->promo = $promo;
        if ($customer !== null) {
            $customer->persist();
        }

        $products = $stepFactory->create(
            \Magento\Catalog\Test\TestStep\CreateProductsStep::class,
            ['products' => $products]
        )->run()['products'];

        foreach ($catalogRules as $catalogRule) {
            foreach ($products as $product) {
                $catalogPriceRule = $this->createCatalogPriceRule($catalogRule, $product, $customer);
                if ($isCronEnabled) {
                    $cron->run();
                    $cron->run();
                } else {
                    $catalogRuleEdit->open(['id' => $catalogPriceRule->getId()]);
                    $this->catalogRuleNew->getFormPageActions()->saveAndApply();
                }
            }
        }
        return ['products' => $products];
    }

    protected function prepareCondition(FixtureInterface $product, array $catalogPriceRule)
    {
        $conditionEntity = explode('|', trim($catalogPriceRule['data']['rule'], '[]'))[0];
        $actionName = 'get' . $conditionEntity;
        if (method_exists($this, $actionName)) {
            $result = $this->$actionName($product);
            foreach ($result as $key => $value) {
                $catalogPriceRule['data']['rule'] = str_replace($key, $value, $catalogPriceRule['data']['rule']);
            }
            return $catalogPriceRule;
        } else {
            $message = sprintf('Method "%s" does not exist in %s', $actionName, get_class($this));
            throw new \BadMethodCallException($message);
        }
    }

    protected function getCategory(FixtureInterface $product)
    {
        $result['%category_id%'] = $product->getDataFieldConfig('category_ids')['source']->getIds()[0];
        return $result;
    }

    protected function getAttribute(FixtureInterface $product)
    {
        $attributes = $product->getDataFieldConfig('attribute_set_id')['source']
            ->getAttributeSet()->getDataFieldConfig('assigned_attributes')['source']->getAttributes();
        $result['%attribute_id%'] = $attributes[0]->getAttributeCode();
        $result['%attribute_value%'] = $attributes[0]->getOptions()[$this->promo]['id'];
        return $result;
    }

    protected function applyCustomerGroup(array $catalogPriceRule, Customer $customer)
    {
        $customerGroup = $customer->getDataFieldConfig('group_id')['source']->getCustomerGroup();
        $catalogPriceRule['data']['customer_group_ids']['option_0'] = $customerGroup->getCustomerGroupId();

        return $catalogPriceRule;
    }

    protected function createCatalogPriceRule(
        array $catalogPriceRule,
        FixtureInterface $product,
        Customer $customer = null
    ) {
        if (isset($catalogPriceRule['data']['rule'])) {
            $catalogPriceRule = $this->prepareCondition($product, $catalogPriceRule);
        }
        if ($customer !== null) {
            $catalogPriceRule = $this->applyCustomerGroup($catalogPriceRule, $customer);
        }

        $catalogPriceRule = $this->fixtureFactory->createByCode('catalogRule', $catalogPriceRule);
        $catalogPriceRule->persist();

        return $catalogPriceRule;
    }
}