diff --git a/tez-runtime-library/src/test/java/org/apache/tez/runtime/library/common/shuffle/impl/TestSimpleFetchedInputAllocator.java b/tez-runtime-library/src/test/java/org/apache/tez/runtime/library/common/shuffle/impl/TestSimpleFetchedInputAllocator.java index a9a78a976c..0671682cc5 100644 --- a/tez-runtime-library/src/test/java/org/apache/tez/runtime/library/common/shuffle/impl/TestSimpleFetchedInputAllocator.java +++ b/tez-runtime-library/src/test/java/org/apache/tez/runtime/library/common/shuffle/impl/TestSimpleFetchedInputAllocator.java @@ -118,6 +118,9 @@ public void testInMemAllocationWithJvmMaxMemory() throws IOException { // check if requestSize is greater than maxSingleShuffleLimit assertTrue(requestSize > inputManager.maxSingleShuffleLimit); + // check if really test case failure catch in tez ci + assertTrue(requestSize < inputManager.maxSingleShuffleLimit); + // requestSize is greater than the maxSingleShuffleLimit, so allocation is from DISK FetchedInput fi1 = inputManager.allocate(requestSize, compressedSize, new InputAttemptIdentifier(1, 1)); assertEquals(FetchedInput.Type.DISK, fi1.getType());