IGNITE-28123 Under node restarts, partition assignment lookup may complete exceptionally while resolving primary replicas.

In ClientPrimaryReplicaTracker.primaryReplicasAsyncInternal, the exceptional path was not terminated for non-TableNotFound errors, and execution could continue into result construction with partitions == null, causing a secondary failure (NullPointerException from unboxing).

Author: Lalant

modules/client-handler/src/main/java/org/apache/ignite/client/handler/ClientPrimaryReplicaTracker.java

@@ -197,7 +197,9 @@ private CompletableFuture<PrimaryReplicasResult> primaryReplicasAsyncInternal(in
                     throw new CompletionException(cause);
                 }
 
-                assert false : "Unexpected error: " + err;
+                throw err instanceof CompletionException
+                        ? (CompletionException) err
+                        : new CompletionException(err);
             }
 
             PrimaryReplicasResult res = primaryReplicasNoWait(tableId, maxStartTime0, timestamp, true);

modules/client-handler/src/test/java/org/apache/ignite/client/handler/ClientPrimaryReplicaTrackerTest.java

@@ -20,11 +20,13 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNull;
 import static org.junit.jupiter.api.Assertions.assertTrue;
+import static org.junit.jupiter.api.Assertions.fail;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 
 import java.util.List;
 import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.CompletionException;
 import java.util.concurrent.atomic.AtomicLong;
 import org.apache.ignite.client.handler.ClientPrimaryReplicaTracker.PrimaryReplicasResult;
 import org.apache.ignite.internal.TestHybridClock;
@@ -134,6 +136,25 @@ public void testFailedInitialFutureIsRetried() {
         assertEquals("s2", replicas.nodeNames().get(1));
     }
 
+    @Test
+    public void testFailedPrimaryReplicaRequestPropagatesOriginalError() {
+        driver.returnError(true);
+        tracker.start();
+
+        try {
+            tracker.primaryReplicasAsync(TABLE_ID, null).join();
+            fail("Expected completion to fail");
+        } catch (CompletionException e) {
+            Throwable cause = e.getCause();
+            while (cause instanceof CompletionException && cause.getCause() != null) {
+                cause = cause.getCause();
+            }
+
+            assertTrue(cause instanceof RuntimeException);
+            assertEquals("FakePlacementDriver expected error", cause.getMessage());
+        }
+    }
+
     @Test
     public void testOldEventsAreIgnoredByLeaseStartTime() {
         tracker.start();