AssertProductInCart.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Test\Constraint;

use Magento\Catalog\Test\Fixture\CatalogProductSimple;
use Magento\Catalog\Test\Page\Product\CatalogProductView;
use Magento\Checkout\Test\Page\CheckoutCart;
use Magento\Mtf\Client\BrowserInterface;
use Magento\Mtf\Constraint\AbstractConstraint;
use Magento\Mtf\Fixture\FixtureInterface;

class AssertProductInCart extends AbstractConstraint
{
    protected $formPrice;

    protected $fixtureActualPrice;

    protected $fixturePrice;

    public function processAssert(
        CatalogProductView $catalogProductView,
        FixtureInterface $product,
        BrowserInterface $browser,
        CheckoutCart $checkoutCart
    ) {
        // Add product to cart
        $browser->open($_ENV['app_frontend_url'] . $product->getUrlKey() . '.html');
        $catalogProductView->getViewBlock()->fillOptions($product);
        $catalogProductView->getViewBlock()->clickAddToCart();
        $catalogProductView->getMessagesBlock()->waitSuccessMessage();

        // Check price
        $this->countPrices($product, $checkoutCart);
        \PHPUnit\Framework\Assert::assertEquals(
            $this->fixtureActualPrice,
            $this->formPrice,
            'Product price in shopping cart is not correct.'
        );
    }

    protected function countPrices(FixtureInterface $product, CheckoutCart $checkoutCart)
    {
        $this->fixturePrice = $product->getPrice();
        $this->prepareFormPrice($product, $checkoutCart);
        $this->countSpecialPrice($product);
        $this->countCheckoutCartItemPrice($product);
        $this->countCustomOptionsPrice($product);
    }

    protected function countSpecialPrice(FixtureInterface $product)
    {
        $specialPrice = $product->getSpecialPrice();
        if ($specialPrice) {
            $this->fixturePrice = $product->getSpecialPrice();
        }
    }

    protected function prepareFormPrice(FixtureInterface $product, CheckoutCart $checkoutCart)
    {
        $checkoutCart->open();
        $cartItem = $checkoutCart->getCartBlock()->getCartItem($product);
        $this->formPrice = $cartItem->getPrice();
    }

    protected function countCheckoutCartItemPrice(FixtureInterface $product)
    {
        $checkoutData = $product->getCheckoutData();
        $checkoutCartItem = isset($checkoutData['cartItem']) ? $checkoutData['cartItem'] : [];
        if (isset($checkoutCartItem['price'])) {
            $this->fixturePrice = $checkoutCartItem['price'];
        }
        $this->fixtureActualPrice = $this->fixturePrice;
    }

    protected function countCustomOptionsPrice(FixtureInterface $product)
    {
        $customOptions = $product->getCustomOptions();
        $checkoutData = $product->getCheckoutData();
        $checkoutCustomOptions = isset($checkoutData['options']['custom_options'])
            ? $checkoutData['options']['custom_options']
            : [];
        foreach ($checkoutCustomOptions as $checkoutOption) {
            $attributeKey = str_replace('attribute_key_', '', $checkoutOption['title']);
            $optionKey = str_replace('option_key_', '', $checkoutOption['value']);
            $option = $customOptions[$attributeKey]['options'][$optionKey];

            if ('Fixed' == $option['price_type']) {
                $this->fixtureActualPrice += $option['price'];
            } else {
                $this->fixtureActualPrice += ($this->fixturePrice / 100) * $option['price'];
            }
        }
    }

    public function toString()
    {
        return 'Product is correctly displayed in cart.';
    }
}