QueryText.php

Go to the documentation of this file.

<?php
namespace Magento\CatalogSearch\Test\Fixture\CatalogSearchQuery;

use Magento\Mtf\Fixture\DataSource;
use Magento\Mtf\Fixture\FixtureFactory;
use Magento\Mtf\Fixture\InjectableFixture;

class QueryText extends DataSource
{
    protected $product;

    private $products;

    public function __construct(FixtureFactory $fixtureFactory, array $params, array $data = [])
    {
        $this->params = $params;

        $this->data = array_key_exists('search_query', $data) ? $data['search_query'] : null;

        $this->products = $this->createProducts($fixtureFactory, (array)$data['value']);
    }

    private function createProducts(FixtureFactory $fixtureFactory, $productsData)
    {
        $products = [];
        foreach ($productsData as $productStringData) {
            $productData = explode('::', $productStringData);
            if (!empty($productData) && count($productData) > 1) {
                $product = $this->createProduct($fixtureFactory, $productData);

                $searchValue = isset($productData[2]) ? $productData[2] : $productData[1];
                if ($this->data === null) {
                    if ($product->hasData($searchValue)) {
                        $getProperty = 'get' . str_replace(' ', '', ucwords(str_replace('_', ' ', $searchValue)));
                        $this->data = $product->$getProperty();
                    } else {
                        $this->data = $searchValue;
                    }
                }

                $products[] = $product;
            } elseif ($this->data === null) {
                $this->data = (string)$productData;
            }
        }

        return $products;
    }

    private function createProduct(FixtureFactory $fixtureFactory, $productData)
    {
        $fixtureCode = $this->getProductFixtureCode($productData);
        $dataset = $this->getProductDataSet($productData);
        $product = $fixtureFactory->createByCode($fixtureCode, ['dataset' => $dataset]);
        if (!$product->hasData('id')) {
            $product->persist();
        }

        return $product;
    }

    private function getProductFixtureCode($productData)
    {
        $fixtureCode = $productData[0];

        return $fixtureCode;
    }

    private function getProductDataSet($productData)
    {
        return (isset($productData[2]) || null !== $this->data) ? $productData[1] : '';
    }

    public function getFirstProduct()
    {
        return reset($this->products);
    }

    public function getProducts()
    {
        return $this->products;
    }
}