Kolyunya\Codeception\Lib\MarkupValidator\DefaultMarkupProvider

Complexity

Total Complexity

8

Size/Duplication

Total Lines	82
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	3

Importance

Changes

0

5 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	4	1
A	getMarkup()	0	16	3
A	getMarkupFromPhpBrowser()	0	8	1
A	getMarkupFromWebDriver()	0	8	1
A	getModule()	0	10	2

<?php

namespace Kolyunya\Codeception\Lib\MarkupValidator;

use Exception;
use Codeception\Lib\ModuleContainer;
use Codeception\Module\PhpBrowser;
use Codeception\Module\WebDriver;
use Kolyunya\Codeception\Lib\MarkupValidator\MarkupProviderInterface;

/**
 * Default markup provider which attemps to get markup from
 * `PhpBrowser` and `WebDriver` modules.
 */
class DefaultMarkupProvider implements MarkupProviderInterface
{
    /**
     * Module container.
     *
     * @var ModuleContainer
     */
    private $moduleContainer;

    /**
     * {@inheritDoc}
     */
    public function __construct(ModuleContainer $moduleContainer, array $config = array())
    {
        $this->moduleContainer = $moduleContainer;
    }

    /**
     * {@inheritDoc}
     */
    public function getMarkup()
    {
        try {
            return $this->getMarkupFromPhpBrowser();
        } catch (Exception $exception) {
            // Wasn't able to get markup from the `PhpBrowser` module.
        }

        try {
            return $this->getMarkupFromWebDriver();
        } catch (Exception $exception) {
            // Wasn't able to get markup from the `WebDriver` module.
        }

        throw new Exception('Unable to obtain current page markup.');
    }

    /**
     * Returns current page markup form the `PhpBrowser` module.
     *
     * @return string Current page markup.
     */
    private function getMarkupFromPhpBrowser()
    {
        /* @var $phpBrowser PhpBrowser */
        $phpBrowser = $this->getModule('PhpBrowser');
        $markup = $phpBrowser->_getResponseContent();

        return $markup;
    }

    /**
     * Returns current page markup form the `WebDriver` module.
     *
     * @return string Current page markup.
     */
    private function getMarkupFromWebDriver()
    {
        /* @var $webDriver WebDriver */
        $webDriver = $this->getModule('WebDriver');
        $markup = $webDriver->webDriver->getPageSource();

        return $markup;
    }

    /**
     * Returns a module instance by its name.
     *
     * @param string $name Module name.
     * @return object Module instance.
     */
    private function getModule($name)
    {
        if (!$this->moduleContainer->hasModule($name)) {
            throw new Exception(sprintf('«%s» module is not available.', $name));
        }

        $module = $this->moduleContainer->getModule($name);

        return $module;
    }
}