silverstripe-behat-extension/src/SilverStripe/BehatExtension/Context/SilverStripeContext.php

<?php

namespace SilverStripe\BehatExtension\Context;

use Behat\Behat\Context\Step;
use Behat\Behat\Event\ScenarioEvent;
use Behat\MinkExtension\Context\MinkContext;
use Behat\Mink\Driver\GoutteDriver;
use Behat\Mink\Driver\Selenium2Driver;
use Behat\Mink\Exception\UnsupportedDriverActionException;
use Behat\Mink\Exception\ElementNotFoundException;
use InvalidArgumentException;

// Mink etc.
require_once 'vendor/autoload.php';

require_once BASE_PATH . '/vendor/phpunit/phpunit/src/Framework/Assert/Functions.php';

/**
 * SilverStripeContext
 *
 * Generic context wrapper used as a base for Behat FeatureContext.
 */
class SilverStripeContext extends MinkContext implements SilverStripeAwareContextInterface
{
    protected $databaseName;

    /**
     * @var array Partial string match for step names
     * that are considered to trigger Ajax request in the CMS,
     * and hence need special timeout handling.
     * @see \SilverStripe\BehatExtension\Context\BasicContext->handleAjaxBeforeStep().
     */
    protected $ajaxSteps;

    /**
     * @var Int Timeout in milliseconds, after which the interface assumes
     * that an Ajax request has timed out, and continues with assertions.
     */
    protected $ajaxTimeout;

    /**
     * @var String Relative URL to the SilverStripe admin interface.
     */
    protected $adminUrl;

    /**
     * @var String Relative URL to the SilverStripe login form.
     */
    protected $loginUrl;

    /**
     * @var String Relative path to a writeable folder where screenshots can be stored.
     * If set to NULL, no screenshots will be stored.
     */
    protected $screenshotPath;

    protected $context;

    protected $testSessionEnvironment;


    /**
     * Initializes context.
     * Every scenario gets it's own context object.
     *
     * @param   array   $parameters     context parameters (set them up through behat.yml)
     */
    public function __construct(array $parameters)
    {
        // Initialize your context here
        $this->context = $parameters;
        $this->testSessionEnvironment = new \TestSessionEnvironment();
    }

    public function setDatabase($databaseName)
    {
        $this->databaseName = $databaseName;
    }

    public function setAjaxSteps($ajaxSteps)
    {
        if ($ajaxSteps) {
            $this->ajaxSteps = $ajaxSteps;
        }
    }

    public function getAjaxSteps()
    {
        return $this->ajaxSteps;
    }

    public function setAjaxTimeout($ajaxTimeout)
    {
        $this->ajaxTimeout = $ajaxTimeout;
    }

    public function getAjaxTimeout()
    {
        return $this->ajaxTimeout;
    }

    public function setAdminUrl($adminUrl)
    {
        $this->adminUrl = $adminUrl;
    }

    public function getAdminUrl()
    {
        return $this->adminUrl;
    }

    public function setLoginUrl($loginUrl)
    {
        $this->loginUrl = $loginUrl;
    }

    public function getLoginUrl()
    {
        return $this->loginUrl;
    }

    public function setScreenshotPath($screenshotPath)
    {
        $this->screenshotPath = $screenshotPath;
    }

    public function getScreenshotPath()
    {
        return $this->screenshotPath;
    }

    public function getRegionMap()
    {
        return $this->regionMap;
    }

    public function setRegionMap($regionMap)
    {
        $this->regionMap = $regionMap;
    }

