org.specrunner.htmlunit.actions.PluginOpen Maven / Gradle / Ivy

Go to download
/*
    SpecRunner - Acceptance Test Driven Development Tool
    Copyright (C) 2011-2016  Thiago Santos

    This program 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.

    This program 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 this program.  If not, see 
 */
package org.specrunner.htmlunit.actions;

import java.net.MalformedURLException;
import java.net.URL;

import nu.xom.Node;

import org.specrunner.context.IContext;
import org.specrunner.plugins.ActionType;
import org.specrunner.plugins.PluginException;
import org.specrunner.plugins.type.Command;
import org.specrunner.result.IResultSet;
import org.specrunner.result.status.Success;
import org.specrunner.util.UtilLog;

import com.gargoylesoftware.htmlunit.WebClient;

/**
 * Open a given URL.
 * 
 * @author Thiago Santos
 * 
 */
public class PluginOpen extends AbstractPluginUrlAware {

    @Override
    public ActionType getActionType() {
        return Command.INSTANCE;
    }

    @Override
    protected void doEnd(IContext context, IResultSet result, WebClient client) throws PluginException {
        Node node = context.getNode();
        String u = getUrl() != null ? getUrl() : node.getValue();
        String tmp = getBrowserName();
        if (!u.startsWith("http")) {
            String baseUrl = (String) context.getByName(PluginStartIn.getBaseForBrowser(tmp));
            if (baseUrl == null) {
                throw new PluginException("Could not find base url for browser '" + tmp + "'.");
            }
            String target;
            try {
                URL base = new URL(baseUrl);
                target = new URL(base, u).toString();
                if (UtilLog.LOG.isInfoEnabled()) {
                    UtilLog.LOG.info("Relative url resolved from '" + u + "' to '" + target + "'.");
                }
            } catch (MalformedURLException e) {
                throw new PluginException("Could not resolve '" + baseUrl + "' with " + u + ".", e);
            }
            u = target;
        }
        PluginException error = null;
        try {
            if (UtilLog.LOG.isInfoEnabled()) {
                UtilLog.LOG.info("Openning '" + u + "' on browser named '" + tmp + "'.");
            }
            client.getPage(u);
        } catch (Exception e) {
            error = new PluginException(e);
        }
        if (error != null && !isIgnorable(error)) {
            throw error;
        } else {
            result.addResult(Success.INSTANCE, context.newBlock(node, this));
        }
    }

    /**
     * Check if an error can be ignored.
     * 
     * @param error
     *            The error.
     * @return true, if ignorable, false, otherwise.
     */
    protected boolean isIgnorable(PluginException error) {
        String msg = error.getMessage();
        return msg.contains("ScriptException") || msg.endsWith(".js");
    }
}