FileSetGetHandlerTest.java

/*
 * Copyright 2011 Google Inc.
 * 
 * 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 com.google.jstestdriver.server.handlers;

import com.google.jstestdriver.BrowserInfo;
import com.google.jstestdriver.CapturedBrowsers;
import com.google.jstestdriver.MockTime;
import com.google.jstestdriver.SlaveBrowser;
import com.google.jstestdriver.SlaveBrowser.BrowserState;
import com.google.jstestdriver.Time;
import com.google.jstestdriver.browser.BrowserIdStrategy;
import com.google.jstestdriver.model.NullPathPrefix;
import com.google.jstestdriver.runner.RunnerType;

import junit.framework.TestCase;

import org.joda.time.Instant;

import java.io.IOException;
import java.io.PrintWriter;
import java.io.StringWriter;



/**
 * @author corysmith@google.com (Cory Smith)
 *
 */
public class FileSetGetHandlerTest extends TestCase {
  public void testQueueingSingleThreaded() throws Exception {
    String browserId = "foo";
    MockTime time = new MockTime(System.currentTimeMillis());
    SlaveBrowser browser = new SlaveBrowser(time,
        browserId,
        new BrowserInfo(),
        10000,
        new NullPathPrefix(),
        "quirks",
        RunnerType.CLIENT,
        BrowserState.READY,
        new Instant(0));
    CapturedBrowsers capturedBrowsers = new CapturedBrowsers(
        new BrowserIdStrategy(time));
    capturedBrowsers.addSlave(browser);

    String sessionIdOne = doRequest(browserId, null, "start", capturedBrowsers, time);
    assertNotNull("Expected the response to be not null.", sessionIdOne);
    assertFalse("Expected sessionId, got failed", "FAILED".equals(sessionIdOne));

    String failed = doRequest(browserId, null, "start", capturedBrowsers, time);
    assertTrue("Expected to be failed, but got:" + failed, "FAILED".equals(failed));
    for (int i = 0; i < 4; i++) {
      time.add(SlaveBrowser.SESSION_TIMEOUT - 1);
      //heartbeat
      doRequest(browserId, sessionIdOne, null, capturedBrowsers, time);
      failed = doRequest(browserId, null, "start", capturedBrowsers, time);
      assertTrue("Expected to be failed because the session is active, but got:" + failed, "FAILED".equals(failed));
    }
    doRequest(browserId, sessionIdOne, "stop", capturedBrowsers, time);
    String sessionIdTwo = doRequest(browserId, null, "start", capturedBrowsers, time);
    assertNotNull("Expected the response to be not null.", sessionIdTwo);
    assertFalse("Expected sessionId, got failed", "FAILED".equals(sessionIdTwo));
  }

  public void testQueueingSingleThreadedTimeoutForceUnlock() throws Exception {
    String browserId = "foo";
    MockTime time = new MockTime(System.currentTimeMillis());
    SlaveBrowser browser =
        new SlaveBrowser(time, browserId, new BrowserInfo(), 10000, new NullPathPrefix(), "quirks",
            RunnerType.CLIENT, BrowserState.READY, new Instant(0));
    CapturedBrowsers capturedBrowsers = new CapturedBrowsers(new BrowserIdStrategy(time));
    capturedBrowsers.addSlave(browser);

    String sessionIdOne = doRequest(browserId, null, "start", capturedBrowsers, time);
    assertNotNull("Expected the response to be not null.", sessionIdOne);
    assertFalse("Expected sessionId, got failed", "FAILED".equals(sessionIdOne));

    String failed = doRequest(browserId, null, "start", capturedBrowsers, time);
    assertTrue("Expected to be failed, but got:" + failed, "FAILED".equals(failed));
    time.add(SlaveBrowser.SESSION_TIMEOUT + 2);
    String sessionIdTwo = doRequest(browserId, null, "start", capturedBrowsers, time);
    assertNotNull("Expected the response to be not null.", sessionIdTwo);
    assertFalse("Expected sessionId, got failed", "FAILED".equals(sessionIdTwo));
  }


  private String doRequest(String id, String sessionId, String session,
      CapturedBrowsers capturedBrowsers, Time time) throws IOException {
    FakeHttpServletRequest request = new FakeHttpServletRequest();
    request.setParameter("id", id);
    request.setParameter("session", session);
    request.setParameter("sessionId", sessionId);

    StringWriter stringWriter = new StringWriter();
    new FileSetGetHandler(request, new PrintWriter(stringWriter), capturedBrowsers, time)
        .handleIt();
    return stringWriter.toString();
  }
}