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

Polish AbstractWebSocketIntegrationTests

Browse Source
This commit is contained in:
Rossen Stoyanchev
2019-08-19 16:56:41 +03:00
parent 4386bf05de
commit 5cfe491602
1 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
spring-webflux/src/test/java/org/springframework/web/reactive/socket/AbstractWebSocketIntegrationTests.java
View File

@@ -34,6 +34,7 @@ import org.junit.jupiter.params.provider.MethodSource;
import org.xnio.OptionMap;
import org.xnio.Xnio;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import reactor.util.function.Tuple3;
import org.springframework.context.ApplicationContext;
@@ -98,11 +99,19 @@ public abstract class AbstractWebSocketIntegrationTests {
servers.put(new ReactorHttpServer(), ReactorNettyConfig.class);
servers.put(new UndertowHttpServer(), UndertowConfig.class);
Flux<WebSocketClient> f1 = Flux.fromArray(clients).concatMap(c -> Flux.just(c).repeat(servers.size() - 1));
Flux<HttpServer> f2 = Flux.fromIterable(servers.keySet()).repeat(clients.length);
Flux<Class<?>> f3 = Flux.fromIterable(servers.values()).repeat(clients.length);
// Try each client once against each server..
return Flux.zip(f1, f2, f3).map(Tuple3::toArray).collectList().block()
Flux<WebSocketClient> f1 = Flux.fromArray(clients)
.concatMap(c -> Mono.just(c).repeat(servers.size() - 1));
Flux<Map.Entry<HttpServer, Class<?>>> f2 = Flux.fromIterable(servers.entrySet())
.repeat(clients.length - 1)
.share();
return Flux.zip(f1, f2.map(Map.Entry::getKey), f2.map(Map.Entry::getValue))
.map(Tuple3::toArray)
.collectList()
.block()
.toArray(new Object[clients.length * servers.size()][2]);
}