コード例 #1
0
  @Test
  public void testEmitInSequence() {
    // test the happy path, emit batches in sequence
    Map meta = emitter.emitPartitionBatchNew(null, collectorMock, partition, null);
    String collected = collectorMock.getBuffer();
    assertTrue(collected.startsWith("message" + 0));
    // System.out.println("collected: " + collected);
    collectorMock.clear();

    emitter.emitPartitionBatchNew(null, collectorMock, partition, meta);
    collected = collectorMock.getBuffer();
    // System.out.println("collected: " + collected);
    assertTrue(collected.startsWith("message" + batchSize));
  }
コード例 #2
0
  @Test
  public void testReEmit() {
    // test we can re-emit the second batch
    Map meta = emitter.emitPartitionBatchNew(null, collectorMock, partition, null);
    collectorMock.clear();

    // emit second batch
    Map meta1 = emitter.emitPartitionBatchNew(null, collectorMock, partition, meta);
    String collected0 = collectorMock.getBuffer();
    collectorMock.clear();

    // re-emit second batch
    emitter.emitPartitionBatch(null, collectorMock, partition, meta1);
    String collected1 = collectorMock.getBuffer();
    assertTrue(collected0.equals(collected1));
  }