/*
 * Copyright 2010 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.browser;

import com.google.common.collect.Lists;
import com.google.jstestdriver.BrowserInfo;
import com.google.jstestdriver.FakeJsTestDriverClient;
import com.google.jstestdriver.FileUploader;
import com.google.jstestdriver.ResponseStream;
import com.google.jstestdriver.SlaveBrowser;
import com.google.jstestdriver.model.JstdTestCase;
import com.google.jstestdriver.util.NullStopWatch;

import junit.framework.TestCase;

import java.util.Collection;
import java.util.Collections;

/**
 * @author corbinrsmith@gmail.com (Cory Smith)
 */
public class BrowserManagedRunnerTest extends TestCase {

  public void testCall() throws Exception {
    String browserId = "5";
    String serverAddress = "http://foo";
    
    final BrowserInfo browserInfo = new BrowserInfo();
    browserInfo.setId(Long.parseLong(browserId));
    browserInfo.setServerReceivedHeartbeat(true);
    browserInfo.setReady(true);
    final FakeJsTestDriverClient client =
        new FakeJsTestDriverClient(Lists.newArrayList(browserInfo));
    final FakeBrowserRunner runner = new FakeBrowserRunner();
    final FakeBrowserActionRunner browserActionRunner = new FakeBrowserActionRunner();
    int browserTimeout = 70000;
    runCaptureUrlTest(browserId, serverAddress, client, runner, browserActionRunner, browserTimeout);
    assertTrue(runner.serverAddress,
      runner.serverAddress.contains(String.valueOf(browserTimeout)));
    runCaptureUrlTest(browserId, serverAddress, client, runner, browserActionRunner, 1);
    assertTrue(runner.serverAddress,
      runner.serverAddress.contains(String.valueOf(SlaveBrowser.TIMEOUT)));
  }

  /**
   * @param browserId
   * @param serverAddress
   * @param client
   * @param runner
   * @param browserActionRunner
   * @param browserTimeout
   * @throws Exception
   */
  private void runCaptureUrlTest(String browserId, String serverAddress,
      final FakeJsTestDriverClient client, final FakeBrowserRunner runner,
      final FakeBrowserActionRunner browserActionRunner, int browserTimeout) throws Exception {
    final BrowserCallable<Collection<ResponseStream>> browserRunner =
        new BrowserCallable<Collection<ResponseStream>>(browserActionRunner, browserId,
            new BrowserControl(runner, serverAddress, new NullStopWatch(), client,
                Collections.<JstdTestCase>emptyList(), browserTimeout));
    browserRunner.call();
  }

  private static final class FakeBrowserActionRunner extends BrowserActionRunner {
    public FakeBrowserActionRunner() {
      super(null, null, null, new NullStopWatch(), null, null);
    }
    
    @Override
    public Collection<ResponseStream> call() {
      return null;
    }
  }

  private static final class FakeBrowserRunner implements BrowserRunner {
    private String serverAddress;

    public void stopBrowser() {
      
    }

    public void startBrowser(String serverAddress) {
      this.serverAddress = serverAddress;
    }

    public int getTimeout() {
      return 0;
    }

    public int getNumStartupTries() {
      return 1;
    }

    public long getHeartbeatTimeout() {
      return SlaveBrowser.TIMEOUT;
    }

    public int getUploadSize() {
      return FileUploader.CHUNK_SIZE;
    }
  }
}