one more part of spring-projects/spring-graphql#904

nkonev · May 25, 2024 · 8e826b9 · 8e826b9
1 parent c726a25
commit 8e826b9
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/...test/java/name/nkonev/multipart/spring/graphql/server/webmvc/GraphQlHttpHandlerTests.java b/...test/java/name/nkonev/multipart/spring/graphql/server/webmvc/GraphQlHttpHandlerTests.java
@@ -144,7 +144,9 @@ private MockHttpServletResponse handleMultipartRequest(
 
         ServerRequest request = ServerRequest.create(servletRequest, MESSAGE_READERS);
         ServerResponse response = handler.handleMultipartRequest(request);
-
+        if (response instanceof AsyncServerResponse asyncResponse) {
+            asyncResponse.block();
+        }
         MockHttpServletResponse servletResponse = new MockHttpServletResponse();
         response.writeTo(servletRequest, servletResponse, new DefaultContext());
         return servletResponse;