Examples with ResponseEntity cn.taketoday.http.ResponseEntity used on opensource projects

Example 1 with ResponseEntity

use of cn.taketoday.http.ResponseEntity in project today-infrastructure by TAKETODAY.

the class AbstractReactiveWebServerFactoryTests method givenAnInflightRequestWhenTheServerIsStoppedThenGracefulShutdownCallbackIsCalledWithRequestsActive.

@Test
void givenAnInflightRequestWhenTheServerIsStoppedThenGracefulShutdownCallbackIsCalledWithRequestsActive() throws Exception {
    AbstractReactiveWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingHandler blockingHandler = new BlockingHandler();
    this.webServer = factory.getWebServer(blockingHandler);
    this.webServer.start();
    Mono<ResponseEntity<Void>> request = getWebClient(this.webServer.getPort()).build().get().retrieve().toBodilessEntity();
    AtomicReference<ResponseEntity<Void>> responseReference = new AtomicReference<>();
    CountDownLatch responseLatch = new CountDownLatch(1);
    request.subscribe((response) -> {
        responseReference.set(response);
        responseLatch.countDown();
    });
    blockingHandler.awaitQueue();
    AtomicReference<GracefulShutdownResult> result = new AtomicReference<>();
    this.webServer.shutDownGracefully(result::set);
    assertThat(responseReference.get()).isNull();
    try {
        this.webServer.stop();
    } catch (Exception ex) {
    // Continue
    }
    System.out.println("Stopped");
    Awaitility.await().atMost(Duration.ofSeconds(5)).until(() -> GracefulShutdownResult.REQUESTS_ACTIVE == result.get());
    blockingHandler.completeOne();
}

Example 2 with ResponseEntity

use of cn.taketoday.http.ResponseEntity in project today-infrastructure by TAKETODAY.

the class AbstractReactiveWebServerFactoryTests method whenARequestIsActiveThenStopWillComplete.

@Test
void whenARequestIsActiveThenStopWillComplete() throws InterruptedException, BrokenBarrierException {
    AbstractReactiveWebServerFactory factory = getFactory();
    BlockingHandler blockingHandler = new BlockingHandler();
    this.webServer = factory.getWebServer(blockingHandler);
    this.webServer.start();
    Mono<ResponseEntity<Void>> request = getWebClient(this.webServer.getPort()).build().get().retrieve().toBodilessEntity();
    AtomicReference<ResponseEntity<Void>> responseReference = new AtomicReference<>();
    CountDownLatch responseLatch = new CountDownLatch(1);
    request.subscribe((response) -> {
        responseReference.set(response);
        responseLatch.countDown();
    });
    blockingHandler.awaitQueue();
    try {
        this.webServer.stop();
    } catch (Exception ex) {
    // Continue
    }
    blockingHandler.completeOne();
}

Example 3 with ResponseEntity

use of cn.taketoday.http.ResponseEntity in project today-infrastructure by TAKETODAY.

the class AbstractReactiveWebServerFactoryTests method whenARequestRemainsInFlightThenShutDownGracefullyDoesNotInvokeCallbackUntilTheRequestCompletes.

@Test
void whenARequestRemainsInFlightThenShutDownGracefullyDoesNotInvokeCallbackUntilTheRequestCompletes() throws Exception {
    AbstractReactiveWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingHandler blockingHandler = new BlockingHandler();
    this.webServer = factory.getWebServer(blockingHandler);
    this.webServer.start();
    Mono<ResponseEntity<Void>> request = getWebClient(this.webServer.getPort()).build().get().retrieve().toBodilessEntity();
    AtomicReference<ResponseEntity<Void>> responseReference = new AtomicReference<>();
    CountDownLatch responseLatch = new CountDownLatch(1);
    request.subscribe((response) -> {
        responseReference.set(response);
        responseLatch.countDown();
    });
    blockingHandler.awaitQueue();
    AtomicReference<GracefulShutdownResult> result = new AtomicReference<>();
    this.webServer.shutDownGracefully(result::set);
    assertThat(responseReference.get()).isNull();
    blockingHandler.completeOne();
    assertThat(responseLatch.await(5, TimeUnit.SECONDS)).isTrue();
    Awaitility.await().atMost(Duration.ofSeconds(30)).until(() -> GracefulShutdownResult.IDLE == result.get());
}

Example 4 with ResponseEntity

use of cn.taketoday.http.ResponseEntity in project today-framework by TAKETODAY.

the class HttpEntityMethodProcessorTests method handleReturnValueWithETagAndETagFilter.

// SPR-13423
@Test
public void handleReturnValueWithETagAndETagFilter() throws Exception {
    String eTagValue = "\"deadb33f8badf00d\"";
    String content = "body";
    Method method = getClass().getDeclaredMethod("handle");
    MethodParameter returnType = new MethodParameter(method, -1);
    FilterChain chain = (req, res) -> {
        ResponseEntity<String> returnValue = ResponseEntity.ok().eTag(eTagValue).body(content);
        try {
            ServletRequestContext requestToUse = new ServletRequestContext(null, (HttpServletRequest) req, (HttpServletResponse) res);
            new HttpEntityMethodProcessor(Collections.singletonList(new StringHttpMessageConverter()), null).handleReturnValue(requestToUse, new HandlerMethod(this, method), returnValue);
            assertThat(this.servletResponse.getContentAsString()).as("Response body was cached? It should be written directly to the raw response").isEqualTo(content);
        } catch (Exception ex) {
            throw new IllegalStateException(ex);
        }
    };
    this.servletRequest.setMethod("GET");
    new ShallowEtagHeaderFilter().doFilter(this.servletRequest, this.servletResponse, chain);
    assertThat(this.servletResponse.getStatus()).isEqualTo(200);
    assertThat(this.servletResponse.getHeader(HttpHeaders.ETAG)).isEqualTo(eTagValue);
    assertThat(this.servletResponse.getContentAsString()).isEqualTo(content);
}

Example 5 with ResponseEntity

use of cn.taketoday.http.ResponseEntity in project today-framework by TAKETODAY.

the class AbstractReactiveWebServerFactoryTests method givenAnInflightRequestWhenTheServerIsStoppedThenGracefulShutdownCallbackIsCalledWithRequestsActive.

@Test
void givenAnInflightRequestWhenTheServerIsStoppedThenGracefulShutdownCallbackIsCalledWithRequestsActive() throws Exception {
    AbstractReactiveWebServerFactory factory = getFactory();
    factory.setShutdown(Shutdown.GRACEFUL);
    BlockingHandler blockingHandler = new BlockingHandler();
    this.webServer = factory.getWebServer(blockingHandler);
    this.webServer.start();
    Mono<ResponseEntity<Void>> request = getWebClient(this.webServer.getPort()).build().get().retrieve().toBodilessEntity();
    AtomicReference<ResponseEntity<Void>> responseReference = new AtomicReference<>();
    CountDownLatch responseLatch = new CountDownLatch(1);
    request.subscribe((response) -> {
        responseReference.set(response);
        responseLatch.countDown();
    });
    blockingHandler.awaitQueue();
    AtomicReference<GracefulShutdownResult> result = new AtomicReference<>();
    this.webServer.shutDownGracefully(result::set);
    assertThat(responseReference.get()).isNull();
    try {
        this.webServer.stop();
    } catch (Exception ex) {
    // Continue
    }
    System.out.println("Stopped");
    Awaitility.await().atMost(Duration.ofSeconds(5)).until(() -> GracefulShutdownResult.REQUESTS_ACTIVE == result.get());
    blockingHandler.completeOne();
}