2929
3030import static org .apache .lucene .tests .util .LuceneTestCase .newDirectory ;
3131import static org .apache .lucene .tests .util .LuceneTestCase .random ;
32+ import static org .hamcrest .MatcherAssert .assertThat ;
33+ import static org .hamcrest .Matchers .equalTo ;
3234
3335public class BlockLoaderTestRunner {
3436 private final BlockLoaderTestCase .Params params ;
@@ -74,10 +76,9 @@ private Object load(BlockLoader blockLoader, LeafReaderContext context, MapperSe
7476 // `columnAtATimeReader` is tried first, we mimic `ValuesSourceReaderOperator`
7577 var columnAtATimeReader = blockLoader .columnAtATimeReader (context );
7678 if (columnAtATimeReader != null ) {
77- var block = (TestBlock ) columnAtATimeReader .read (TestBlock .factory (context .reader ().numDocs ()), TestBlock .docs (0 ));
78- if (block .size () == 0 ) {
79- return null ;
80- }
79+ BlockLoader .Docs docs = TestBlock .docs (0 );
80+ var block = (TestBlock ) columnAtATimeReader .read (TestBlock .factory (context .reader ().numDocs ()), docs );
81+ assertThat (block .size (), equalTo (1 ));
8182 return block .get (0 );
8283 }
8384
@@ -99,9 +100,8 @@ private Object load(BlockLoader blockLoader, LeafReaderContext context, MapperSe
99100 BlockLoader .Builder builder = blockLoader .builder (TestBlock .factory (context .reader ().numDocs ()), 1 );
100101 blockLoader .rowStrideReader (context ).read (0 , storedFieldsLoader , builder );
101102 var block = (TestBlock ) builder .build ();
102- if (block .size () == 0 ) {
103- return null ;
104- }
103+ assertThat (block .size (), equalTo (1 ));
104+
105105 return block .get (0 );
106106 }
107107
0 commit comments