6 changed files with 132 additions and 17 deletions
@ -0,0 +1,87 @@
@@ -0,0 +1,87 @@
|
||||
/* |
||||
* Copyright 2002-present the original author or authors. |
||||
* |
||||
* 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 |
||||
* |
||||
* https://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 org.springframework.http.client; |
||||
|
||||
import java.io.IOException; |
||||
import java.io.InputStream; |
||||
import java.net.URI; |
||||
import java.net.http.HttpClient; |
||||
import java.net.http.HttpRequest; |
||||
import java.net.http.HttpResponse; |
||||
import java.net.http.HttpTimeoutException; |
||||
import java.time.Duration; |
||||
import java.util.concurrent.CompletableFuture; |
||||
import java.util.concurrent.ExecutorService; |
||||
import java.util.concurrent.Executors; |
||||
|
||||
import org.junit.jupiter.api.AfterEach; |
||||
import org.junit.jupiter.api.BeforeEach; |
||||
import org.junit.jupiter.api.Test; |
||||
|
||||
import org.springframework.http.HttpHeaders; |
||||
import org.springframework.http.HttpMethod; |
||||
|
||||
import static org.assertj.core.api.Assertions.assertThatThrownBy; |
||||
import static org.mockito.Mockito.any; |
||||
import static org.mockito.Mockito.mock; |
||||
import static org.mockito.Mockito.when; |
||||
|
||||
/** |
||||
* Unit tests for {@link JdkClientHttpRequest}. |
||||
*/ |
||||
class JdkClientHttpRequestTests { |
||||
|
||||
private final HttpClient client = mock(HttpClient.class); |
||||
|
||||
private ExecutorService executor; |
||||
|
||||
|
||||
@BeforeEach |
||||
void setup() { |
||||
executor = Executors.newSingleThreadExecutor(); |
||||
} |
||||
|
||||
@AfterEach |
||||
void tearDown() { |
||||
executor.shutdownNow(); |
||||
} |
||||
|
||||
|
||||
@Test |
||||
void futureCancelledAfterTimeout() { |
||||
CompletableFuture<HttpResponse<InputStream>> future = new CompletableFuture<>(); |
||||
when(client.sendAsync(any(HttpRequest.class), any(HttpResponse.BodyHandler.class))).thenReturn(future); |
||||
|
||||
assertThatThrownBy(() -> createRequest(Duration.ofMillis(10)).executeInternal(new HttpHeaders(), null)) |
||||
.isExactlyInstanceOf(HttpTimeoutException.class); |
||||
} |
||||
|
||||
@Test |
||||
void futureCancelled() { |
||||
CompletableFuture<HttpResponse<InputStream>> future = new CompletableFuture<>(); |
||||
future.cancel(true); |
||||
when(client.sendAsync(any(HttpRequest.class), any(HttpResponse.BodyHandler.class))).thenReturn(future); |
||||
|
||||
assertThatThrownBy(() -> createRequest(null).executeInternal(new HttpHeaders(), null)) |
||||
.isExactlyInstanceOf(IOException.class); |
||||
} |
||||
|
||||
private JdkClientHttpRequest createRequest(Duration timeout) { |
||||
return new JdkClientHttpRequest(client, URI.create("http://abc.com"), HttpMethod.GET, executor, timeout); |
||||
} |
||||
|
||||
} |
||||
Loading…
Reference in new issue