org.springframework.mock.http.server.reactive.MockServerHttpResponse Java Examples

@Override
public Mono<ClientHttpResponse> connect(HttpMethod httpMethod, URI uri,
		Function<? super ClientHttpRequest, Mono<Void>> requestCallback) {

	MonoProcessor<ClientHttpResponse> result = MonoProcessor.create();

	MockClientHttpRequest mockClientRequest = new MockClientHttpRequest(httpMethod, uri);
	MockServerHttpResponse mockServerResponse = new MockServerHttpResponse();

	mockClientRequest.setWriteHandler(requestBody -> {
		log("Invoking HttpHandler for ", httpMethod, uri);
		ServerHttpRequest mockServerRequest = adaptRequest(mockClientRequest, requestBody);
		ServerHttpResponse responseToUse = prepareResponse(mockServerResponse, mockServerRequest);
		this.handler.handle(mockServerRequest, responseToUse).subscribe(aVoid -> {}, result::onError);
		return Mono.empty();
	});

	mockServerResponse.setWriteHandler(responseBody ->
			Mono.fromRunnable(() -> {
				log("Creating client response for ", httpMethod, uri);
				result.onNext(adaptResponse(mockServerResponse, responseBody));
			}));

	log("Writing client request for ", httpMethod, uri);
	requestCallback.apply(mockClientRequest).subscribe(aVoid -> {}, result::onError);

	return result;
}
 

private ClientHttpResponse adaptResponse(MockServerHttpResponse response, Flux<DataBuffer> body) {
	HttpStatus status = Optional.ofNullable(response.getStatusCode()).orElse(HttpStatus.OK);
	MockClientHttpResponse clientResponse = new MockClientHttpResponse(status);
	clientResponse.getHeaders().putAll(response.getHeaders());
	clientResponse.getCookies().putAll(response.getCookies());
	clientResponse.setBody(body);
	return clientResponse;
}
 

@Override
public Mono<ClientHttpResponse> connect(HttpMethod httpMethod, URI uri,
		Function<? super ClientHttpRequest, Mono<Void>> requestCallback) {

	MonoProcessor<ClientHttpResponse> result = MonoProcessor.create();

	MockClientHttpRequest mockClientRequest = new MockClientHttpRequest(httpMethod, uri);
	MockServerHttpResponse mockServerResponse = new MockServerHttpResponse();

	mockClientRequest.setWriteHandler(requestBody -> {
		log("Invoking HttpHandler for ", httpMethod, uri);
		ServerHttpRequest mockServerRequest = adaptRequest(mockClientRequest, requestBody);
		ServerHttpResponse responseToUse = prepareResponse(mockServerResponse, mockServerRequest);
		this.handler.handle(mockServerRequest, responseToUse).subscribe(aVoid -> {}, result::onError);
		return Mono.empty();
	});

	mockServerResponse.setWriteHandler(responseBody ->
			Mono.fromRunnable(() -> {
				log("Creating client response for ", httpMethod, uri);
				result.onNext(adaptResponse(mockServerResponse, responseBody));
			}));

	log("Writing client request for ", httpMethod, uri);
	requestCallback.apply(mockClientRequest).subscribe(aVoid -> {}, result::onError);

	return result;
}
 

private ClientHttpResponse adaptResponse(MockServerHttpResponse response, Flux<DataBuffer> body) {
	HttpStatus status = Optional.ofNullable(response.getStatusCode()).orElse(HttpStatus.OK);
	MockClientHttpResponse clientResponse = new MockClientHttpResponse(status);
	clientResponse.getHeaders().putAll(response.getHeaders());
	clientResponse.getCookies().putAll(response.getCookies());
	clientResponse.setBody(body);
	return clientResponse;
}
 

@Before
public void setup() {
	MockServerHttpRequest request = MockServerHttpRequest.post("/path").build();
	MockServerHttpResponse response = new MockServerHttpResponse();
	DefaultWebSessionManager sessionManager = new DefaultWebSessionManager();
	exchange = new DefaultServerWebExchange(request, response, sessionManager);
	pattern = "/pattern";
	path = "/path";

	matcher = new PathMatcherServerWebExchangeMatcher(pattern);
	matcher.setPathMatcher(pathMatcher);
}
 

@Test
public void notAbstractServerHttpResponse() {
	ServerWebExchange mockExchange = mock(ServerWebExchange.class);
	ServerHttpResponseDecorator responseDecorator = new ServerHttpResponseDecorator(
			new MockServerHttpResponse());
	responseDecorator.setStatusCode(OK);

	when(mockExchange.getRequest())
			.thenReturn(MockServerHttpRequest.get(ROUTE_URI).build());
	when(mockExchange.getResponse()).thenReturn(responseDecorator);

	Tags tags = tagsProvider.apply(mockExchange);
	assertThat(tags).isEqualTo(DEFAULT_TAGS);
}
 

@Test
void shouldHandleResponseStatusException() {
    MockServerHttpResponse response = exchange.getResponse();
    handler.handle(exchange, new ResponseStatusException(HttpStatus.CONFLICT)).block();

    assertThat(response.getStatusCode(), is(HttpStatus.CONFLICT));
    assertThat(response.getHeaders().getContentType(), is(MediaTypes.PROBLEM));
    assertThat(response.getBodyAsString().block(), containsString(Status.CONFLICT.getReasonPhrase()));
}
 

@Test
void shouldSetHttpResponse() {
    MockServerHttpResponse response = webExchange.getResponse();

    AdviceUtils.setHttpResponse(responseEntity, webExchange, mapper)
            .map(it -> response.setComplete())
            .block();

    assertThat(response.getStatusCode(), is(HttpStatus.CONFLICT));
    assertThat(response.getHeaders().getContentType(), is(MediaTypes.PROBLEM));
    assertThat(response.getBodyAsString().block(), containsString(Status.CONFLICT.getReasonPhrase()));

}
 

private MockServerWebExchange(MockServerHttpRequest request, WebSessionManager sessionManager) {
	super(request, new MockServerHttpResponse(), sessionManager,
			ServerCodecConfigurer.create(), new AcceptHeaderLocaleContextResolver());
}
 

@Override
public MockServerHttpResponse getResponse() {
	return (MockServerHttpResponse) super.getResponse();
}
 

private MockServerWebExchange(MockServerHttpRequest request, WebSessionManager sessionManager) {
	super(request, new MockServerHttpResponse(), sessionManager,
			ServerCodecConfigurer.create(), new AcceptHeaderLocaleContextResolver());
}
 

@Override
public MockServerHttpResponse getResponse() {
	return (MockServerHttpResponse) super.getResponse();
}