Spring/spring-framework
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Polishing in client tests

Browse Source
This commit is contained in:
rstoyanchev
2025-01-07 15:26:34 +00:00
parent 3f7d3cbbd3
commit b9efa91ecc
4 changed files with 47 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
spring-web/src/test/java/org/springframework/http/client/AbstractMockWebServerTests.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2024 the original author or authors.
* Copyright 2002-2025 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.
@@ -16,8 +16,6 @@
package org.springframework.http.client;
import java.util.Collections;
import okhttp3.mockwebserver.Dispatcher;
import okhttp3.mockwebserver.MockResponse;
import okhttp3.mockwebserver.MockWebServer;
@@ -25,7 +23,6 @@ import okhttp3.mockwebserver.RecordedRequest;
import org.junit.jupiter.api.AfterEach;
import org.junit.jupiter.api.BeforeEach;
import org.springframework.http.MediaType;
import org.springframework.util.StringUtils;
import static org.assertj.core.api.Assertions.assertThat;
@@ -41,8 +38,6 @@ public abstract class AbstractMockWebServerTests {
protected String baseUrl;
protected static final MediaType textContentType =
new MediaType("text", "plain", Collections.singletonMap("charset", "UTF-8"));
@BeforeEach
void setUp() throws Exception {
@@ -58,13 +53,14 @@ public abstract class AbstractMockWebServerTests {
this.server.shutdown();
}
protected class TestDispatcher extends Dispatcher {
@Override
public MockResponse dispatch(RecordedRequest request) {
try {
if (request.getPath().equals("/echo")) {
assertThat(request.getHeader("Host"))
.contains("localhost:" + port);
assertThat(request.getHeader("Host")).contains("localhost:" + port);
MockResponse response = new MockResponse()
.setHeaders(request.getHeaders())
.setHeader("Content-Length", request.getBody().size())
@@ -80,8 +76,7 @@ public abstract class AbstractMockWebServerTests {
return new MockResponse().setResponseCode(404);
}
else if (request.getPath().equals("/status/299")) {
assertThat(request.getHeader("Expect"))
.contains("299");
assertThat(request.getHeader("Expect")).contains("299");
return new MockResponse().setResponseCode(299);
}
else if(request.getPath().startsWith("/params")) {
@@ -112,8 +107,8 @@ public abstract class AbstractMockWebServerTests {
}
return new MockResponse().setResponseCode(404);
}
catch (Throwable exc) {
return new MockResponse().setResponseCode(500).setBody(exc.toString());
catch (Throwable ex) {
return new MockResponse().setResponseCode(500).setBody(ex.toString());
}
}
}

36
spring-web/src/test/java/org/springframework/http/client/BufferingClientHttpRequestFactoryTests.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2024 the original author or authors.
* Copyright 2002-2025 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.
@@ -18,7 +18,6 @@ package org.springframework.http.client;
import java.net.URI;
import java.nio.charset.StandardCharsets;
import java.util.Arrays;
import org.junit.jupiter.api.Test;
@@ -37,30 +36,29 @@ class BufferingClientHttpRequestFactoryTests extends AbstractHttpRequestFactoryT
@Test
void repeatableRead() throws Exception {
ClientHttpRequest request = factory.createRequest(URI.create(baseUrl + "/echo"), HttpMethod.PUT);
assertThat(request.getMethod()).as("Invalid HTTP method").isEqualTo(HttpMethod.PUT);
String headerName = "MyHeader";
String headerValue1 = "value1";
request.getHeaders().add(headerName, headerValue1);
String headerValue2 = "value2";
request.getHeaders().add(headerName, headerValue2);
assertThat(request.getMethod()).isEqualTo(HttpMethod.PUT);
String header = "MyHeader";
request.getHeaders().add(header, "value1");
request.getHeaders().add(header, "value2");
byte[] body = "Hello World".getBytes(StandardCharsets.UTF_8);
request.getHeaders().setContentLength(body.length);
FileCopyUtils.copy(body, request.getBody());
request.getHeaders().setContentLength(body.length);
try (ClientHttpResponse response = request.execute()) {
assertThat(response.getStatusCode()).as("Invalid status code").isEqualTo(HttpStatus.OK);
assertThat(response.getStatusCode()).as("Invalid status code").isEqualTo(HttpStatus.OK);
assertThat(response.getStatusCode()).isEqualTo(HttpStatus.OK);
assertThat(response.getHeaders().containsHeader(headerName)).as("Header not found").isTrue();
assertThat(response.getHeaders().containsHeader(headerName)).as("Header not found").isTrue();
assertThat(response.getHeaders().get(headerName)).as("Header value not found").isEqualTo(Arrays.asList(headerValue1, headerValue2));
assertThat(response.getHeaders().get(headerName)).as("Header value not found").isEqualTo(Arrays.asList(headerValue1, headerValue2));
assertThat(response.getHeaders().get(header)).containsExactly("value1", "value2");
assertThat(response.getHeaders().get(header)).containsExactly("value1", "value2");
byte[] result = FileCopyUtils.copyToByteArray(response.getBody());
assertThat(Arrays.equals(body, result)).as("Invalid body").isTrue();
FileCopyUtils.copyToByteArray(response.getBody());
assertThat(Arrays.equals(body, result)).as("Invalid body").isTrue();
assertThat(result).isEqualTo(body);
result = FileCopyUtils.copyToByteArray(response.getBody());
assertThat(result).isEqualTo(body);
}
}

27
spring-web/src/test/java/org/springframework/http/client/InterceptingClientHttpRequestFactoryTests.java
View File

@@ -56,7 +56,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void shouldInvokeInterceptors() throws Exception {
void invokeInterceptors() throws Exception {
List<ClientHttpRequestInterceptor> interceptors = new ArrayList<>();
interceptors.add(new NoOpInterceptor());
interceptors.add(new NoOpInterceptor());
@@ -74,7 +74,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void shouldSkipIntercetor() throws Exception {
void skipInterceptor() throws Exception {
List<ClientHttpRequestInterceptor> interceptors = new ArrayList<>();
interceptors.add((request, body, execution) -> responseMock);
interceptors.add(new NoOpInterceptor());
@@ -89,7 +89,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void interceptorShouldUpdateRequestHeader() throws Exception {
void updateRequestHeader() throws Exception {
final String headerName = "Foo";
final String headerValue = "Bar";
final String otherValue = "Baz";
@@ -115,7 +115,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void interceptorShouldUpdateRequestAttribute() throws Exception {
void updateRequestAttribute() throws Exception {
final String attrName = "Foo";
final String attrValue = "Bar";
@@ -137,7 +137,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void interceptorShouldUpdateRequestURI() throws Exception {
void updateRequestURI() throws Exception {
final URI changedUri = URI.create("https://example.com/2");
ClientHttpRequestInterceptor interceptor = (request, body, execution) -> execution.execute(new HttpRequestWrapper(request) {
@@ -161,7 +161,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void interceptorShouldUpdateRequestMethod() throws Exception {
void updateRequestMethod() throws Exception {
final HttpMethod changedMethod = HttpMethod.POST;
ClientHttpRequestInterceptor interceptor = (request, body, execution) -> execution.execute(new HttpRequestWrapper(request) {
@@ -185,7 +185,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void interceptorShouldUpdateRequestBody() throws Exception {
void updateRequestBody() throws Exception {
final byte[] changedBody = "Foo".getBytes();
ClientHttpRequestInterceptor interceptor = (request, body, execution) -> execution.execute(request, changedBody);
requestFactory = new InterceptingClientHttpRequestFactory(requestFactoryMock, Collections.singletonList(interceptor));
@@ -197,7 +197,7 @@ class InterceptingClientHttpRequestFactoryTests {
}
@Test
void interceptorShouldAlwaysExecuteNextInterceptor() throws Exception {
void multipleExecutions() throws Exception {
List<ClientHttpRequestInterceptor> interceptors = new ArrayList<>();
interceptors.add(new MultipleExecutionInterceptor());
interceptors.add(new NoOpInterceptor());
@@ -217,18 +217,21 @@ class InterceptingClientHttpRequestFactoryTests {
private int invocationCount = 0;
@Override
public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution)
throws IOException {
public ClientHttpResponse intercept(
HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
invocationCount++;
return execution.execute(request, body);
}
}
private static class MultipleExecutionInterceptor implements ClientHttpRequestInterceptor {
@Override
public ClientHttpResponse intercept(HttpRequest request, byte[] body, ClientHttpRequestExecution execution)
throws IOException {
public ClientHttpResponse intercept(
HttpRequest request, byte[] body, ClientHttpRequestExecution execution) throws IOException {
// execute another request first
execution.execute(new MockClientHttpRequest(), body);
return execution.execute(request, body);

12
spring-web/src/test/java/org/springframework/http/client/JdkClientHttpRequestFactoryTests.java
View File

@@ -1,5 +1,5 @@
/*
* Copyright 2002-2024 the original author or authors.
* Copyright 2002-2025 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.
@@ -39,6 +39,7 @@ class JdkClientHttpRequestFactoryTests extends AbstractHttpRequestFactoryTests {
private static @Nullable String originalPropertyValue;
@BeforeAll
static void setProperty() {
originalPropertyValue = System.getProperty("jdk.httpclient.allowRestrictedHeaders");
@@ -55,6 +56,7 @@ class JdkClientHttpRequestFactoryTests extends AbstractHttpRequestFactoryTests {
}
}
@Override
protected ClientHttpRequestFactory createRequestFactory() {
return new JdkClientHttpRequestFactory();
@@ -69,7 +71,8 @@ class JdkClientHttpRequestFactoryTests extends AbstractHttpRequestFactoryTests {
@Test
void customizeDisallowedHeaders() throws IOException {
ClientHttpRequest request = this.factory.createRequest(URI.create(this.baseUrl + "/status/299"), HttpMethod.PUT);
URI uri = URI.create(this.baseUrl + "/status/299");
ClientHttpRequest request = this.factory.createRequest(uri, HttpMethod.PUT);
request.getHeaders().set("Expect", "299");
try (ClientHttpResponse response = request.execute()) {
@@ -79,8 +82,9 @@ class JdkClientHttpRequestFactoryTests extends AbstractHttpRequestFactoryTests {
@Test // gh-31451
public void contentLength0() throws IOException {
BufferingClientHttpRequestFactory bufferingFactory = new BufferingClientHttpRequestFactory(this.factory);
ClientHttpRequest request = bufferingFactory.createRequest(URI.create(this.baseUrl + "/methods/get"), HttpMethod.GET);
URI uri = URI.create(this.baseUrl + "/methods/get");
ClientHttpRequest request =
new BufferingClientHttpRequestFactory(this.factory).createRequest(uri, HttpMethod.GET);
try (ClientHttpResponse response = request.execute()) {
assertThat(response.getStatusCode()).as("Invalid response status").isEqualTo(HttpStatus.OK);