StoreTree.php

Go to the documentation of this file.

<?php
namespace Magento\Catalog\Test\Block\Adminhtml\Product\Edit\Section\Websites;

use Magento\Mtf\Client\Locator;
use Magento\Mtf\Client\Element\SimpleElement;

class StoreTree extends SimpleElement
{
    protected $website = './/label[contains(text(), "%s")]';

    protected $selectedWebsite = '(.//input[contains(@name, "product[website_ids]") and not(@value="0")])[%d]/../label';

    public function setValue($values)
    {
        $this->clearValue();

        $values = is_array($values) ? $values : [$values];
        foreach ($values as $value) {
            $website = $this->find(sprintf($this->website, $value), Locator::SELECTOR_XPATH);
            if (!$website->isVisible()) {
                throw new \Exception("Can't find website: \"{$value}\".");
            }
            if (!$website->isSelected()) {
                $website->click();
            }
        }
    }

    public function getValue()
    {
        $values = [];

        $count = 1;
        $website = $this->find(sprintf($this->selectedWebsite, $count), Locator::SELECTOR_XPATH);
        while ($website->isVisible()) {
            $values[] = $website->getText();
            ++$count;
            $website = $this->find(sprintf($this->selectedWebsite, $count), Locator::SELECTOR_XPATH);
        }
        return $values;
    }

    private function clearValue()
    {
        foreach ($this->getValue() as $value) {
            $this->find(sprintf($this->website, $value), Locator::SELECTOR_XPATH)->click();
        }
    }
}