AssertProductInventoryThreshold.php

Go to the documentation of this file.

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

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

class AssertProductInventoryThreshold extends AbstractConstraint
{
    const SUCCESS_MESSAGE = 'Only %s left';

    public function processAssert(
        BrowserInterface $browser,
        FixtureInterface $product,
        FixtureFactory $fixtureFactory,
        CatalogProductView $catalogProductView,
        array $threshold
    ) {
        foreach ($threshold as $thresholdItem) {
            $this->buyProduct($fixtureFactory, $product, $thresholdItem['qty']);
            $browser->open($_ENV['app_frontend_url'] . $product->getUrlKey() . '.html');
            $isThresholdMessageDisplayed = $catalogProductView->getViewBlock()
                ->isThresholdMessageDisplayed($browser, $product);
            \PHPUnit\Framework\Assert::assertEquals(
                $isThresholdMessageDisplayed,
                $thresholdItem['is_message_displayed'],
                'Product inventory threshold message display is not correct.'
            );
            if ($thresholdItem['is_message_displayed']) {
                \PHPUnit\Framework\Assert::assertEquals(
                    sprintf(self::SUCCESS_MESSAGE, $thresholdItem['expected']),
                    $catalogProductView->getViewBlock()->getThresholdMessage(),
                    'Product inventory success message is not displayed.'
                );
            }
        }
    }

    private function buyProduct(FixtureFactory $fixtureFactory, FixtureInterface $product, $qty)
    {
        $newProduct = $fixtureFactory->createByCode(
            'catalogProductSimple',
            ['data' => ['sku' => $product->getSku(), 'checkout_data' => ['qty' => $qty]]]
        );
        $order = $fixtureFactory->createByCode('orderInjectable', [
            'dataset' => 'guest',
            'data' => [
                'entity_id' => [
                    'products' => [$newProduct]
                ]
            ]
        ]);
        $order->persist();
    }

    public function toString()
    {
        return 'Product inventory threshold message display is correct.';
    }
}