Enforce timeout in WebFlux multipart tests

rstoyanchev · rstoyanchev · commit 01426481ea02 · 2021-09-17T14:53:45.000+01:00
diff --git a/spring-webflux/src/test/java/org/springframework/web/reactive/function/MultipartIntegrationTests.java b/spring-webflux/src/test/java/org/springframework/web/reactive/function/MultipartIntegrationTests.java
@@ -1,5 +1,5 @@
 /*
- * Copyright 2002-2019 the original author or authors.
+ * Copyright 2002-2021 the original author or authors.
  *
  * Licensed under the Apache License, Version 2.0 (the "License");
  * you may not use this file except in compliance with the License.
@@ -20,6 +20,7 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.time.Duration;
 import java.util.Map;
 
 import reactor.core.publisher.Mono;
@@ -72,7 +73,8 @@ void multipartData(HttpServer httpServer) throws Exception {
 		StepVerifier
 				.create(result)
 				.consumeNextWith(entity -> assertThat(entity.getStatusCode()).isEqualTo(HttpStatus.OK))
-				.verifyComplete();
+				.expectComplete()
+				.verify(Duration.ofSeconds(5));
 	}
 
 	@ParameterizedHttpServerTest
@@ -89,7 +91,8 @@ void parts(HttpServer httpServer) throws Exception {
 		StepVerifier
 				.create(result)
 				.consumeNextWith(entity -> assertThat(entity.getStatusCode()).isEqualTo(HttpStatus.OK))
-				.verifyComplete();
+				.expectComplete()
+				.verify(Duration.ofSeconds(5));
 	}
 
 	@ParameterizedHttpServerTest
@@ -119,7 +122,8 @@ void transferTo(HttpServer httpServer) throws Exception {
 						fail("IOException", ex);
 					}
 				})
-				.verifyComplete();
+				.expectComplete()
+				.verify(Duration.ofSeconds(5));
 	}
 
 	private MultiValueMap<String, HttpEntity<?>> generateBody() {
@@ -193,8 +197,7 @@ private Mono<Path> createTempFile() {
 				catch (IOException ex) {
 					return Mono.error(ex);
 				}
-			})
-					.subscribeOn(Schedulers.boundedElastic());
+			}).subscribeOn(Schedulers.boundedElastic());
 		}
 
 	}
