NIFI-1272: Added unit test to verify behavior
Project: http://git-wip-us.apache.org/repos/asf/nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/nifi/commit/9ca0f95d Tree: http://git-wip-us.apache.org/repos/asf/nifi/tree/9ca0f95d Diff: http://git-wip-us.apache.org/repos/asf/nifi/diff/9ca0f95d Branch: refs/heads/master Commit: 9ca0f95d0685fc1a76a50eae978638916f1dbbcd Parents: fa4c531 Author: Mark Payne <[email protected]> Authored: Fri Dec 18 11:51:42 2015 -0500 Committer: Mark Payne <[email protected]> Committed: Fri Dec 18 11:51:42 2015 -0500 ---------------------------------------------------------------------- .../repository/TestStandardProcessSession.java | 14 ++++++++++++++ 1 file changed, 14 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/nifi/blob/9ca0f95d/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/test/java/org/apache/nifi/controller/repository/TestStandardProcessSession.java ---------------------------------------------------------------------- diff --git a/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/test/java/org/apache/nifi/controller/repository/TestStandardProcessSession.java b/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/test/java/org/apache/nifi/controller/repository/TestStandardProcessSession.java index d549a00..644018f 100644 --- a/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/test/java/org/apache/nifi/controller/repository/TestStandardProcessSession.java +++ b/nifi-nar-bundles/nifi-framework-bundle/nifi-framework/nifi-framework-core/src/test/java/org/apache/nifi/controller/repository/TestStandardProcessSession.java @@ -1106,6 +1106,20 @@ public class TestStandardProcessSession { } @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() .id(1L)
