diff options
Diffstat (limited to 'src/test/java/com/google/devtools/build/lib/remote/ByteStreamUploaderTest.java')
-rw-r--r-- | src/test/java/com/google/devtools/build/lib/remote/ByteStreamUploaderTest.java | 9 |
1 files changed, 9 insertions, 0 deletions
diff --git a/src/test/java/com/google/devtools/build/lib/remote/ByteStreamUploaderTest.java b/src/test/java/com/google/devtools/build/lib/remote/ByteStreamUploaderTest.java index 8cd78ff10c..732afac1fa 100644 --- a/src/test/java/com/google/devtools/build/lib/remote/ByteStreamUploaderTest.java +++ b/src/test/java/com/google/devtools/build/lib/remote/ByteStreamUploaderTest.java @@ -263,11 +263,18 @@ public class ByteStreamUploaderTest { Chunker chunker = new Chunker(blob, CHUNK_SIZE); AtomicInteger numWriteCalls = new AtomicInteger(); + CountDownLatch blocker = new CountDownLatch(1); serviceRegistry.addService(new ByteStreamImplBase() { @Override public StreamObserver<WriteRequest> write(StreamObserver<WriteResponse> response) { numWriteCalls.incrementAndGet(); + try { + // Ensures that the first upload does not finish, before the second upload is started. + blocker.await(); + } catch (InterruptedException e) { + Thread.currentThread().interrupt(); + } return new StreamObserver<WriteRequest>() { @@ -295,6 +302,8 @@ public class ByteStreamUploaderTest { Future<?> upload1 = uploader.uploadBlobAsync(chunker); Future<?> upload2 = uploader.uploadBlobAsync(chunker); + blocker.countDown(); + assertThat(upload1).isSameAs(upload2); upload1.get(); |