From 405f89cfd3c0c03554c5cf3d067e782bc9f69486 Mon Sep 17 00:00:00 2001
From: "sdk-sentinel-publisher[bot]"
 <314421413+sdk-sentinel-publisher[bot]@users.noreply.github.com>
Date: Sat, 15 Aug 2026 01:11:21 +0000
Subject: [PATCH] Fix CI flake detected by SDK Sentinel (java)

---
 .../temporal/internal/worker/AsyncPollerTest.java   | 13 ++++++++++++-
 1 file changed, 12 insertions(+), 1 deletion(-)

diff --git a/temporal-sdk/src/test/java/io/temporal/internal/worker/AsyncPollerTest.java b/temporal-sdk/src/test/java/io/temporal/internal/worker/AsyncPollerTest.java
index 5faa34ca7c..22a6c2b495 100644
--- a/temporal-sdk/src/test/java/io/temporal/internal/worker/AsyncPollerTest.java
+++ b/temporal-sdk/src/test/java/io/temporal/internal/worker/AsyncPollerTest.java
@@ -336,7 +336,15 @@ public void testAsyncPollFailed()
   @Test
   public void testSuspendPolling()
       throws InterruptedException, ExecutionException, AsyncPoller.PollTaskAsyncAbort {
-    CountingSlotSupplier<SlotInfo> slotSupplierInner = new CountingSlotSupplier<>(1);
+    CountDownLatch reserveAttemptLatch = new CountDownLatch(2);
+    CountingSlotSupplier<SlotInfo> slotSupplierInner =
+        new CountingSlotSupplier<SlotInfo>(1) {
+          @Override
+          public SlotSupplierFuture reserveSlot(SlotReserveContext<SlotInfo> ctx) throws Exception {
+            reserveAttemptLatch.countDown();
+            return super.reserveSlot(ctx);
+          }
+        };
     TrackingSlotSupplier<?> slotSupplier =
         new TrackingSlotSupplier<>(slotSupplierInner, new NoopScope());
     DummyTaskExecutor executor = new DummyTaskExecutor(slotSupplier);
@@ -377,6 +385,9 @@ public void testSuspendPolling()
           assertEquals(1, slotSupplierInner.reservedCount.get());
           assertEquals(0, slotSupplier.getUsedSlots().size());
         });
+    // Wait until the poll loop is reserving a slot for its next iteration. Suspending before this
+    // point would correctly prevent that reservation, making the expected count racy.
+    assertTrue(reserveAttemptLatch.await(5, TimeUnit.SECONDS));
     // Suspend polling again, this will not affect the already issued poll request
     poller.suspendPolling();
     completePoll.get().apply();