org.xframium.page.keyWord.step.spi.KWSDumpState.java Source code

Introduction

Here is the source code for org.xframium.page.keyWord.step.spi.KWSDumpState.java

Source

/*******************************************************************************
 * xFramium
 *
 * Copyright 2016 by Moreland Labs, Ltd. (http://www.morelandlabs.com)
 *
 * Some open source application is free software: you can redistribute 
 * it and/or modify it under the terms of the GNU General Public 
 * License as published by the Free Software Foundation, either 
 * version 3 of the License, or (at your option) any later version.
 *  
 * Some open source application is distributed in the hope that it will 
 * be useful, but WITHOUT ANY WARRANTY; without even the implied warranty 
 * of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *  
 * You should have received a copy of the GNU General Public License
 * along with xFramium.  If not, see <http://www.gnu.org/licenses/>.
 *
 * @license GPL-3.0+ <http://spdx.org/licenses/GPL-3.0+>
 *******************************************************************************/
package org.xframium.page.keyWord.step.spi;

import java.io.BufferedOutputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.OutputStream;
import java.util.Map;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.xframium.application.ApplicationRegistry;
import org.xframium.device.data.DataManager;
import org.xframium.page.Page;
import org.xframium.page.PageManager;
import org.xframium.page.StepStatus;
import org.xframium.page.data.PageData;
import org.xframium.page.keyWord.step.AbstractKeyWordStep;
import org.xframium.spi.RunDetails;
import org.xframium.utility.XMLEscape;

public class KWSDumpState extends AbstractKeyWordStep {

    /* (non-Javadoc)
     * @see com.perfectoMobile.page.keyWord.step.AbstractKeyWordStep#_executeStep(com.perfectoMobile.page.Page, org.openqa.selenium.WebDriver, java.util.Map, java.util.Map)
     */
    @Override
    public synchronized boolean _executeStep(Page pageObject, WebDriver webDriver, Map<String, Object> contextMap,
            Map<String, PageData> dataMap, Map<String, Page> pageMap) {
        long startTime = System.currentTimeMillis();
        File rootFolder = new File(DataManager.instance().getReportFolder(), RunDetails.instance().getRootFolder());
        File useFolder = new File(rootFolder, "artifacts");
        useFolder.mkdirs();

        File screenFile = null;
        File domFile = null;
        ;

        if (webDriver instanceof TakesScreenshot) {
            OutputStream os = null;
            try {
                byte[] screenShot = ((TakesScreenshot) webDriver).getScreenshotAs(OutputType.BYTES);

                screenFile = File.createTempFile("state", ".png", useFolder);
                screenFile.getParentFile().mkdirs();
                os = new BufferedOutputStream(new FileOutputStream(screenFile));
                os.write(screenShot);
                os.flush();
                os.close();
            } catch (Exception e) {
                log.error("Error taking screenshot", e);
                try {
                    os.close();
                } catch (Exception e2) {
                }
            } finally {
                if (os != null)
                    try {
                        os.close();
                    } catch (Exception e) {
                    }
            }
        }

        FileOutputStream outputStream = null;
        try {
            File xmlFile = File.createTempFile("dom-", ".xml", useFolder);
            domFile = new File(xmlFile.getParentFile(), xmlFile.getName().replace(".xml", ".html"));
            String pageSource = webDriver.getPageSource();
            outputStream = new FileOutputStream(xmlFile);
            if (ApplicationRegistry.instance().getAUT().isWeb())
                outputStream.write(XMLEscape.toXML(pageSource).getBytes());
            else
                outputStream.write(XMLEscape.toHTML(pageSource).getBytes());

            outputStream.flush();
            outputStream.close();

            StringBuilder stringBuilder = new StringBuilder();
            stringBuilder.append(
                    "<html><head><link href=\"http://www.xframium.org/output/assets/css/prism.css\" rel=\"stylesheet\"><script src=\"http://www.xframium.org/output/assets/js/prism.js\"></script>");
            stringBuilder.append("</script></head><body><pre class\"line-numbers\"><code class=\"language-markup\">"
                    + pageSource.replace("<", "&lt;").replace(">", "&gt;").replace("\t", "  ")
                    + "</code></pre></body></html>");

            outputStream = new FileOutputStream(domFile);
            outputStream.write(stringBuilder.toString().getBytes());
            outputStream.flush();
            outputStream.close();
        } catch (Exception e) {
            throw new IllegalArgumentException("Could not write to file", e);
        } finally {
            if (outputStream != null)
                try {
                    outputStream.close();
                } catch (Exception e) {
                }
        }

        String files = domFile.getName() + (screenFile != null ? ("," + screenFile.getName()) : "");

        PageManager.instance().addExecutionLog(getExecutionId(webDriver), getDeviceName(webDriver), files,
                getName(), "KWSDumpState", startTime, System.currentTimeMillis() - startTime, StepStatus.SUCCESS,
                "", null, getThreshold(), getDescription(), false, null);

        return true;
    }

    @Override
    public boolean isRecordable() {
        return false;
    }

}