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

Merge branch '5.3.x'

Browse Source
This commit is contained in:
Arjen Poutsma
2022-08-30 15:15:57 +02:00
parent 3eaf259f2d f91943977c
commit 4b18cc94ea
2 changed files with 52 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
spring-webflux/src/test/java/org/springframework/web/reactive/function/server/DefaultRenderingResponseTests.java
View File

@@ -16,6 +16,7 @@
package org.springframework.web.reactive.function.server;
import java.time.Duration;
import java.time.ZonedDateTime;
import java.time.format.DateTimeFormatter;
import java.time.temporal.ChronoUnit;
@@ -37,6 +38,7 @@ import org.springframework.http.ResponseCookie;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.reactive.result.view.AbstractView;
import org.springframework.web.reactive.result.view.RedirectView;
import org.springframework.web.reactive.result.view.View;
import org.springframework.web.reactive.result.view.ViewResolver;
import org.springframework.web.reactive.result.view.ViewResolverSupport;
@@ -47,8 +49,10 @@ import org.springframework.web.testfixture.server.MockServerWebExchange;
import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.BDDMockito.given;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
/**
* @author Arjen Poutsma
@@ -155,6 +159,41 @@ public class DefaultRenderingResponseTests {
.verify();
}
@Test
public void writeTo() {
Map<String, Object> model = Collections.singletonMap("foo", "bar");
RenderingResponse renderingResponse = RenderingResponse.create("view")
.status(HttpStatus.FOUND)
.modelAttributes(model)
.build().block(Duration.of(5, ChronoUnit.MILLIS));
assertThat(renderingResponse).isNotNull();
MockServerWebExchange exchange = MockServerWebExchange.from(MockServerHttpRequest.get("http://localhost"));
MediaType contentType = MediaType.APPLICATION_PDF;
exchange.getResponse().getHeaders().setContentType(contentType);
ViewResolver viewResolver = mock(ViewResolver.class);
RedirectView view = mock(RedirectView.class);
given(viewResolver.resolveViewName(eq("view"), any())).willReturn(Mono.just(view));
given(view.render(model, contentType, exchange)).willReturn(Mono.empty());
List<ViewResolver> viewResolvers = new ArrayList<>();
viewResolvers.add(viewResolver);
HandlerStrategies mockConfig = mock(HandlerStrategies.class);
given(mockConfig.viewResolvers()).willReturn(viewResolvers);
ServerResponse.Context context = mock(ServerResponse.Context.class);
given(context.viewResolvers()).willReturn(viewResolvers);
Mono<Void> result = renderingResponse.writeTo(exchange, context);
StepVerifier.create(result)
.expectComplete()
.verify();
verify(view).setStatusCode(HttpStatus.FOUND);
}
@Test
public void defaultContentType() {
Mono<RenderingResponse> result = RenderingResponse.create("view").build();