co.paralleluniverse.fibers.okhttp.RecordingCallback.java Source code

Introduction

Here is the source code for co.paralleluniverse.fibers.okhttp.RecordingCallback.java

Source

/*
 * COMSAT
 * Copyright (c) 2013-2015, Parallel Universe Software Co. All rights reserved.
 *
 * This program and the accompanying materials are dual-licensed under
 * either the terms of the Eclipse Public License v1.0 as published by
 * the Eclipse Foundation
 *
 *   or (per the licensee's choosing)
 *
 * under the terms of the GNU Lesser General Public License version 3.0
 * as published by the Free Software Foundation.
 */
/*
 * Based on the corresponding class in okhttp-tests.
 * Copyright 2014 Square, Inc.
 * Licensed under the Apache License, Version 2.0 (the "License").
 */
package co.paralleluniverse.fibers.okhttp;

import com.squareup.okhttp.Callback;
import com.squareup.okhttp.Request;
import com.squareup.okhttp.Response;
import com.squareup.okhttp.ResponseBody;
import java.io.IOException;
import java.net.URL;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.concurrent.TimeUnit;
import okio.Buffer;

/**
 * Records received HTTP responses so they can be later retrieved by tests.
 */
public class RecordingCallback implements Callback {
    public static final long TIMEOUT_MILLIS = TimeUnit.SECONDS.toMillis(10);

    private final List<RecordedResponse> responses = new ArrayList<>();

    @Override
    public synchronized void onFailure(Request request, IOException e) {
        responses.add(new RecordedResponse(request, null, null, null, e));
        notifyAll();
    }

    @Override
    public synchronized void onResponse(Response response) throws IOException {
        Buffer buffer = new Buffer();
        ResponseBody body = response.body();
        body.source().readAll(buffer);

        responses.add(new RecordedResponse(response.request(), response, null, buffer.readUtf8(), null));
        notifyAll();
    }

    /**
     * Returns the recorded response triggered by {@code request}. Throws if the
     * response isn't enqueued before the timeout.
     */
    public synchronized RecordedResponse await(URL url) throws Exception {
        long timeoutMillis = TimeUnit.NANOSECONDS.toMillis(System.nanoTime()) + TIMEOUT_MILLIS;
        while (true) {
            for (Iterator<RecordedResponse> i = responses.iterator(); i.hasNext();) {
                RecordedResponse recordedResponse = i.next();
                if (recordedResponse.request.url().equals(url)) {
                    i.remove();
                    return recordedResponse;
                }
            }

            long nowMillis = TimeUnit.NANOSECONDS.toMillis(System.nanoTime());
            if (nowMillis >= timeoutMillis)
                break;
            wait(timeoutMillis - nowMillis);
        }

        throw new AssertionError("Timed out waiting for response to " + url);
    }

    public synchronized void assertNoResponse(URL url) throws Exception {
        for (RecordedResponse recordedResponse : responses) {
            if (recordedResponse.request.url().equals(url)) {
                throw new AssertionError("Expected no response for " + url);
            }
        }
    }
}