Test case for race condition in WriteBarrier

asw12 · asw12 · commit 114df9161fe9 · 2023-12-19T15:47:29.000-08:00
This is just a quick-and-dirty test case scrapped together from
ControllerInputIntegrationTests.
diff --git a/spring-web/src/testFixtures/java/org/springframework/web/testfixture/http/server/reactive/bootstrap/AbstractHttpHandlerIntegrationTests.java b/spring-web/src/testFixtures/java/org/springframework/web/testfixture/http/server/reactive/bootstrap/AbstractHttpHandlerIntegrationTests.java
@@ -125,9 +125,9 @@ public static Flux<Long> testInterval(Duration period, int count) {
 
 	static Stream<Named<HttpServer>> httpServers() {
 		return Stream.of(
-				named("Jetty", new JettyHttpServer()),
-				named("Reactor Netty", new ReactorHttpServer()),
-				named("Tomcat", new TomcatHttpServer()),
+				// named("Jetty", new JettyHttpServer()),
+				// named("Reactor Netty", new ReactorHttpServer()),
+				// named("Tomcat", new TomcatHttpServer()),
 				named("Undertow", new UndertowHttpServer())
 		);
 	}
diff --git a/spring-webflux/src/test/java/org/springframework/web/reactive/result/method/annotation/ChannelSendOperatorRaceConditionIntegrationTest.java b/spring-webflux/src/test/java/org/springframework/web/reactive/result/method/annotation/ChannelSendOperatorRaceConditionIntegrationTest.java
@@ -0,0 +1,45 @@
+package org.springframework.web.reactive.result.method.annotation;
+
+import org.reactivestreams.Publisher;
+import org.springframework.context.ApplicationContext;
+import org.springframework.context.annotation.AnnotationConfigApplicationContext;
+import org.springframework.http.HttpHeaders;
+import org.springframework.web.bind.annotation.GetMapping;
+import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.reactive.config.EnableWebFlux;
+import org.springframework.web.testfixture.http.server.reactive.bootstrap.HttpServer;
+import reactor.core.publisher.Flux;
+import reactor.core.scheduler.Schedulers;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+class ChannelSendOperatorRaceConditionIntegrationTest extends AbstractRequestMappingIntegrationTests {
+
+	@Override
+	protected ApplicationContext initApplicationContext() {
+		return new AnnotationConfigApplicationContext(WebConfig.class, TestRestController.class);
+	}
+
+
+	@ParameterizedHttpServerTest
+	void handlePublishOn(HttpServer httpServer) throws Exception {
+		startServer(httpServer);
+
+		String expected = "Hello world!";
+		assertThat(performGet("/test", new HttpHeaders(), String.class).getBody()).isEqualTo(expected);
+	}
+
+	@EnableWebFlux
+	static class WebConfig {
+	}
+	@RestController
+	@SuppressWarnings("unused")
+	private static class TestRestController {
+
+		@GetMapping("/test")
+		public Publisher<String> handlerWithPublishOn() {
+			return Flux.just("Hello ", "world", "!")
+					.publishOn(Schedulers.boundedElastic());
+		}
+	}
+}
