WebsiteIds.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Test\Fixture\Product;

use Magento\Mtf\Fixture\DataSource;
use Magento\Mtf\Fixture\FixtureFactory;
use Magento\Store\Test\Fixture\Store;

class WebsiteIds extends DataSource
{
    private $stores = [];

    private $websites = [];

    private $fixtureFactory;

    private $fixtureData = null;

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

    public function getData($key = null)
    {
        if (empty($this->fixtureData)) {
            throw new \Exception("Data must be set");
        }

        foreach ($this->fixtureData as $dataset) {
            if (is_array($dataset) && isset($dataset['websites'])) {
                foreach ($dataset['websites'] as $website) {
                    $this->websites[] = $website;
                }
            } else {
                $this->createStore($dataset);
            }
        }

        return parent::getData($key);
    }

    private function createStore($dataset)
    {
        if ($dataset instanceof Store) {
            $store = $dataset;
        } elseif (is_array($dataset)) {
            $store = isset($dataset['store']) ? $dataset['store'] :
                (isset($dataset['dataset']) ? $this->fixtureFactory->createByCode('store', $dataset) : null);
        }
        if (isset($store)) {
            $this->setWebsiteStoreData($store);
        }
    }

    private function setWebsiteStoreData(Store $store)
    {
        if (!$store->getStoreId()) {
            $store->persist();
        }
        $website = $store->getDataFieldConfig('group_id')['source']
            ->getStoreGroup()->getDataFieldConfig('website_id')['source']->getWebsite();
        $this->data[] = $website->getName();
        $this->websites[] = $website;
        $this->stores[] = $store;
    }

    public function getStores()
    {
        return $this->stores;
    }

    public function getWebsites()
    {
        return $this->websites;
    }
}