CreateSalesRuleEntityTest.php

Go to the documentation of this file.

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

use Magento\Catalog\Test\Fixture\CatalogProductSimple;
use Magento\Customer\Test\Fixture\Customer;
use Magento\Mtf\Fixture\FixtureFactory;
use Magento\Mtf\TestCase\Injectable;
use Magento\SalesRule\Test\Block\Adminhtml\Promo\Quote\Edit\Section\ManageCouponCode;
use Magento\SalesRule\Test\Fixture\SalesRule;
use Magento\SalesRule\Test\Page\Adminhtml\PromoQuoteEdit;
use Magento\SalesRule\Test\Page\Adminhtml\PromoQuoteIndex;
use Magento\SalesRule\Test\Page\Adminhtml\PromoQuoteNew;

class CreateSalesRuleEntityTest extends Injectable
{
    /* tags */
    const MVP = 'yes';
    const TEST_TYPE = 'extended_acceptance_test';
    /* end tags */

    protected $promoQuoteNew;

    protected $promoQuoteEdit;

    protected $promoQuoteIndex;

    protected $salesRuleName;

    protected $fixtureFactory;

    public function __inject(
        PromoQuoteNew $promoQuoteNew,
        PromoQuoteIndex $promoQuoteIndex,
        PromoQuoteEdit $promoQuoteEdit,
        FixtureFactory $fixtureFactory
    ) {
        $this->promoQuoteNew = $promoQuoteNew;
        $this->promoQuoteIndex = $promoQuoteIndex;
        $this->promoQuoteEdit = $promoQuoteEdit;
        $this->fixtureFactory = $fixtureFactory;
    }

    public function testCreateSalesRule(
        SalesRule $salesRule,
        CatalogProductSimple $productForSalesRule1,
        CatalogProductSimple $productForSalesRule2 = null,
        Customer $customer = null,
        $conditionEntity = null,
        SalesRule $salesRuleEdit = null
    ) {
        $replace = null;
        $this->salesRuleName = $salesRule->getName();

        // Prepare data
        if ($customer !== null) {
            $customer->persist();
        }
        $productForSalesRule1->persist();
        if ($productForSalesRule2 !== null) {
            $productForSalesRule2->persist();
            if ($conditionEntity !== null) {
                $replace = $this->prepareCondition($productForSalesRule2, $conditionEntity);
            }
        }

        // Steps
        $this->promoQuoteNew->open();
        $this->promoQuoteNew->getSalesRuleForm()->fill($salesRule, null, $replace);

        if ($salesRule->getCouponType() == "Auto") {
            $this->promoQuoteNew->getFormPageActions()->saveAndContinue();
            $form = $this->promoQuoteEdit->getSalesRuleForm();
            $form->openSection('manage_coupon_code');
            $section = $form->getSection('manage_coupon_code');
            $section->fill($salesRuleEdit);
            $section->generateCouponCodes();
            $couponCode = $section->getGeneratedCouponCode();
            $this->promoQuoteEdit->getFormPageActions()->save();

            return ["couponCode" => $couponCode];
        } else {
            $this->promoQuoteNew->getFormPageActions()->save();
        }
    }

    protected function prepareCondition(CatalogProductSimple $productSimple, $conditionEntity)
    {
        $result = [];

        switch ($conditionEntity) {
            case 'category':
                $result['%category_id%'] = $productSimple->getDataFieldConfig('category_ids')['source']->getIds()[0];
                break;
            case 'attribute':
                $attributes = $productSimple->getDataFieldConfig('attribute_set_id')['source']
                    ->getAttributeSet()->getDataFieldConfig('assigned_attributes')['source']->getAttributes();

                $result['%attribute_name%'] = $attributes[0]->getFrontendLabel();
                $result['%attribute_value%'] = $attributes[0]->getOptions()[0]['view'];
                break;
        }

        return [
            'conditions' => [
                'conditions_serialized' => $result,
            ],
        ];
    }

    public function tearDown()
    {
        $filter = [
            'name' => $this->salesRuleName,
        ];

        $this->promoQuoteIndex->open();
        $this->promoQuoteIndex->getPromoQuoteGrid()->searchAndOpen($filter);
        $this->promoQuoteEdit->getFormPageActions()->delete();
        $this->promoQuoteEdit->getModalBlock()->acceptAlert();
    }
}