Examples with ClientRequest org.springframework.web.reactive.function.client.ClientRequest used on opensource projects

Example 31 with ClientRequest

use of org.springframework.web.reactive.function.client.ClientRequest in project spring-boot by spring-projects.

the class MetricsWebClientFilterFunctionTests method filterWhenCancelAfterResponseThrownShouldNotRecordTimer.

@Test
void filterWhenCancelAfterResponseThrownShouldNotRecordTimer() {
    ClientRequest request = ClientRequest.create(HttpMethod.GET, URI.create("https://example.com/projects/spring-boot")).build();
    given(this.response.rawStatusCode()).willReturn(HttpStatus.OK.value());
    Mono<ClientResponse> filter = this.filterFunction.filter(request, this.exchange);
    StepVerifier.create(filter).expectNextCount(1).thenCancel().verify(Duration.ofSeconds(5));
    assertThat(this.registry.get("http.client.requests").tags("method", "GET", "uri", "/projects/spring-boot", "status", "200").timer().count()).isEqualTo(1);
    assertThatThrownBy(() -> this.registry.get("http.client.requests").tags("method", "GET", "uri", "/projects/spring-boot", "status", "CLIENT_ERROR").timer()).isInstanceOf(MeterNotFoundException.class);
}

Example 32 with ClientRequest

use of org.springframework.web.reactive.function.client.ClientRequest in project spring-boot by spring-projects.

the class MetricsWebClientFilterFunctionTests method filterShouldRecordTimer.

@Test
void filterShouldRecordTimer() {
    ClientRequest request = ClientRequest.create(HttpMethod.GET, URI.create("https://example.com/projects/spring-boot")).build();
    given(this.response.rawStatusCode()).willReturn(HttpStatus.OK.value());
    this.filterFunction.filter(request, this.exchange).block(Duration.ofSeconds(5));
    assertThat(this.registry.get("http.client.requests").tags("method", "GET", "uri", "/projects/spring-boot", "status", "200").timer().count()).isEqualTo(1);
}

Example 33 with ClientRequest

use of org.springframework.web.reactive.function.client.ClientRequest in project spring-security by spring-projects.

the class SecurityReactorContextConfigurationTests method createPublisherWhenLastOperatorAddedThenSecurityContextAttributesAvailable.

@Test
public void createPublisherWhenLastOperatorAddedThenSecurityContextAttributesAvailable() {
    // Trigger the importing of SecurityReactorContextConfiguration via
    // OAuth2ImportSelector
    this.spring.register(SecurityConfig.class).autowire();
    // Setup for SecurityReactorContextSubscriberRegistrar
    RequestContextHolder.setRequestAttributes(new ServletRequestAttributes(this.servletRequest, this.servletResponse));
    SecurityContextHolder.getContext().setAuthentication(this.authentication);
    ClientResponse clientResponseOk = ClientResponse.create(HttpStatus.OK).build();
    // @formatter:off
    ExchangeFilterFunction filter = (req, next) -> Mono.subscriberContext().filter((ctx) -> ctx.hasKey(SecurityReactorContextSubscriber.SECURITY_CONTEXT_ATTRIBUTES)).map((ctx) -> ctx.get(SecurityReactorContextSubscriber.SECURITY_CONTEXT_ATTRIBUTES)).cast(Map.class).map((attributes) -> {
        if (attributes.containsKey(HttpServletRequest.class) && attributes.containsKey(HttpServletResponse.class) && attributes.containsKey(Authentication.class)) {
            return clientResponseOk;
        } else {
            return ClientResponse.create(HttpStatus.NOT_FOUND).build();
        }
    });
    // @formatter:on
    ClientRequest clientRequest = ClientRequest.create(HttpMethod.GET, URI.create("https://example.com")).build();
    MockExchangeFunction exchange = new MockExchangeFunction();
    Map<Object, Object> expectedContextAttributes = new HashMap<>();
    expectedContextAttributes.put(HttpServletRequest.class, this.servletRequest);
    expectedContextAttributes.put(HttpServletResponse.class, this.servletResponse);
    expectedContextAttributes.put(Authentication.class, this.authentication);
    Mono<ClientResponse> clientResponseMono = filter.filter(clientRequest, exchange).flatMap((response) -> filter.filter(clientRequest, exchange));
    // @formatter:off
    StepVerifier.create(clientResponseMono).expectAccessibleContext().contains(SecurityReactorContextSubscriber.SECURITY_CONTEXT_ATTRIBUTES, expectedContextAttributes).then().expectNext(clientResponseOk).verifyComplete();
// @formatter:on
}

Example 34 with ClientRequest

use of org.springframework.web.reactive.function.client.ClientRequest in project spring-framework by spring-projects.

the class WiretapConnectorTests method captureAndClaim.

@Test
public void captureAndClaim() {
    ClientHttpRequest request = new MockClientHttpRequest(HttpMethod.GET, "/test");
    ClientHttpResponse response = new MockClientHttpResponse(HttpStatus.OK);
    ClientHttpConnector connector = (method, uri, fn) -> fn.apply(request).then(Mono.just(response));
    ClientRequest clientRequest = ClientRequest.create(HttpMethod.GET, URI.create("/test")).header(WebTestClient.WEBTESTCLIENT_REQUEST_ID, "1").build();
    WiretapConnector wiretapConnector = new WiretapConnector(connector);
    ExchangeFunction function = ExchangeFunctions.create(wiretapConnector);
    function.exchange(clientRequest).block(ofMillis(0));
    ExchangeResult result = wiretapConnector.getExchangeResult("1", null, Duration.ZERO);
    assertThat(result.getMethod()).isEqualTo(HttpMethod.GET);
    assertThat(result.getUrl().toString()).isEqualTo("/test");
}

Example 35 with ClientRequest

use of org.springframework.web.reactive.function.client.ClientRequest in project spring-security by spring-projects.

the class ServerOAuth2AuthorizedClientExchangeFilterFunctionTests method filterWhenDefaultOAuth2AuthorizedClientFalseThenEmpty.

@Test
public void filterWhenDefaultOAuth2AuthorizedClientFalseThenEmpty() {
    ClientRequest request = ClientRequest.create(HttpMethod.GET, URI.create("https://example.com")).build();
    OAuth2User user = new DefaultOAuth2User(AuthorityUtils.createAuthorityList("ROLE_USER"), Collections.singletonMap("user", "rob"), "user");
    OAuth2AuthenticationToken authentication = new OAuth2AuthenticationToken(user, user.getAuthorities(), "client-id");
    // @formatter:off
    this.function.filter(request, this.exchange).subscriberContext(ReactiveSecurityContextHolder.withAuthentication(authentication)).block();
    // @formatter:on
    List<ClientRequest> requests = this.exchange.getRequests();
    assertThat(requests).hasSize(1);
    verifyZeroInteractions(this.clientRegistrationRepository, this.authorizedClientRepository);
}