From ac7f7baf8bc5f13ac8adc2622e8f4e3b3322eb4d Mon Sep 17 00:00:00 2001
From: Violeta Georgieva <violetag@vmware.com>
Date: Thu, 5 Oct 2023 13:12:56 +0300
Subject: [PATCH] Remove duplications (#2925)

---
 .../reactor/netty/ContextPropagationTest.java | 49 ++++++++++---------
 1 file changed, 27 insertions(+), 22 deletions(-)

diff --git a/reactor-netty-http/src/contextPropagationTest/java/reactor/netty/ContextPropagationTest.java b/reactor-netty-http/src/contextPropagationTest/java/reactor/netty/ContextPropagationTest.java
index 76f80db934..6beb893841 100644
--- a/reactor-netty-http/src/contextPropagationTest/java/reactor/netty/ContextPropagationTest.java
+++ b/reactor-netty-http/src/contextPropagationTest/java/reactor/netty/ContextPropagationTest.java
@@ -30,6 +30,8 @@
 import io.netty.handler.ssl.util.InsecureTrustManagerFactory;
 import io.netty.handler.ssl.util.SelfSignedCertificate;
 import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.MethodSource;
 import reactor.core.publisher.Hooks;
@@ -54,8 +56,17 @@ class ContextPropagationTest {
 	static final ConnectionProvider provider = ConnectionProvider.create("testContextPropagation", 1);
 	static final ContextRegistry registry = ContextRegistry.getInstance();
 
+	static SelfSignedCertificate ssc;
+
+	HttpServer baseServer;
 	DisposableServer disposableServer;
-	SelfSignedCertificate ssc;
+	Http2SslContextSpec serverCtx;
+
+
+	@BeforeAll
+	static void createSelfSignedCertificate() throws Exception {
+		ssc = new SelfSignedCertificate();
+	}
 
 	@AfterAll
 	static void disposePool() {
@@ -63,20 +74,22 @@ static void disposePool() {
 		        .block(Duration.ofSeconds(30));
 	}
 
-	@ParameterizedTest
-	@MethodSource("httpClientCombinations")
-	void testContextPropagation(HttpClient client) throws Exception {
-		ssc = new SelfSignedCertificate();
-		Http2SslContextSpec serverCtxHttp = Http2SslContextSpec.forServer(ssc.certificate(), ssc.privateKey());
-		HttpServer server =
+	@BeforeEach
+	void setUp() {
+		serverCtx = Http2SslContextSpec.forServer(ssc.certificate(), ssc.privateKey());
+		baseServer =
 				HttpServer.create()
 				          .wiretap(true)
 				          .httpRequestDecoder(spec -> spec.h2cMaxContentLength(256))
 				          .handle((in, out) -> out.send(in.receive().retain()));
+	}
 
-		server = client.configuration().sslProvider() != null ?
-				server.secure(spec -> spec.sslContext(serverCtxHttp)).protocol(HttpProtocol.HTTP11, HttpProtocol.H2) :
-				server.protocol(HttpProtocol.HTTP11, HttpProtocol.H2C);
+	@ParameterizedTest
+	@MethodSource("httpClientCombinations")
+	void testContextPropagation(HttpClient client) {
+		HttpServer server = client.configuration().sslProvider() != null ?
+				baseServer.secure(spec -> spec.sslContext(serverCtx)).protocol(HttpProtocol.HTTP11, HttpProtocol.H2) :
+				baseServer.protocol(HttpProtocol.HTTP11, HttpProtocol.H2C);
 
 		disposableServer = server.bindNow();
 
@@ -108,7 +121,7 @@ void testContextPropagation(HttpClient client) throws Exception {
 
 	@ParameterizedTest
 	@MethodSource("httpClientCombinations")
-	void testAutomaticContextPropagation(HttpClient client) throws Exception {
+	void testAutomaticContextPropagation(HttpClient client) {
 		String reactorCoreVersionMinor = System.getProperty("reactorCoreVersionMinor");
 		String contextPropagationVersionMicro =  System.getProperty("contextPropagationVersionMicro");
 
@@ -116,17 +129,9 @@ void testAutomaticContextPropagation(HttpClient client) throws Exception {
 				reactorCoreVersionMinor != null && !reactorCoreVersionMinor.isEmpty() && Integer.parseInt(reactorCoreVersionMinor) >= 6 && // 3.6.x
 						contextPropagationVersionMicro != null && !contextPropagationVersionMicro.isEmpty() && Integer.parseInt(contextPropagationVersionMicro) >= 5; // 1.0.5 or above
 
-		ssc = new SelfSignedCertificate();
-		Http2SslContextSpec serverCtxHttp = Http2SslContextSpec.forServer(ssc.certificate(), ssc.privateKey());
-		HttpServer server =
-				HttpServer.create()
-				          .wiretap(true)
-				          .httpRequestDecoder(spec -> spec.h2cMaxContentLength(256))
-				          .handle((in, out) -> out.send(in.receive().retain()));
-
-		server = client.configuration().sslProvider() != null ?
-				server.secure(spec -> spec.sslContext(serverCtxHttp)).protocol(HttpProtocol.HTTP11, HttpProtocol.H2) :
-				server.protocol(HttpProtocol.HTTP11, HttpProtocol.H2C);
+		HttpServer server = client.configuration().sslProvider() != null ?
+				baseServer.secure(spec -> spec.sslContext(serverCtx)).protocol(HttpProtocol.HTTP11, HttpProtocol.H2) :
+				baseServer.protocol(HttpProtocol.HTTP11, HttpProtocol.H2C);
 
 		disposableServer = server.bindNow();