    /**
     * Returns MinkElement based off region defined in .yml file.
     * Also supports direct CSS selectors and regions identified by a "data-title" attribute.
     * When using the "data-title" attribute, ensure not to include double quotes.
     *
     * @param string $region Region name or CSS selector
     * @return MinkElement
     */
    public function getRegionObj($region)
    {
        // Try to find regions directly by CSS selector.
        try {
            $regionObj = $this->getSession()->getPage()->find(
                'css',
                // Escape CSS selector
                (false !== strpos($region, "'")) ? str_replace("'", "\'", $region) : $region
            );
            if ($regionObj) {
                return $regionObj;
            }
        } catch (\Symfony\Component\CssSelector\Exception\SyntaxErrorException $e) {
            // fall through to next case
        }

        // Fall back to region identified by data-title.
        // Only apply if no double quotes exist in search string,
        // which would break the CSS selector.
        if (false === strpos($region, '"')) {
            $regionObj = $this->getSession()->getPage()->find(
                'css',
                '[data-title="' . $region . '"]'
            );
            if ($regionObj) {
                return $regionObj;
            }
        }

        // Look for named region
        if (!$this->regionMap) {
            throw new \LogicException("Cannot find 'region_map' in the behat.yml");
        }
        if (!array_key_exists($region, $this->regionMap)) {
            throw new \LogicException("Cannot find the specified region in the behat.yml");
        }
        $regionObj = $this->getSession()->getPage()->find('css', $region);
        if (!$regionObj) {
            throw new ElementNotFoundException("Cannot find the specified region on the page");
        }

        return $regionObj;
    }

    /**
     * @BeforeScenario
     */
    public function before(ScenarioEvent $event)
    {
        if (!isset($this->databaseName)) {
            throw new \LogicException(
                'Context\'s $databaseName has to be set when implementing SilverStripeAwareContextInterface.'
            );
        }

        $state = $this->getTestSessionState();
        $this->testSessionEnvironment->startTestSession($state);

        // Optionally import database
        if (!empty($state['importDatabasePath'])) {
            $this->testSessionEnvironment->importDatabase(
                $state['importDatabasePath'],
                !empty($state['requireDefaultRecords']) ? $state['requireDefaultRecords'] : false
            );
        } elseif (!empty($state['requireDefaultRecords']) && $state['requireDefaultRecords']) {
            $this->testSessionEnvironment->requireDefaultRecords();
        }

        // Fixtures
        $fixtureFile = (!empty($state['fixture'])) ? $state['fixture'] : null;
        if ($fixtureFile) {
            $this->testSessionEnvironment->loadFixtureIntoDb($fixtureFile);
        }

        if ($screenSize = getenv('BEHAT_SCREEN_SIZE')) {
            list($screenWidth, $screenHeight) = explode('x', $screenSize);
            $this->getSession()->resizeWindow((int)$screenWidth, (int)$screenHeight);
        } else {
            $this->getSession()->resizeWindow(1024, 768);
        }
    }

    /**
     * Returns a parameter map of state to set within the test session.
     * Takes TESTSESSION_PARAMS environment variable into account for run-specific configurations.
     *
     * @return array
     */
    public function getTestSessionState()
    {
        $extraParams = array();
        parse_str(getenv('TESTSESSION_PARAMS'), $extraParams);
        return array_merge(
            array(
                'database' => $this->databaseName,
                'mailer' => 'SilverStripe\BehatExtension\Utility\TestMailer',
            ),
            $extraParams
        );
    }

    /**
     * Parses given URL and returns its components
     *
     * @param $url
     * @return array|mixed Parsed URL
     */
    public function parseUrl($url)
    {
        $url = parse_url($url);
        $url['vars'] = array();
        if (!isset($url['fragment'])) {
            $url['fragment'] = null;
        }
        if (isset($url['query'])) {
            parse_str($url['query'], $url['vars']);
        }

        return $url;
    }

