From e90015245c328755d26d185d6c13f15c1c7f30f7 Mon Sep 17 00:00:00 2001 From: Josh Rosen Date: Thu, 30 Apr 2015 17:02:18 -0700 Subject: [PATCH] Add test for empty iterator in UnsafeSorter --- .../apache/spark/unsafe/sort/UnsafeSorterSuite.java | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/core/src/test/java/org/apache/spark/unsafe/sort/UnsafeSorterSuite.java b/core/src/test/java/org/apache/spark/unsafe/sort/UnsafeSorterSuite.java index 95d54ea79cf40..52bb27a5b572f 100644 --- a/core/src/test/java/org/apache/spark/unsafe/sort/UnsafeSorterSuite.java +++ b/core/src/test/java/org/apache/spark/unsafe/sort/UnsafeSorterSuite.java @@ -22,6 +22,7 @@ import org.junit.Assert; import org.junit.Test; +import static org.mockito.Mockito.*; import org.apache.spark.HashPartitioner; import org.apache.spark.unsafe.PlatformDependent; @@ -43,6 +44,18 @@ private static String getStringFromDataPage(Object baseObject, long baseOffset) return new String(strBytes); } + @Test + public void testSortingEmptyInput() { + final UnsafeSorter sorter = new UnsafeSorter( + new TaskMemoryManager(new ExecutorMemoryManager(MemoryAllocator.HEAP)), + mock(UnsafeSorter.RecordComparator.class), + mock(UnsafeSorter.PrefixComputer.class), + mock(UnsafeSorter.PrefixComparator.class), + 100); + final Iterator iter = sorter.getSortedIterator(); + assert(!iter.hasNext()); + } + /** * Tests the type of sorting that's used in the non-combiner path of sort-based shuffle. */