com.github.restdriver.clientdriver.integration.VerifyWithinTest.java Source code

Introduction

Here is the source code for com.github.restdriver.clientdriver.integration.VerifyWithinTest.java

Source

/**
 * Copyright  2010-2011 Nokia
 *
 * 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.github.restdriver.clientdriver.integration;

import static com.github.restdriver.clientdriver.RestClientDriver.*;
import static org.hamcrest.MatcherAssert.*;
import static org.hamcrest.Matchers.*;

import java.io.IOException;
import java.util.concurrent.TimeUnit;

import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.DefaultHttpClient;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.rules.ExpectedException;

import com.github.restdriver.clientdriver.ClientDriver;
import com.github.restdriver.clientdriver.ClientDriverFactory;
import com.github.restdriver.clientdriver.exception.ClientDriverFailedExpectationException;

public class VerifyWithinTest {

    private ClientDriver clientDriver;

    @Rule
    public ExpectedException thrown = ExpectedException.none();

    @Before
    public void before() {
        clientDriver = new ClientDriverFactory().createClientDriver();
    }

    @Test
    public void defaultBehaviourIsToVerifyImmediately() throws Exception {

        thrown.expect(ClientDriverFailedExpectationException.class);

        clientDriver.addExpectation(onRequestTo("/foo"), giveEmptyResponse());

        // Don't actually do anything
        clientDriver.verify();

    }

    @Test
    public void expectationWhichMatchesAnyTimesDoesNotWait() throws Exception {

        clientDriver.addExpectation(onRequestTo("/foo"), giveEmptyResponse().within(10, TimeUnit.SECONDS))
                .anyTimes();

        long start = System.currentTimeMillis();

        clientDriver.verify();

        long end = System.currentTimeMillis();

        assertThat(end - start, is(lessThan(5000L)));

    }

    @Test
    public void verifyingSingleRequestWithinATimePeriodWorks() throws Exception {

        clientDriver.addExpectation(onRequestTo("/foo"), giveEmptyResponse().within(2000, TimeUnit.MILLISECONDS));

        Thread thread = new Thread(new Runnable() {
            @Override
            public void run() {
                schnooze(500, TimeUnit.MILLISECONDS);
                hitThat(clientDriver.getBaseUrl() + "/foo");
            }
        });
        thread.setDaemon(true);

        thread.start();

        clientDriver.verify();

    }

    @Test
    public void singleRequestThatIsNotMatchedInTimeFailsToVerify() throws Exception {

        thrown.expect(ClientDriverFailedExpectationException.class);

        clientDriver.addExpectation(onRequestTo("/foo"), giveEmptyResponse().within(200, TimeUnit.MILLISECONDS));

        Thread thread = new Thread(new Runnable() {
            @Override
            public void run() {
                schnooze(500, TimeUnit.MILLISECONDS);
                hitThat(clientDriver.getBaseUrl() + "/foo");
            }
        });
        thread.setDaemon(true);

        thread.start();

        clientDriver.verify();

    }

    @Test
    public void multipleRequestsWhichAreMatchedInTimeWork() throws Exception {

        clientDriver.addExpectation(onRequestTo("/foo1"), giveEmptyResponse().within(5, TimeUnit.SECONDS));

        clientDriver.addExpectation(onRequestTo("/foo2"), giveEmptyResponse().within(5, TimeUnit.SECONDS));

        Thread thread1 = new Thread(new Runnable() {
            @Override
            public void run() {
                schnooze(500, TimeUnit.MILLISECONDS);
                hitThat(clientDriver.getBaseUrl() + "/foo1");
            }
        });
        thread1.setDaemon(true);

        Thread thread2 = new Thread(new Runnable() {
            @Override
            public void run() {
                schnooze(500, TimeUnit.MILLISECONDS);
                hitThat(clientDriver.getBaseUrl() + "/foo2");
            }
        });
        thread2.setDaemon(true);

        thread1.start();
        thread2.start();

        clientDriver.verify();

    }

    @Test
    public void multipleRequestsOneOfWhichWasNotMatchedInTimeFailsToVerify() throws Exception {

        thrown.expect(ClientDriverFailedExpectationException.class);

        clientDriver.addExpectation(onRequestTo("/foo1"), giveEmptyResponse().within(200, TimeUnit.MILLISECONDS));

        clientDriver.addExpectation(onRequestTo("/foo2"), giveEmptyResponse().within(1, TimeUnit.SECONDS));

        Thread thread1 = new Thread(new Runnable() {
            @Override
            public void run() {
                schnooze(500, TimeUnit.MILLISECONDS);
                hitThat(clientDriver.getBaseUrl() + "/foo1");
            }
        });
        thread1.setDaemon(true);

        Thread thread2 = new Thread(new Runnable() {
            @Override
            public void run() {
                schnooze(500, TimeUnit.MILLISECONDS);
                hitThat(clientDriver.getBaseUrl() + "/foo2");
            }
        });
        thread2.setDaemon(true);

        thread1.start();
        thread2.start();

        clientDriver.verify();

    }

    private static void schnooze(long interval, TimeUnit unit) {
        try {
            unit.sleep(interval);
        } catch (InterruptedException e) {
            throw new RuntimeException(e);
        }
    }

    private static void hitThat(String url) {
        try {
            HttpClient client = new DefaultHttpClient();
            HttpGet get = new HttpGet(url);
            client.execute(get);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

}