package ru.yandex.autotests.direct.web.pages;

import ru.yandex.qatools.allure.webdriver.rules.WebDriverConfiguration;
import ru.yandex.qatools.allure.webdriver.steps.WebDriverSteps;
import ru.yandex.qatools.htmlelements.element.HtmlElement;

/**
 * @author Roman Kuhta (kuhtich@yandex-team.ru)
 */
public class ExtendedHtmlElement extends HtmlElement {
    protected WebDriverConfiguration config;

    public void setConfig(WebDriverConfiguration config) {
        this.config = config;
    }

    public void takeScreenshot() {
        WebDriverSteps.getInstance(WebDriverSteps.class, config).takeScreenshot();
    }
}
