build-packaging/tomcat-server/org.eclipse.virgo.tomcat.smoketest/src/test/java/org/eclipse/virgo/tomcat/smoketest/UrlWaitLatch.java

/*******************************************************************************
 * Copyright (c) 2008, 2010 VMware Inc.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   VMware Inc. - initial contribution
 *******************************************************************************/

package org.eclipse.virgo.tomcat.smoketest;

import static org.junit.Assert.fail;

import org.apache.commons.httpclient.HttpClient;
import org.apache.commons.httpclient.HttpMethod;
import org.apache.commons.httpclient.UsernamePasswordCredentials;
import org.apache.commons.httpclient.auth.AuthScope;
import org.apache.commons.httpclient.methods.GetMethod;

public class UrlWaitLatch {

    public static final long HALF_SECOND = 500;

    public static final long ONE_MINUTE = 60 * 1000;

    public static void waitFor(String url) {
        waitFor(url, HALF_SECOND, ONE_MINUTE);
    }

    public static void waitFor(String url, String username, String password) {
        waitFor(url, username, password, HALF_SECOND, ONE_MINUTE);
    }

    public static void waitFor(String url, long interval, long duration) {
        HttpClient client = new HttpClient();
        wait(url, client, interval, duration);
    }

    public static void waitFor(String url, String username, String password, long interval, long duration) {
        HttpClient client = new HttpClient();
        client.getParams().setAuthenticationPreemptive(true);
        client.getState().setCredentials(AuthScope.ANY, new UsernamePasswordCredentials(username, password));
        wait(url, client, interval, duration);
    }

    private static void wait(String url, HttpClient client, long interval, long duration) {
        HttpMethod get = new GetMethod(url);

        try {
            long startTime = System.currentTimeMillis();

            int statusCode = 999;
            while (System.currentTimeMillis() - startTime < duration) {
                statusCode = client.executeMethod(get);
                if (statusCode < 400) {
                    return;
                }
                Thread.sleep(interval);
            }

            fail(String.format("After %d ms, status code was %d", duration, statusCode));
        } catch (Exception e) {
            throw new RuntimeException(e);
        } finally {
            get.releaseConnection();
        }
    }

}