예제 #1
0
 @Parameterized.Parameters(name = "{0} bits")
 public static List<Object[]> formats() {
   ArrayList<Object[]> parameters = new ArrayList<>();
   for (BitmapDocumentFormat format : BitmapDocumentFormat.values()) {
     parameters.add(new Object[] {format});
   }
   return parameters;
 }
예제 #2
0
  @Test
  public void shouldReadPagesOfDocumentsFromSearcher() throws Exception {
    final int labelId = 7;
    final int pageSize = 2;
    // given
    Query query = mock(Query.class);
    IndexSearcher searcher = mock(IndexSearcher.class);
    ScoreDoc doc1 = new ScoreDoc(37, 0.0f);
    ScoreDoc doc2 = new ScoreDoc(16, 0.0f);
    ScoreDoc doc3 = new ScoreDoc(11, 0.0f);
    when(searcher.searchAfter(any(ScoreDoc.class), same(query), anyInt()))
        .thenReturn(
            docs(doc1, doc2), // page1
            docs(doc3) // page2
            );
    when(searcher.doc(37))
        .thenReturn(document(format.rangeField(0x1), format.labelField(labelId, 0x01)));
    when(searcher.doc(16))
        .thenReturn(document(format.rangeField(0x2), format.labelField(labelId, 0x03)));
    when(searcher.doc(11))
        .thenReturn(document(format.rangeField(0x3), format.labelField(labelId, 0x30)));

    PrimitiveLongIterator iterator =
        flatten(new PageOfRangesIterator(format, searcher, pageSize, query, labelId));

    // when
    List<Long> longs = primitivesList(iterator);

    // then
    assertEquals(
        asList(
            /*doc1:*/ (1L << format.bitmapFormat().shift),
            /*doc2:*/ (2L << format.bitmapFormat().shift),
            (2L << format.bitmapFormat().shift) + 1,
            /*doc3:*/ (3L << format.bitmapFormat().shift) + 4,
            (3L << format.bitmapFormat().shift) + 5),
        longs);

    ArgumentCaptor<ScoreDoc> prefixCollector = ArgumentCaptor.forClass(ScoreDoc.class);
    verify(searcher, times(2)).searchAfter(prefixCollector.capture(), same(query), eq(2));
    assertEquals(asList(null, doc2), prefixCollector.getAllValues());

    verify(searcher, times(3)).doc(anyInt());
    verifyNoMoreInteractions(searcher);
  }