diff --git a/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/SortMergeResultPartitionReadSchedulerTest.java b/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/SortMergeResultPartitionReadSchedulerTest.java index 0d1613e14017e..a3b459f0f8fdb 100644 --- a/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/SortMergeResultPartitionReadSchedulerTest.java +++ b/flink-runtime/src/test/java/org/apache/flink/runtime/io/network/partition/SortMergeResultPartitionReadSchedulerTest.java @@ -266,16 +266,16 @@ void testRequestBufferTimeout() throws Exception { SortMergeResultPartitionReadScheduler readScheduler = new SortMergeResultPartitionReadScheduler( bufferPool, executorService, this, bufferRequestTimeout); + long startTimestamp = System.nanoTime(); readScheduler.createSubpartitionReader( new NoOpBufferAvailablityListener(), 0, partitionedFile); // request and use all buffers of buffer pool. readScheduler.run(); assertThat(bufferPool.getAvailableBuffers()).isZero(); - long startTimestamp = System.nanoTime(); assertThatThrownBy(readScheduler::allocateBuffers).isInstanceOf(TimeoutException.class); long requestDuration = System.nanoTime() - startTimestamp; - assertThat(requestDuration > bufferRequestTimeout.toNanos()).isTrue(); + assertThat(requestDuration >= bufferRequestTimeout.toNanos()).isTrue(); readScheduler.release(); }