@SuppressWarnings("unchecked")
  private Reader<Text, Text> getReader(int i) throws IOException {
    Reader<Text, Text> readerMock = mock(Reader.class);
    when(readerMock.getPosition()).thenReturn(0l).thenReturn(10l).thenReturn(20l);
    when(readerMock.nextRawKey(any(DataInputBuffer.class))).thenAnswer(getKeyAnswer("Segment" + i));
    doAnswer(getValueAnswer("Segment" + i))
        .when(readerMock)
        .nextRawValue(any(DataInputBuffer.class));

    return readerMock;
  }
Exemple #2
0
 public long getPosition() throws IOException {
   return reader.getPosition();
 }