Repository: incubator-nifi Updated Branches: refs/heads/develop 8d745c2b7 -> ba7855646
NIFI-594: Removing the output to System.out and providing some simple checks to ensure extraction of attributes is consistent with the sample file used in the test. Project: http://git-wip-us.apache.org/repos/asf/incubator-nifi/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-nifi/commit/ba785564 Tree: http://git-wip-us.apache.org/repos/asf/incubator-nifi/tree/ba785564 Diff: http://git-wip-us.apache.org/repos/asf/incubator-nifi/diff/ba785564 Branch: refs/heads/develop Commit: ba785564687d7118551c84a4ca46ddfb2b615c96 Parents: 8d745c2 Author: Aldrin Piri <[email protected]> Authored: Fri Jun 5 10:36:19 2015 -0400 Committer: Aldrin Piri <[email protected]> Committed: Fri Jun 5 10:43:33 2015 -0400 ---------------------------------------------------------------------- .../hl7/TestExtractHL7Attributes.java | 47 ++++++++++++++++---- 1 file changed, 38 insertions(+), 9 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-nifi/blob/ba785564/nifi/nifi-nar-bundles/nifi-hl7-bundle/nifi-hl7-processors/src/test/java/org/apache/nifi/processors/hl7/TestExtractHL7Attributes.java ---------------------------------------------------------------------- diff --git a/nifi/nifi-nar-bundles/nifi-hl7-bundle/nifi-hl7-processors/src/test/java/org/apache/nifi/processors/hl7/TestExtractHL7Attributes.java b/nifi/nifi-nar-bundles/nifi-hl7-bundle/nifi-hl7-processors/src/test/java/org/apache/nifi/processors/hl7/TestExtractHL7Attributes.java index d173c27..d91a633 100644 --- a/nifi/nifi-nar-bundles/nifi-hl7-bundle/nifi-hl7-processors/src/test/java/org/apache/nifi/processors/hl7/TestExtractHL7Attributes.java +++ b/nifi/nifi-nar-bundles/nifi-hl7-bundle/nifi-hl7-processors/src/test/java/org/apache/nifi/processors/hl7/TestExtractHL7Attributes.java @@ -16,33 +16,62 @@ */ package org.apache.nifi.processors.hl7; +import org.apache.nifi.util.MockFlowFile; +import org.apache.nifi.util.TestRunner; +import org.apache.nifi.util.TestRunners; +import org.junit.Assert; +import org.junit.BeforeClass; +import org.junit.Test; + import java.io.IOException; import java.nio.file.Paths; import java.util.Map; import java.util.SortedMap; import java.util.TreeMap; -import org.apache.nifi.processors.hl7.ExtractHL7Attributes; -import org.apache.nifi.util.MockFlowFile; -import org.apache.nifi.util.TestRunner; -import org.apache.nifi.util.TestRunners; -import org.junit.Test; - public class TestExtractHL7Attributes { + @BeforeClass + public static void setup() { + System.setProperty("org.slf4j.simpleLogger.log.org.apache.nifi", "DEBUG"); + } + @Test public void testExtract() throws IOException { - System.setProperty("org.slf4j.simpleLogger.log.org.apache.nifi", "DEBUG"); final TestRunner runner = TestRunners.newTestRunner(ExtractHL7Attributes.class); runner.enqueue(Paths.get("src/test/resources/hypoglycemia.hl7")); runner.run(); runner.assertAllFlowFilesTransferred(ExtractHL7Attributes.REL_SUCCESS, 1); + final MockFlowFile out = runner.getFlowFilesForRelationship(ExtractHL7Attributes.REL_SUCCESS).get(0); final SortedMap<String, String> sortedAttrs = new TreeMap<>(out.getAttributes()); + + int mshSegmentCount = 0; + int obrSegmentCount = 0; + int obxSegmentCount = 0; + int pidSegmentCount = 0; + for (final Map.Entry<String, String> entry : sortedAttrs.entrySet()) { - System.out.println(entry.getKey() + " : " + entry.getValue()); + final String entryKey = entry.getKey(); + if (entryKey.startsWith("MSH")) { + mshSegmentCount++; + continue; + } else if (entryKey.startsWith("OBR")) { + obrSegmentCount++; + continue; + } else if (entryKey.startsWith("OBX")) { + obxSegmentCount++; + continue; + } else if (entryKey.startsWith("PID")) { + pidSegmentCount++; + continue; + } } - } + Assert.assertEquals("Did not have the proper number of MSH segments", 8, mshSegmentCount); + Assert.assertEquals("Did not have the proper number of OBR segments", 9, obrSegmentCount); + Assert.assertEquals("Did not have the proper number of OBX segments", 9, obxSegmentCount); + Assert.assertEquals("Did not have the proper number of PID segments", 6, pidSegmentCount); + } }
