WebsiteId.php

Go to the documentation of this file.

<?php
namespace Magento\Customer\Test\Fixture\Customer;

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

class WebsiteId extends DataSource
{
    private $store;

    private $website;

    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");
        }

        if (isset($this->fixtureData['website'])) {
            $this->website = $this->fixtureData['website'];
            $this->data = $this->fixtureData['website']->getName();
        } else {
            if (isset($this->fixtureData['dataset'])) {
                $store = $this->fixtureFactory->createByCode('store', $this->fixtureData);

                if (!$store->getStoreId()) {
                    $store->persist();
                }

                $website = $store->getDataFieldConfig('group_id')['source']
                    ->getStoreGroup()->getDataFieldConfig('website_id')['source']->getWebsite();

                $this->data = $website->getName();
                $this->website = $website;
                $this->store = $store;
            }
        }

        return parent::getData($key);
    }

    public function getStore()
    {
        return $this->store;
    }

    public function getWebsite()
    {
        return $this->website;
    }
}