Merge pull request #9024 from yosefe/topic/java-test-wait-for-active-…

…message-to-v1.14.x JAVA/TEST: Wait for active message to arrive before checking data - v1.14.x
openucx · Apr 19, 2023 · 816f0e0 · 816f0e0
2 parents 22604a3 + a777090
commit 816f0e0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindings/java/src/test/java/org/openucx/jucx/UcpEndpointTest.java b/bindings/java/src/test/java/org/openucx/jucx/UcpEndpointTest.java
@@ -735,7 +735,8 @@ public void onSuccess(UcpRequest request) {
             sendData.getMemory().getAddress(), 2L, UcpConstants.UCP_AM_FLAG_PERSISTENT_DATA, null,
             new UcpRequestParams().setMemoryType(memType).setMemoryHandle(sendData.getMemory()));
 
-        while (!Arrays.stream(requests).allMatch(r -> (r != null) && r.isCompleted())) {
+        while (!Arrays.stream(requests).allMatch(r -> (r != null) && r.isCompleted()) ||
+             (persistantAmData.get() == null)) {
             worker1.progress();
             worker2.progress();
         }