AssertProductOnCustomWebsite.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\AbstractAssertForm;
use Magento\Mtf\Fixture\FixtureInterface;

class AssertProductOnCustomWebsite extends AbstractAssertForm
{
    const NOT_FOUND_MESSAGE = 'Whoops, our bad...';

    protected $productViewPage;

    protected $browser;

    public function processAssert(
        BrowserInterface $browser,
        CatalogProductView $catalogProductView,
        FixtureInterface $product
    ) {
        $this->browser = $browser;
        $this->productViewPage = $catalogProductView;

        $this->verifyProductOnMainWebsite($product);
        $this->verifyProductOnCustomWebsite($product);
    }

    protected function verifyProductOnMainWebsite(FixtureInterface $product)
    {
        $this->browser->open($_ENV['app_frontend_url'] . $product->getUrlKey() . '.html');

        \PHPUnit\Framework\Assert::assertEquals(
            self::NOT_FOUND_MESSAGE,
            $this->productViewPage->getTitleBlock()->getTitle(),
            'Product ' . $product->getName() . ' is available on Main Website, but should not.'
        );
    }

    protected function verifyProductOnCustomWebsite(FixtureInterface $product)
    {
        $websites = $product->getDataFieldConfig('website_ids')['source']->getWebsites();
        foreach ($websites as $website) {
            $this->browser->open(
                $_ENV['app_frontend_url'] . 'websites/' . $website->getCode() . '/' . $product->getUrlKey() . '.html'
            );

            \PHPUnit\Framework\Assert::assertEquals(
                $product->getName(),
                $this->productViewPage->getViewBlock()->getProductName(),
                'Product ' . $product->getName() . ' is not available on ' . $website->getName() . ' website.'
            );
        }
    }

    public function toString()
    {
        return 'Product on product view page displayed in appropriate Website.';
    }
}