All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
jvmTest.okhttp3.WholeOperationTimeoutTest Maven / Gradle / Ivy
/*
* Copyright (C) 2018 Square, 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 okhttp3;
import java.io.IOException;
import java.io.InterruptedIOException;
import java.net.HttpURLConnection;
import java.time.Duration;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
import java.util.concurrent.atomic.AtomicReference;
import mockwebserver3.MockResponse;
import mockwebserver3.MockWebServer;
import okhttp3.testing.Flaky;
import okio.BufferedSink;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Tag;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.Timeout;
import org.junit.jupiter.api.extension.RegisterExtension;
import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.fail;
@Timeout(30)
@Tag("Slow")
public final class WholeOperationTimeoutTest {
/** A large response body. Smaller bodies might successfully read after the socket is closed! */
private static final String BIG_ENOUGH_BODY = TestUtil.repeat('a', 64 * 1024);
@RegisterExtension public final OkHttpClientTestRule clientTestRule = new OkHttpClientTestRule();
private MockWebServer server;
private final OkHttpClient client = clientTestRule.newClient();
@BeforeEach
public void setUp(MockWebServer server) throws Exception {
this.server = server;
}
@Test public void defaultConfigIsNoTimeout() throws Exception {
Request request = new Request.Builder()
.url(server.url("/"))
.build();
Call call = client.newCall(request);
assertThat(call.timeout().timeoutNanos()).isEqualTo(0);
}
@Test public void configureClientDefault() throws Exception {
Request request = new Request.Builder()
.url(server.url("/"))
.build();
OkHttpClient timeoutClient = client.newBuilder()
.callTimeout(Duration.ofMillis(456))
.build();
Call call = timeoutClient.newCall(request);
assertThat(call.timeout().timeoutNanos()).isEqualTo(TimeUnit.MILLISECONDS.toNanos(456));
}
@Test public void timeoutWritingRequest() throws Exception {
server.enqueue(new MockResponse());
Request request = new Request.Builder()
.url(server.url("/"))
.post(sleepingRequestBody(500))
.build();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
try {
call.execute();
fail();
} catch (IOException e) {
assertThat(e.getMessage()).isEqualTo("timeout");
assertThat(call.isCanceled()).isTrue();
}
}
@Test public void timeoutWritingRequestWithEnqueue() throws Exception {
server.enqueue(new MockResponse());
Request request = new Request.Builder()
.url(server.url("/"))
.post(sleepingRequestBody(500))
.build();
final CountDownLatch latch = new CountDownLatch(1);
final AtomicReference exceptionRef = new AtomicReference<>();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
call.enqueue(new Callback() {
@Override public void onFailure(Call call, IOException e) {
exceptionRef.set(e);
latch.countDown();
}
@Override public void onResponse(Call call, Response response) throws IOException {
response.close();
latch.countDown();
}
});
latch.await();
assertThat(call.isCanceled()).isTrue();
assertThat(exceptionRef.get()).isNotNull();
}
@Test public void timeoutProcessing() throws Exception {
server.enqueue(new MockResponse()
.setHeadersDelay(500, TimeUnit.MILLISECONDS));
Request request = new Request.Builder()
.url(server.url("/"))
.build();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
try {
call.execute();
fail();
} catch (IOException e) {
assertThat(e.getMessage()).isEqualTo("timeout");
assertThat(call.isCanceled()).isTrue();
}
}
@Test public void timeoutProcessingWithEnqueue() throws Exception {
server.enqueue(new MockResponse()
.setHeadersDelay(500, TimeUnit.MILLISECONDS));
Request request = new Request.Builder()
.url(server.url("/"))
.build();
final CountDownLatch latch = new CountDownLatch(1);
final AtomicReference exceptionRef = new AtomicReference<>();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
call.enqueue(new Callback() {
@Override public void onFailure(Call call, IOException e) {
exceptionRef.set(e);
latch.countDown();
}
@Override public void onResponse(Call call, Response response) throws IOException {
response.close();
latch.countDown();
}
});
latch.await();
assertThat(call.isCanceled()).isTrue();
assertThat(exceptionRef.get()).isNotNull();
}
@Test public void timeoutReadingResponse() throws Exception {
server.enqueue(new MockResponse()
.setBody(BIG_ENOUGH_BODY));
Request request = new Request.Builder()
.url(server.url("/"))
.build();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
Response response = call.execute();
Thread.sleep(500);
try {
response.body().source().readUtf8();
fail();
} catch (IOException e) {
assertThat(e.getMessage()).isEqualTo("timeout");
assertThat(call.isCanceled()).isTrue();
}
}
@Test public void timeoutReadingResponseWithEnqueue() throws Exception {
server.enqueue(new MockResponse()
.setBody(BIG_ENOUGH_BODY));
Request request = new Request.Builder()
.url(server.url("/"))
.build();
final CountDownLatch latch = new CountDownLatch(1);
final AtomicReference exceptionRef = new AtomicReference<>();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
call.enqueue(new Callback() {
@Override public void onFailure(Call call, IOException e) {
latch.countDown();
}
@Override public void onResponse(Call call, Response response) throws IOException {
try {
Thread.sleep(500);
} catch (InterruptedException e) {
throw new AssertionError();
}
try {
response.body().source().readUtf8();
fail();
} catch (IOException e) {
exceptionRef.set(e);
} finally {
latch.countDown();
}
}
});
latch.await();
assertThat(call.isCanceled()).isTrue();
assertThat(exceptionRef.get()).isNotNull();
}
@Test public void singleTimeoutForAllFollowUpRequests() throws Exception {
server.enqueue(new MockResponse()
.setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
.setHeader("Location", "/b")
.setHeadersDelay(100, TimeUnit.MILLISECONDS));
server.enqueue(new MockResponse()
.setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
.setHeader("Location", "/c")
.setHeadersDelay(100, TimeUnit.MILLISECONDS));
server.enqueue(new MockResponse()
.setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
.setHeader("Location", "/d")
.setHeadersDelay(100, TimeUnit.MILLISECONDS));
server.enqueue(new MockResponse()
.setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
.setHeader("Location", "/e")
.setHeadersDelay(100, TimeUnit.MILLISECONDS));
server.enqueue(new MockResponse()
.setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
.setHeader("Location", "/f")
.setHeadersDelay(100, TimeUnit.MILLISECONDS));
server.enqueue(new MockResponse());
Request request = new Request.Builder()
.url(server.url("/a"))
.build();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
try {
call.execute();
fail();
} catch (IOException e) {
assertThat(e.getMessage()).isEqualTo("timeout");
assertThat(call.isCanceled()).isTrue();
}
}
@Test
public void timeoutFollowingRedirectOnNewConnection() throws Exception {
MockWebServer otherServer = new MockWebServer();
server.enqueue(
new MockResponse()
.setResponseCode(HttpURLConnection.HTTP_MOVED_TEMP)
.setHeader("Location", otherServer.url("/")));
otherServer.enqueue(new MockResponse().setHeadersDelay(500, TimeUnit.MILLISECONDS));
Request request = new Request.Builder().url(server.url("/")).build();
Call call = client.newCall(request);
call.timeout().timeout(250, TimeUnit.MILLISECONDS);
try {
call.execute();
fail();
} catch (IOException e) {
assertThat(e.getMessage()).isEqualTo("timeout");
assertThat(call.isCanceled()).isTrue();
}
}
@Flaky
@Test public void noTimeout() throws Exception {
// Flaky https://github.com/square/okhttp/issues/5304
server.enqueue(new MockResponse()
.setHeadersDelay(250, TimeUnit.MILLISECONDS)
.setBody(BIG_ENOUGH_BODY));
Request request = new Request.Builder()
.url(server.url("/"))
.post(sleepingRequestBody(250))
.build();
Call call = client.newCall(request);
call.timeout().timeout(2000, TimeUnit.MILLISECONDS);
Response response = call.execute();
Thread.sleep(250);
response.body().source().readUtf8();
response.close();
assertThat(call.isCanceled()).isFalse();
}
private RequestBody sleepingRequestBody(final int sleepMillis) {
return new RequestBody() {
@Override public MediaType contentType() {
return MediaType.parse("text/plain");
}
@Override public void writeTo(BufferedSink sink) throws IOException {
try {
sink.writeUtf8("abc");
sink.flush();
Thread.sleep(sleepMillis);
sink.writeUtf8("def");
} catch (InterruptedException e) {
throw new InterruptedIOException();
}
}
};
}
}
