@Test
public void testEventForPartitionedDataset() {
final String topic = "testTopic";
final String kafka = "kafka";
final String hdfs = "hdfs";
final String path = "/data/tracking/PageViewEvent";
final String partitionName = "hourly/2018/08/15/15";
State state = new State();
LineageInfo lineageInfo = getLineageInfo();
DatasetDescriptor source = new DatasetDescriptor(kafka, topic);
lineageInfo.setSource(source, state);
DatasetDescriptor destinationDataset = new DatasetDescriptor(hdfs, path);
PartitionDescriptor destination = new PartitionDescriptor(partitionName, destinationDataset);
lineageInfo.putDestination(destination, 0, state);
Map<String, Set<LineageEventBuilder>> events = LineageInfo.load(state);
LineageEventBuilder event = first(events.get("0"));
verify(event, topic, source, destination);
GobblinTrackingEvent trackingEvent = event.build();
Assert.assertEquals(LineageEventBuilder.isLineageEvent(trackingEvent), true);
Assert.assertEquals(LineageEventBuilder.fromEvent(trackingEvent), event);
}