    /**
     * Checks whether current URL is close enough to the given URL.
     * Unless specified in $url, get vars will be ignored
     * Unless specified in $url, fragment identifiers will be ignored
     *
     * @param $url string URL to compare to current URL
     * @return boolean Returns true if the current URL is close enough to the given URL, false otherwise.
     */
    public function isCurrentUrlSimilarTo($url)
    {
        $current = $this->parseUrl($this->getSession()->getCurrentUrl());
        $test = $this->parseUrl($url);

        if ($current['path'] !== $test['path']) {
            return false;
        }

        if (isset($test['fragment']) && $current['fragment'] !== $test['fragment']) {
            return false;
        }

        foreach ($test['vars'] as $name => $value) {
            if (!isset($current['vars'][$name]) || $current['vars'][$name] !== $value) {
                return false;
            }
        }

        return true;
    }

    /**
     * Returns base URL parameter set in MinkExtension.
     * It simplifies configuration by allowing to specify this parameter
     * once but makes code dependent on MinkExtension.
     *
     * @return string
     */
    public function getBaseUrl()
    {
        return $this->getMinkParameter('base_url') ?: '';
    }

    /**
     * Joins URL parts into an URL using forward slash.
     * Forward slash usages are normalised to one between parts.
     * This method takes variable number of parameters.
     *
     * @param $...
     * @return string
     * @throws \InvalidArgumentException
     */
    public function joinUrlParts()
    {
        if (0 === func_num_args()) {
            throw new \InvalidArgumentException('Need at least one argument');
        }

        $parts = func_get_args();
        $trimSlashes = function (&$part) {
            $part = trim($part, '/');
        };
        array_walk($parts, $trimSlashes);

        return implode('/', $parts);
    }

