Separate step for retrieve in RSocketRequester

Closes gh-24073
This commit is contained in:
Rossen Stoyanchev
2019-11-25 11:47:41 +00:00
parent b234c77b67
commit 6ed1b5835b
3 changed files with 42 additions and 32 deletions

View File

@@ -40,6 +40,7 @@ import reactor.test.StepVerifier;
import org.springframework.core.io.buffer.DefaultDataBufferFactory;
import org.springframework.lang.Nullable;
import org.springframework.messaging.rsocket.RSocketRequester.RequestSpec;
import org.springframework.messaging.rsocket.RSocketRequester.RetrieveSpec;
import org.springframework.util.MimeType;
import org.springframework.util.MimeTypeUtils;
@@ -90,7 +91,7 @@ public class DefaultRSocketRequesterTests {
testSendMono(spec -> spec.data(Mono.delay(MILLIS_10).then(), Void.class), "");
}
private void testSendMono(Function<RequestSpec, RequestSpec> mapper, String expectedValue) {
private void testSendMono(Function<RequestSpec, RetrieveSpec> mapper, String expectedValue) {
mapper.apply(this.requester.route("toA")).send().block(Duration.ofSeconds(5));
assertThat(this.rsocket.getSavedMethodName()).isEqualTo("fireAndForget");
@@ -114,7 +115,7 @@ public class DefaultRSocketRequesterTests {
testSendFlux(spec -> spec.data(stringFlux.cast(Object.class), Object.class), values);
}
private void testSendFlux(Function<RequestSpec, RequestSpec> mapper, String... expectedValues) {
private void testSendFlux(Function<RequestSpec, RetrieveSpec> mapper, String... expectedValues) {
this.rsocket.reset();
mapper.apply(this.requester.route("toA")).retrieveFlux(String.class).blockLast(Duration.ofSeconds(5));