TestLauncher.java

/**
 * Sahi - Web Automation and Test Tool
 *
 * Copyright  2006  V Narayan Raman
 *
 * 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.sf.sahi.test;



import net.sf.sahi.config.Configuration;
import net.sf.sahi.playback.ScriptFactory;
import net.sf.sahi.nashorn.NashornScriptRunner;
import net.sf.sahi.session.Session;
import net.sf.sahi.session.Status;
import org.apache.log4j.Logger;

/**
 * @author nraman Launches browser with test and kills it on completion
 */
public class TestLauncher {

  private final String scriptName;

  private final String startURL;

  private String sessionId;

  private String childSessionId;

  private String browser;

  private static Logger logger = Logger.getLogger(TestLauncher.class);

  private String browserOption;

  private int threadNo;

  @SuppressWarnings("unused")
  private boolean isMultiThreaded;

  private NashornScriptRunner scriptRunner;

  private String browserProcessName;

  private BrowserLauncher browserLauncher;

  private boolean useSystemProxy;

  private boolean isSingleSession;

  public TestLauncher(final String scriptName, final String startURL) {
    this.scriptName = scriptName;
    this.startURL = startURL;
  }

  public void setSessionId(final String sessionId, String childSessionId) {
    this.sessionId = sessionId;
    this.childSessionId = childSessionId;
  }

  public void setBrowser(final String browser) {
    this.browser = browser;
  }

  public void setBrowserOption(final String browserOption) {
    this.browserOption = browserOption;
  }

  public String getStartURL() {
    return startURL;
  }

  public String getScriptName() {
    return scriptName;
  }

  public void execute(Session session) throws Exception {
    this.execute(session, true, true);
  }

  public void execute(Session session, boolean async, boolean setDefaultReporters) throws Exception {
    logger.debug("Running Script: " + scriptName);
    scriptRunner = new NashornScriptRunner(new ScriptFactory().getScript(scriptName), session.getSuite(), this, setDefaultReporters);
    session.setScriptRunner(scriptRunner);
    if (!isSingleSession) {
      launchBrowser();
    }
    if (async) scriptRunner.execute();
    else scriptRunner.executeAndWait(); // seems not to be used?
  }

  private void launchBrowser() throws Exception {
    if (browserOption != null) {
      browserOption = browserOption
        .replaceAll("[$]threadNo", "" + threadNo)
        .replaceAll("[$]userDir", Configuration.getAbsoluteUserPath(".").replace('\\', '/'))
        .replaceAll("[$]port", String.valueOf(Configuration.getPort()));
    }
    browserLauncher = new BrowserLauncher(browser, browserProcessName, browserOption, useSystemProxy);
    browserLauncher.openURL(browserLauncher.getPlayerAutoURL(childSessionId, startURL, isSingleSession));
  }

  public void kill() {
    logger.debug("Killing " + scriptName);
    if (!isSingleSession) browserLauncher.kill();
  }

  public String getChildSessionId() {
    return childSessionId;
  }

  public void setThreadNo(int threadNo, boolean isMultiThreaded) {
    this.threadNo = threadNo;
    this.isMultiThreaded = isMultiThreaded;
  }

  public int getThreadNo() {
    return threadNo;
  }

  public Status getStatus() {
    return scriptRunner.getScriptStatus();
  }

  public NashornScriptRunner getScriptRunner() {
    return scriptRunner;
  }

  public void setBrowserProcessName(String browserProcessName) {
    this.browserProcessName = browserProcessName;
  }

  public void setUseSystemProxy(boolean useSystemProxy) {
    this.useSystemProxy = useSystemProxy;
  }

  public void setIsSingleSession(boolean isSingleSession) {
    this.isSingleSession = isSingleSession;
  }
}