Polishing

Issue: SPR-14952
This commit is contained in:
Sebastien Deleuze 2016-11-25 12:54:48 +01:00
parent 4660ef701e
commit 0fcae5d2c5
3 changed files with 24 additions and 24 deletions

View File

@ -50,9 +50,9 @@ public class MockServerHttpResponse implements ServerHttpResponse {
private final MultiValueMap<String, ResponseCookie> cookies = new LinkedMultiValueMap<>();
private Publisher<? extends DataBuffer> body;
private Flux<DataBuffer> body;
private Publisher<? extends Publisher<? extends DataBuffer>> bodyWithFlushes;
private Flux<Publisher<DataBuffer>> bodyWithFlushes;
private DataBufferFactory bufferFactory = new DefaultDataBufferFactory();
@ -77,24 +77,24 @@ public class MockServerHttpResponse implements ServerHttpResponse {
return this.cookies;
}
public Publisher<? extends DataBuffer> getBody() {
public Publisher<DataBuffer> getBody() {
return this.body;
}
public Publisher<? extends Publisher<? extends DataBuffer>> getBodyWithFlush() {
public Publisher<Publisher<DataBuffer>> getBodyWithFlush() {
return this.bodyWithFlushes;
}
@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
this.body = body;
return Flux.from(this.body).then();
this.body = Flux.from(body);
return this.body.then();
}
@Override
public Mono<Void> writeAndFlushWith(Publisher<? extends Publisher<? extends DataBuffer>> body) {
this.bodyWithFlushes = body;
return Flux.from(this.bodyWithFlushes).then();
this.bodyWithFlushes = Flux.from(body).map(p -> Flux.from(p));
return this.bodyWithFlushes.then();
}
@Override

View File

@ -49,9 +49,9 @@ public class MockServerHttpResponse implements ServerHttpResponse {
private final MultiValueMap<String, ResponseCookie> cookies = new LinkedMultiValueMap<>();
private Publisher<? extends DataBuffer> body;
private Flux<DataBuffer> body;
private Publisher<? extends Publisher<? extends DataBuffer>> bodyWithFlushes;
private Flux<Publisher<DataBuffer>> bodyWithFlushes;
private DataBufferFactory bufferFactory = new DefaultDataBufferFactory();
@ -77,24 +77,24 @@ public class MockServerHttpResponse implements ServerHttpResponse {
return this.cookies;
}
public Publisher<? extends DataBuffer> getBody() {
public Publisher<DataBuffer> getBody() {
return this.body;
}
public Publisher<? extends Publisher<? extends DataBuffer>> getBodyWithFlush() {
public Publisher<Publisher<DataBuffer>> getBodyWithFlush() {
return this.bodyWithFlushes;
}
@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
this.body = body;
return Flux.from(this.body).then();
this.body = Flux.from(body);
return this.body.then();
}
@Override
public Mono<Void> writeAndFlushWith(Publisher<? extends Publisher<? extends DataBuffer>> body) {
this.bodyWithFlushes = body;
return Flux.from(this.bodyWithFlushes).then();
this.bodyWithFlushes = Flux.from(body).map(p -> Flux.from(p));
return this.bodyWithFlushes.then();
}
@Override

View File

@ -41,9 +41,9 @@ public class MockClientHttpRequest extends AbstractClientHttpRequest {
private final DataBufferFactory bufferFactory = new DefaultDataBufferFactory();
private Publisher<? extends DataBuffer> body;
private Flux<DataBuffer> body;
private Publisher<? extends Publisher<? extends DataBuffer>> bodyWithFlushes;
private Flux<Publisher<DataBuffer>> bodyWithFlushes;
public MockClientHttpRequest() {
@ -91,21 +91,21 @@ public class MockClientHttpRequest extends AbstractClientHttpRequest {
@Override
public Mono<Void> writeWith(Publisher<? extends DataBuffer> body) {
this.body = body;
return applyBeforeCommit().then(Flux.from(this.body).then());
this.body = Flux.from(body);
return applyBeforeCommit().then(this.body.then());
}
@Override
public Mono<Void> writeAndFlushWith(Publisher<? extends Publisher<? extends DataBuffer>> body) {
this.bodyWithFlushes = body;
return applyBeforeCommit().then(Flux.from(this.bodyWithFlushes).then());
this.bodyWithFlushes = Flux.from(body).map(p -> Flux.from(p));
return applyBeforeCommit().then(this.bodyWithFlushes.then());
}
public Publisher<? extends DataBuffer> getBody() {
public Publisher<DataBuffer> getBody() {
return body;
}
public Publisher<? extends Publisher<? extends DataBuffer>> getBodyWithFlush() {
public Publisher<Publisher<DataBuffer>> getBodyWithFlush() {
return bodyWithFlushes;
}