IGNITE-12607 PartitionsExchangeAwareTest is flaky - Fixes #7339.
diff --git a/modules/core/src/test/java/org/apache/ignite/internal/processors/cache/distributed/PartitionsExchangeAwareTest.java b/modules/core/src/test/java/org/apache/ignite/internal/processors/cache/distributed/PartitionsExchangeAwareTest.java
index 2070dcd..48512f0 100644
--- a/modules/core/src/test/java/org/apache/ignite/internal/processors/cache/distributed/PartitionsExchangeAwareTest.java
+++ b/modules/core/src/test/java/org/apache/ignite/internal/processors/cache/distributed/PartitionsExchangeAwareTest.java
@@ -263,10 +263,14 @@
 
         assertEquals(updatePossible, GridTestUtils.waitForCondition(new GridAbsPredicate() {
             @Override public boolean apply() {
-                assertTrue(txUpdateFut.isDone() == atomicUpdateFut.isDone());
-
-                return atomicUpdateFut.isDone();
+                return atomicUpdateFut.isDone() && txUpdateFut.isDone();
             }
         }, TIMEOUT_SECONDS * 1000));
+
+        if (!updatePossible) {
+            assertFalse(atomicUpdateFut.isDone());
+
+            assertFalse(txUpdateFut.isDone());
+        }
     }
 }