Adding a test for multiple attribute paths

Project: http://git-wip-us.apache.org/repos/asf/incubator-nifi/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-nifi/commit/78ad0a31
Tree: http://git-wip-us.apache.org/repos/asf/incubator-nifi/tree/78ad0a31
Diff: http://git-wip-us.apache.org/repos/asf/incubator-nifi/diff/78ad0a31

Branch: refs/heads/NIFI-360
Commit: 78ad0a3147cf1f3573f94f79685ebb868b5d190d
Parents: 974617d
Author: Aldrin Piri <aldrinp...@gmail.com>
Authored: Mon Feb 16 18:20:31 2015 -0500
Committer: Aldrin Piri <aldrinp...@gmail.com>
Committed: Mon Feb 16 18:20:31 2015 -0500

----------------------------------------------------------------------
 .../standard/TestEvaluateJsonPath.java          | 22 ++++++++++++++++++++
 1 file changed, 22 insertions(+)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-nifi/blob/78ad0a31/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestEvaluateJsonPath.java
----------------------------------------------------------------------
diff --git 
a/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestEvaluateJsonPath.java
 
b/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestEvaluateJsonPath.java
index 308440f..60e19d9 100644
--- 
a/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestEvaluateJsonPath.java
+++ 
b/nifi/nifi-nar-bundles/nifi-standard-bundle/nifi-standard-processors/src/test/java/org/apache/nifi/processors/standard/TestEvaluateJsonPath.java
@@ -96,6 +96,28 @@ public class TestEvaluateJsonPath {
     }
 
     @Test
+    public void testExtractPath_destinationAttributes_twoPaths() throws 
Exception {
+        final TestRunner testRunner = TestRunners.newTestRunner(new 
EvaluateJsonPath());
+        testRunner.setProperty(EvaluateJsonPath.DESTINATION, 
EvaluateJsonPath.DESTINATION_ATTRIBUTE);
+
+        String jsonPathIdAttrKey = "evaluatejson.id";
+        String jsonPathNameAttrKey = "evaluatejson.name";
+
+        testRunner.setProperty(jsonPathIdAttrKey, "$[0]._id");
+        testRunner.setProperty(jsonPathNameAttrKey, "$[0].name");
+
+        testRunner.enqueue(JSON_SNIPPET);
+        testRunner.run();
+
+        Relationship expectedRel = EvaluateJsonPath.REL_MATCH;
+
+        testRunner.assertAllFlowFilesTransferred(expectedRel, 1);
+        final MockFlowFile out = 
testRunner.getFlowFilesForRelationship(expectedRel).get(0);
+        Assert.assertEquals("Transferred flow file did not have the correct 
result for id attribute", "54df94072d5dbf7dc6340cc5", 
out.getAttribute(jsonPathIdAttrKey));
+        Assert.assertEquals("Transferred flow file did not have the correct 
result for name attribute", "{\"first\":\"Shaffer\",\"last\":\"Pearson\"}", 
out.getAttribute(jsonPathNameAttrKey));
+    }
+
+    @Test
     public void testExtractPath_destinationContent() throws Exception {
         String jsonPathAttrKey = "JsonPath";
 

Reply via email to