io.nosqlbench.driver.webdriver.WebContext Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of driver-web Show documentation

A nosqlbench driver module; Provides the ability to drive steps through the webdriver API.

There is a newer version: 4.15.102

package io.nosqlbench.driver.webdriver;

import org.openqa.selenium.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;

public class WebContext {
    private final static Logger logger = LoggerFactory.getLogger(WebContext.class);

    private final WebDriver rootwindow;
    WebDriver focus;

    LinkedList elements = new LinkedList<>();
    private LinkedList cookies = new LinkedList<>();
    private HashMap vars = new HashMap<>();
    private Alert alert;

    public WebContext(WebDriver initial) {
        this.focus = initial;
        this.rootwindow = initial;
    }

    public void pushElement(WebElement element) {
        elements.push(element);
    }

    public WebElement peekElement() {
        return elements.peek();
    }

    public LinkedList getCookies() {
        return cookies;
    }

    public LinkedList getElements() {
        return elements;
    }

    public void setElements(List elements) {
        this.elements.clear();
        this.elements.addAll(elements);
    }

    public void pushCookie(Cookie cookie) {
        this.cookies.push(cookie);
    }

    public void setCookies(Collection cookies) {
        this.cookies.clear();
        this.cookies.addAll(cookies);

    }

    public void setAlert(Alert alert) {
        this.alert = alert;
    }

    public Alert getAlert() {
        return alert;
    }

    public WebDriver driver() {
        return focus;
    }

    public void setFocus(WebDriver driver) {
        this.focus =driver;
    }

    public void clearAlert() {
        this.alert=null;
    }

    public void setVar(String key, Object value) {
        this.vars.put(key,value);
        logger.debug("context vars: '" + key + "'='" + value.toString() + "'");
    }

    public  T getVar(String name, Class type) {
        Object o = this.vars.get(name);
        if (o==null) { return null; }

        if (type.isAssignableFrom(o.getClass())) {
            return type.cast(o);
        }
        throw new RuntimeException("Could not cast named var '" + name + "' to a " + type.getCanonicalName());
    }
}