ProductAssert.php

Go to the documentation of this file.

<?php
namespace Magento\Setup\Fixtures\FixturesAsserts;

class ProductAssert
{
    protected $productCollectionFactory;

    protected $expressionFactory;

    public function __construct(
        \Magento\Catalog\Model\ResourceModel\Product\CollectionFactory $productCollectionFactory,
        \Magento\Framework\DB\Sql\ColumnValueExpressionFactory $expressionFactory
    ) {
        $this->productCollectionFactory = $productCollectionFactory;
        $this->expressionFactory = $expressionFactory;
    }

    public function assertProductsCount($skuPattern, $expectedCount)
    {
        $productSkuPattern = str_replace('%s', '[0-9]+', $skuPattern);
        $productCollection = $this->productCollectionFactory->create();
        $productCollection
            ->getSelect()
            ->where('sku ?', $this->expressionFactory->create([
                'expression' => 'REGEXP \'^' . $productSkuPattern . '$\''
            ]));

        if ($expectedCount !== count($productCollection)) {
            throw new \AssertionError(
                sprintf(
                    'Expected amount of products with sku pattern "%s" not equals actual amount',
                    $skuPattern
                )
            );
        }
    }

    public function assertProductType($expectedProductType, $product)
    {
        if ($expectedProductType !== $product->getTypeId()) {
            throw new \AssertionError('Product type is wrong');
        }
    }
}