NIFI-1272: Added unit test to verify behavior

This commit is contained in:
Mark Payne 2015-12-18 11:51:42 -05:00
parent fa4c5314ca
commit 9ca0f95d06
1 changed files with 14 additions and 0 deletions

View File

@ -1105,6 +1105,20 @@ public class TestStandardProcessSession {
assertEquals(ProvenanceEventType.CONTENT_MODIFIED, event.getEventType());
}
@Test
public void testGetWithCount() {
for (int i = 0; i < 8; i++) {
final FlowFileRecord flowFile = new StandardFlowFileRecord.Builder()
.id(i)
.addAttribute("uuid", "000000000000-0000-0000-0000-0000000" + i)
.build();
this.flowFileQueue.put(flowFile);
}
final List<FlowFile> flowFiles = session.get(7);
assertEquals(7, flowFiles.size());
}
@Test
public void testAttributesModifiedEmitted() throws IOException {
final FlowFileRecord flowFile = new StandardFlowFileRecord.Builder()