Polishing

samples/webflux-security/src/test/java/io/spring/sample/graphql/WebFluxSecuritySampleTests.java

@@ -16,7 +16,9 @@
 package io.spring.sample.graphql;
 
 import java.net.URI;
+import java.time.Duration;
 
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 
@@ -38,8 +40,7 @@ class WebFluxSecuritySampleTests {
 	@LocalServerPort
 	private int port;
 
-	@Autowired
-	private WebGraphQlTester graphQlTester;
+	private WebSocketGraphQlTester graphQlTester;
 
 	@BeforeEach
 	public void setUp() {
@@ -47,6 +48,11 @@ class WebFluxSecuritySampleTests {
 		this.graphQlTester = WebSocketGraphQlTester.create(url, new ReactorNettyWebSocketClient());
 	}
 
+	@AfterEach
+	void tearDown() {
+		this.graphQlTester.stop().block(Duration.ofSeconds(5));
+	}
+
 	@Test
 	void printError() {
 		this.graphQlTester.documentName("employeesNamesAndSalaries")