net.mindengine.galen.components.mocks.driver.MockedDriver.java Source code

Introduction

Here is the source code for net.mindengine.galen.components.mocks.driver.MockedDriver.java

Source

/*******************************************************************************
* Copyright 2015 Ivan Shubin http://mindengine.net
* 
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
* 
*   http://www.apache.org/licenses/LICENSE-2.0
* 
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
******************************************************************************/
package net.mindengine.galen.components.mocks.driver;

import com.fasterxml.jackson.databind.ObjectMapper;
import net.mindengine.galen.utils.GalenUtils;
import net.mindengine.rainbow4j.Rainbow4J;
import org.openqa.selenium.*;
import org.openqa.selenium.logging.Logs;

import java.awt.image.BufferedImage;
import java.awt.image.DataBufferByte;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;

public class MockedDriver implements WebDriver, TakesScreenshot, JavascriptExecutor {
    private MockedDriverPage page;
    private String currrentUrl;

    private Dimension screenSize = new Dimension(1024, 768);
    private List<String> allExecutedJavascript = new LinkedList<String>();

    public MockedDriver() {
    }

    public MockedDriver(String initialUrl) {
        this.currrentUrl = initialUrl;
        get(initialUrl);
    }

    @Override
    public void get(String url) {
        this.currrentUrl = url;
        InputStream stream = getClass().getResourceAsStream(url);
        if (stream == null) {
            throw new RuntimeException("Cannot find resource: " + url);
        }

        ObjectMapper mapper = new ObjectMapper();
        try {
            this.page = mapper.readValue(stream, MockedDriverPage.class);
        } catch (IOException e) {
            throw new RuntimeException("Failed parsing page: " + url, e);
        }
    }

    @Override
    public String getCurrentUrl() {
        return this.currrentUrl;
    }

    @Override
    public String getTitle() {
        return page.getTitle();
    }

    @Override
    public List<WebElement> findElements(By by) {
        List<WebElement> elements = new LinkedList<WebElement>();

        for (MockedPageItem item : page.getItems()) {
            if (item.matches(by)) {
                elements.add(item.asWebElement());
            }
        }
        return elements;
    }

    @Override
    public WebElement findElement(By by) {
        List<WebElement> elements = findElements(by);
        if (elements.size() > 0) {
            return elements.get(0);
        } else
            throw new NoSuchElementException(by.toString());
    }

    @Override
    public String getPageSource() {
        return null;
    }

    @Override
    public void close() {

    }

    @Override
    public void quit() {

    }

    @Override
    public Set<String> getWindowHandles() {
        return null;
    }

    @Override
    public String getWindowHandle() {
        return null;
    }

    @Override
    public TargetLocator switchTo() {
        return null;
    }

    @Override
    public Navigation navigate() {
        return null;
    }

    @Override
    public Options manage() {
        return new Options() {
            @Override
            public void addCookie(Cookie cookie) {
            }

            @Override
            public void deleteCookieNamed(String s) {
            }

            @Override
            public void deleteCookie(Cookie cookie) {
            }

            @Override
            public void deleteAllCookies() {
            }

            @Override
            public Set<Cookie> getCookies() {
                return null;
            }

            @Override
            public Cookie getCookieNamed(String s) {
                return null;
            }

            @Override
            public Timeouts timeouts() {
                return null;
            }

            @Override
            public ImeHandler ime() {
                return null;
            }

            @Override
            public Window window() {
                return new Window() {
                    @Override
                    public void setSize(Dimension dimension) {
                        MockedDriver.this.screenSize = dimension;
                    }

                    @Override
                    public void setPosition(Point point) {

                    }

                    @Override
                    public Dimension getSize() {
                        return MockedDriver.this.screenSize;
                    }

                    @Override
                    public Point getPosition() {
                        return null;
                    }

                    @Override
                    public void maximize() {

                    }
                };
            }

            @Override
            public Logs logs() {
                return null;
            }
        };
    }

    @Override
    public <X> X getScreenshotAs(OutputType<X> xOutputType) throws WebDriverException {
        if (xOutputType.equals(OutputType.FILE)) {
            return (X) new File(getClass().getResource("/mocks/pages/screenshot.png").getFile());
        } else if (xOutputType.equals(OutputType.BYTES)) {
            File file = new File(getClass().getResource("/mocks/pages/screenshot.png").getFile());

            BufferedImage image = null;
            try {
                image = Rainbow4J.loadImage(file.getAbsolutePath());
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
            return (X) ((DataBufferByte) image.getData().getDataBuffer()).getData();

        } else
            throw new RuntimeException("Cannot make screenshot");
    }

    @Override
    public Object executeScript(String s, Object... objects) {
        allExecutedJavascript.add(s);

        if (s.equals(GalenUtils.JS_RETRIEVE_DEVICE_PIXEL_RATIO)) {
            return 1L;
        } else
            return null;
    }

    @Override
    public Object executeAsyncScript(String s, Object... objects) {
        allExecutedJavascript.add(s);
        return null;
    }

    public List<String> getAllExecutedJavascript() {
        return allExecutedJavascript;
    }
}