Per https://github.com/bazelbuild/bazel/pull/15416#discussion_r871840328, use latch instead of flag to check whether the chunker is closed.

PiperOrigin-RevId: 448473842

src/test/java/com/google/devtools/build/lib/remote/GrpcCacheClientTest.java

diff --git a/src/test/java/com/google/devtools/build/lib/remote/GrpcCacheClientTest.java b/src/test/java/com/google/devtools/build/lib/remote/GrpcCacheClientTest.java
index 3839ed4..5bb3122 100644
--- a/src/test/java/com/google/devtools/build/lib/remote/GrpcCacheClientTest.java
+++ b/src/test/java/com/google/devtools/build/lib/remote/GrpcCacheClientTest.java
@@ -75,6 +75,7 @@
 import java.io.IOException;
 import java.util.List;
 import java.util.Optional;
+import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.ExecutionException;
 import java.util.concurrent.atomic.AtomicBoolean;
 import java.util.concurrent.atomic.AtomicInteger;
@@ -209,7 +210,7 @@
         });
     byte[] data = new byte[20];
     Digest digest = DIGEST_UTIL.compute(data);
-    AtomicBoolean closed = new AtomicBoolean();
+    CountDownLatch latch = new CountDownLatch(1);
     Chunker chunker =
         new Chunker(
             () ->
@@ -218,7 +219,7 @@
                   @Override
                   public void close() throws IOException {
                     super.close();
-                    closed.set(true);
+                    latch.countDown();
                   }
                 },
             data.length,
@@ -231,7 +232,7 @@
 
     // assert
     assertThat(Status.fromThrowable(t.getCause()).getCode()).isEqualTo(Status.Code.DATA_LOSS);
-    assertThat(closed.get()).isTrue();
+    latch.await();
   }
 
   @Test