Skip to content

Commit 578aca2

Browse files
committed
Polish "Use RestClient in tests and infrastructure code"
See gh-51120
1 parent 648e2fb commit 578aca2

1 file changed

Lines changed: 5 additions & 10 deletions

File tree

  • module/spring-boot-cloudfoundry/src/test/java/org/springframework/boot/cloudfoundry/autoconfigure/actuate/endpoint/servlet

module/spring-boot-cloudfoundry/src/test/java/org/springframework/boot/cloudfoundry/autoconfigure/actuate/endpoint/servlet/SecurityServiceTests.java

Lines changed: 5 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,6 @@
2525
import org.springframework.boot.cloudfoundry.autoconfigure.actuate.endpoint.AccessLevel;
2626
import org.springframework.boot.cloudfoundry.autoconfigure.actuate.endpoint.CloudFoundryAuthorizationException;
2727
import org.springframework.boot.cloudfoundry.autoconfigure.actuate.endpoint.CloudFoundryAuthorizationException.Reason;
28-
import org.springframework.boot.restclient.RestTemplateBuilder;
29-
import org.springframework.boot.restclient.test.MockServerRestTemplateCustomizer;
3028
import org.springframework.http.HttpStatus;
3129
import org.springframework.http.MediaType;
3230
import org.springframework.test.web.client.MockRestServiceServer;
@@ -60,24 +58,21 @@ class SecurityServiceTests {
6058

6159
@BeforeEach
6260
void setup() {
63-
MockServerRestTemplateCustomizer mockServerCustomizer = new MockServerRestTemplateCustomizer();
64-
RestTemplateBuilder builder = new RestTemplateBuilder(mockServerCustomizer);
65-
this.securityService = new SecurityService(RestClient.builder(builder.build()), CLOUD_CONTROLLER, false);
66-
this.server = mockServerCustomizer.getServer();
61+
RestClient.Builder builder = RestClient.builder();
62+
this.server = MockRestServiceServer.bindTo(builder).build();
63+
this.securityService = new SecurityService(builder, CLOUD_CONTROLLER, false);
6764
}
6865

6966
@Test
7067
void skipSslValidationWhenTrue() {
71-
RestTemplateBuilder builder = new RestTemplateBuilder();
72-
this.securityService = new SecurityService(RestClient.builder(builder.build()), CLOUD_CONTROLLER, true);
68+
this.securityService = new SecurityService(RestClient.builder(), CLOUD_CONTROLLER, true);
7369
assertThat(this.securityService).extracting("restClient.clientRequestFactory")
7470
.isInstanceOf(SkipSslVerificationHttpRequestFactory.class);
7571
}
7672

7773
@Test
7874
void doNotSkipSslValidationWhenFalse() {
79-
RestTemplateBuilder builder = new RestTemplateBuilder();
80-
this.securityService = new SecurityService(RestClient.builder(builder.build()), CLOUD_CONTROLLER, false);
75+
this.securityService = new SecurityService(RestClient.builder(), CLOUD_CONTROLLER, false);
8176
assertThat(this.securityService).extracting("restClient.clientRequestFactory")
8277
.isNotInstanceOf(SkipSslVerificationHttpRequestFactory.class);
8378
}

0 commit comments

Comments
 (0)