    public function canIntercept()
    {
        $driver = $this->getSession()->getDriver();
        if ($driver instanceof GoutteDriver) {
            return true;
        } else {
            if ($driver instanceof Selenium2Driver) {
                return false;
            }
        }

        throw new UnsupportedDriverActionException('You need to tag the scenario with "@mink:goutte" or
			"@mink:symfony". Intercepting the redirections is not supported by %s', $driver);
    }

    /**
     * @Given /^(.*) without redirection$/
     */
    public function theRedirectionsAreIntercepted($step)
    {
        if ($this->canIntercept()) {
            $this->getSession()->getDriver()->getClient()->followRedirects(false);
        }

        return new Step\Given($step);
    }

    /**
     * Fills in form field with specified id|name|label|value.
     * Overwritten to select the first *visible* element, see https://github.com/Behat/Mink/issues/311
     */
    public function fillField($field, $value)
    {
        $value = $this->fixStepArgument($value);
        $fields = $this->getSession()->getPage()->findAll('named', array(
            'field', $this->getSession()->getSelectorsHandler()->xpathLiteral($field)
        ));
        if ($fields) {
            foreach ($fields as $f) {
                if ($f->isVisible()) {
                    $f->setValue($value);
                    return;
                }
            }
        }

        throw new ElementNotFoundException(
            $this->getSession(),
            'form field',
            'id|name|label|value',
            $field
        );
    }

    /**
     * Overwritten to click the first *visable* link the DOM.
     */
    public function clickLink($link)
    {
        $link = $this->fixStepArgument($link);
        $links = $this->getSession()->getPage()->findAll('named', array(
            'link', $this->getSession()->getSelectorsHandler()->xpathLiteral($link)
        ));
        if ($links) {
            foreach ($links as $l) {
                if ($l->isVisible()) {
                    $l->click();
                    return;
                }
            }
        }
        throw new ElementNotFoundException(
            $this->getSession(),
            'link',
            'id|name|label|value',
            $link
        );
    }

     /**
     * Sets the current date. Relies on the underlying functionality using
     * {@link SS_Datetime::now()} rather than PHP's system time methods like date().
     * Supports ISO fomat: Y-m-d
     * Example: Given the current date is "2009-10-31"
     *
     * @Given /^the current date is "([^"]*)"$/
     */
    public function givenTheCurrentDateIs($date)
    {
        $newDatetime = \DateTime::createFromFormat('Y-m-d', $date);
        if (!$newDatetime) {
            throw new InvalidArgumentException(sprintf('Invalid date format: %s (requires "Y-m-d")', $date));
        }

        $state = $this->testSessionEnvironment->getState();
        $oldDatetime = \DateTime::createFromFormat('Y-m-d H:i:s', isset($state->datetime) ? $state->datetime : null);
        if ($oldDatetime) {
            $newDatetime->setTime($oldDatetime->format('H'), $oldDatetime->format('i'), $oldDatetime->format('s'));
        }
        $state->datetime = $newDatetime->format('Y-m-d H:i:s');
        $this->testSessionEnvironment->applyState($state);
    }

    /**
     * Sets the current time. Relies on the underlying functionality using
     * {@link \SS_Datetime::now()} rather than PHP's system time methods like date().
     * Supports ISO fomat: H:i:s
     * Example: Given the current time is "20:31:50"
     *
     * @Given /^the current time is "([^"]*)"$/
     */
    public function givenTheCurrentTimeIs($time)
    {
        $newDatetime = \DateTime::createFromFormat('H:i:s', $date);
        if (!$newDatetime) {
            throw new InvalidArgumentException(sprintf('Invalid date format: %s (requires "H:i:s")', $date));
        }

        $state = $this->testSessionEnvironment->getState();
        $oldDatetime = \DateTime::createFromFormat('Y-m-d H:i:s', isset($state->datetime) ? $state->datetime : null);
        if ($oldDatetime) {
            $newDatetime->setDate($oldDatetime->format('Y'), $oldDatetime->format('m'), $oldDatetime->format('d'));
        }
        $state->datetime = $newDatetime->format('Y-m-d H:i:s');
        $this->testSessionEnvironment->applyState($state);
    }

    /**
     * Selects option in select field with specified id|name|label|value.
     *
     * @override /^(?:|I )select "(?P<option>(?:[^"]|\\")*)" from "(?P<select>(?:[^"]|\\")*)"$/
     */
    public function selectOption($select, $option)
    {
        // Find field
        $field = $this
            ->getSession()
            ->getPage()
            ->findField($this->fixStepArgument($select));

        // If field is visible then select it as per normal
        if ($field && $field->isVisible()) {
            parent::selectOption($select, $option);
        } else {
            $this->selectOptionWithJavascript($select, $option);
        }
    }

    /**
     * Selects option in select field with specified id|name|label|value using javascript
     * This method uses javascript to allow selection of options that may be
     * overridden by javascript libraries, and thus hide the element.
     *
     * @When /^(?:|I )select "(?P<option>(?:[^"]|\\")*)" from "(?P<select>(?:[^"]|\\")*)" with javascript$/
     */
    public function selectOptionWithJavascript($select, $option)
    {
        $select = $this->fixStepArgument($select);
        $option = $this->fixStepArgument($option);
        $page = $this->getSession()->getPage();

        // Find field
        $field = $page->findField($select);
        if (null === $field) {
            throw new ElementNotFoundException($this->getSession(), 'form field', 'id|name|label|value', $select);
        }

        // Find option
        $opt = $field->find('named', array(
            'option', $this->getSession()->getSelectorsHandler()->xpathLiteral($option)
        ));
        if (null === $opt) {
            throw new ElementNotFoundException($this->getSession(), 'select option', 'value|text', $option);
        }

        // Merge new option in with old handling both multiselect and single select
        $value = $field->getValue();
        $newValue = $opt->getAttribute('value');
        if (is_array($value)) {
            if (!in_array($newValue, $value)) {
                $value[] = $newValue;
            }
        } else {
            $value = $newValue;
        }
        $valueEncoded = json_encode($value);

        // Inject this value via javascript
        $fieldID = $field->getAttribute('ID');
        $script = <<<EOS
			(function($) {
				$("#$fieldID")
					.val($valueEncoded)
					.change()
					.trigger('liszt:updated')
					.trigger('chosen:updated');
			})(jQuery);
EOS;
        $this->getSession()->getDriver()->executeScript($script);
    }
}