Examples with TimestampType org.apache.kafka.common.record.TimestampType used on opensource projects

Example 11 with TimestampType

use of org.apache.kafka.common.record.TimestampType in project kafka by apache.

the class WorkerSinkTaskTest method testTimestampPropagation.

@Test
public void testTimestampPropagation() throws Exception {
    final Long timestamp = System.currentTimeMillis();
    final TimestampType timestampType = TimestampType.CREATE_TIME;
    createTask(initialState);
    expectInitializeTask();
    expectTaskGetTopic(true);
    expectPollInitialAssignment();
    expectConsumerPoll(1, timestamp, timestampType);
    expectConversionAndTransformation(1);
    Capture<Collection<SinkRecord>> records = EasyMock.newCapture(CaptureType.ALL);
    sinkTask.put(EasyMock.capture(records));
    PowerMock.replayAll();
    workerTask.initialize(TASK_CONFIG);
    workerTask.initializeAndStart();
    // iter 1 -- initial assignment
    workerTask.iteration();
    // iter 2 -- deliver 1 record
    workerTask.iteration();
    SinkRecord record = records.getValue().iterator().next();
    assertEquals(timestamp, record.timestamp());
    assertEquals(timestampType, record.timestampType());
    PowerMock.verifyAll();
}