[patch] Extract JSON Objects as JSON Strings

001-bug59609-json-path.diff (text/plain), 11.84 KB, created by Felix Schumacher on 2016-05-24 19:15:24 UTC

(hide)

Description: Extract JSON Objects as JSON Strings

Filename:

MIME Type:

Creator: Felix Schumacher

Created: 2016-05-24 19:15:24 UTC

Size: 11.84 KB

patch

obsolete

>diff --git a/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONManager.java b/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONManager.java >index c297d35..88d4473 100644 >--- a/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONManager.java >+++ b/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONManager.java >@@ -19,10 +19,14 @@ > package org.apache.jmeter.extractor.json.jsonpath; > > import java.text.ParseException; >+import java.util.ArrayList; > import java.util.HashMap; > import java.util.List; > import java.util.Map; > >+import net.minidev.json.JSONArray; >+import net.minidev.json.JSONObject; >+ > import com.jayway.jsonpath.Configuration; > import com.jayway.jsonpath.JsonPath; > import com.jayway.jsonpath.Option; >@@ -60,13 +64,30 @@ public class JSONManager { > * > * @param jsonString JSON String from which data is extracted > * @param jsonPath JSON-PATH expression >- * @return List of String extracted data >+ * @return List of JSON Strings of the extracted data > * @throws ParseException > */ > public List<Object> extractWithJsonPath(String jsonString, String jsonPath) > throws ParseException { > JsonPath jsonPathParser = getJsonPath(jsonPath); >- return jsonPathParser.read(jsonString, >+ List<Object> extractedObjects = jsonPathParser.read(jsonString, > DEFAULT_CONFIGURATION); >+ List<Object> results = new ArrayList<>(extractedObjects.size()); >+ for (Object obj: extractedObjects) { >+ results.add(stringifyJSONObject(obj)); >+ } >+ return results; > } >+ >+ @SuppressWarnings("unchecked") >+ private String stringifyJSONObject(Object obj) { >+ if (obj instanceof Map) { >+ return new JSONObject((Map<String, ?>) obj).toJSONString(); >+ } >+ if (obj instanceof JSONArray) { >+ return ((JSONArray)obj).toJSONString(); >+ } >+ return obj == null ? "" : obj.toString(); //$NON-NLS-1$ >+ } >+ > } >diff --git a/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONPostProcessor.java b/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONPostProcessor.java >index 1f45b35..d8c1856 100644 >--- a/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONPostProcessor.java >+++ b/src/components/org/apache/jmeter/extractor/json/jsonpath/JSONPostProcessor.java >@@ -125,11 +125,12 @@ public class JSONPostProcessor extends AbstractScopedTestElement implements Seri > new StringBuilder(getComputeConcatenation() > ? extractedValues.size() * 20 > : 1); >- for (Object stringExtracted : extractedValues) { >- vars.put(currentRefName + "_" + index, >- stringExtracted != null ? stringExtracted.toString() : ""); //$NON-NLS-1$ >+ for (Object extractedObject : extractedValues) { >+ String extractedString = stringify(extractedObject); >+ vars.put(currentRefName + "_" + index, >+ extractedString); //$NON-NLS-1$ > if (getComputeConcatenation()) { >- concat.append(stringExtracted) >+ concat.append(extractedString) > .append(JSONPostProcessor.JSON_CONCATENATION_SEPARATOR); > } > index++; >@@ -141,9 +142,9 @@ public class JSONPostProcessor extends AbstractScopedTestElement implements Seri > } else if (matchNumber == 0) { > // Random extraction > int matchSize = extractedValues.size(); >- Object obj = extractedValues.get(JMeterUtils.getRandomInt(matchSize)); >- vars.put(currentRefName, >- obj != null ? obj.toString() : ""); //$NON-NLS-1$ >+ int matchNr = JMeterUtils.getRandomInt(matchSize); >+ placeObjectIntoVars(vars, currentRefName, >+ extractedValues, matchNr); > } else { > // extract at position > if (matchNumber > extractedValues.size()) { >@@ -155,20 +156,14 @@ public class JSONPostProcessor extends AbstractScopedTestElement implements Seri > } > vars.put(currentRefName, defaultValues[i]); > } else { >- Object obj = extractedValues.get(matchNumber - 1); >- vars.put(currentRefName, >- obj != null ? obj.toString() : ""); //$NON-NLS-1$ >+ placeObjectIntoVars(vars, currentRefName, extractedValues, matchNumber - 1); > } > } > } else { > // else just one value extracted >- Object obj = extractedValues.get(0); >- String objAsString = >- obj != null ? obj.toString() : ""; //$NON-NLS-1$ >- vars.put(currentRefName, >- objAsString); >+ placeObjectIntoVars(vars, currentRefName, extractedValues, 0); > if (matchNumber < 0 && getComputeConcatenation()) { >- vars.put(currentRefName + ALL_SUFFIX, objAsString); >+ vars.put(currentRefName + ALL_SUFFIX, vars.get(currentRefName)); > } > } > vars.put(currentRefName + REF_MATCH_NR, Integer.toString(extractedValues.size())); >@@ -187,6 +182,16 @@ public class JSONPostProcessor extends AbstractScopedTestElement implements Seri > } > } > >+ private void placeObjectIntoVars(JMeterVariables vars, String currentRefName, >+ List<Object> extractedValues, int matchNr) { >+ vars.put(currentRefName, >+ stringify(extractedValues.get(matchNr))); >+ } >+ >+ private String stringify(Object obj) { >+ return obj == null ? "" : obj.toString(); //$NON-NLS-1$ >+ } >+ > public String getJsonPathExpressions() { > return getPropertyAsString(JSON_PATH_EXPRESSIONS); > } >diff --git a/test/src/org/apache/jmeter/extractor/TestJSONPostProcessor.java b/test/src/org/apache/jmeter/extractor/TestJSONPostProcessor.java >new file mode 100644 >index 0000000..2e1fafd >--- /dev/null >+++ b/test/src/org/apache/jmeter/extractor/TestJSONPostProcessor.java >@@ -0,0 +1,123 @@ >+/* >+ * Licensed to the Apache Software Foundation (ASF) under one or more >+ * contributor license agreements. See the NOTICE file distributed with >+ * this work for additional information regarding copyright ownership. >+ * The ASF licenses this file to You under the Apache License, Version 2.0 >+ * (the "License"); you may not use this file except in compliance with >+ * the License. You may obtain a copy of the License at >+ * >+ * http://www.apache.org/licenses/LICENSE-2.0 >+ * >+ * Unless required by applicable law or agreed to in writing, software >+ * distributed under the License is distributed on an "AS IS" BASIS, >+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. >+ * See the License for the specific language governing permissions and >+ * limitations under the License. >+ * >+ */ >+ >+package org.apache.jmeter.extractor; >+ >+import java.nio.charset.StandardCharsets; >+ >+import net.minidev.json.parser.JSONParser; >+import net.minidev.json.parser.ParseException; >+ >+import org.apache.jmeter.extractor.json.jsonpath.JSONPostProcessor; >+import org.apache.jmeter.samplers.SampleResult; >+import org.apache.jmeter.threads.JMeterContext; >+import org.apache.jmeter.threads.JMeterContextService; >+import org.apache.jmeter.threads.JMeterVariables; >+import org.junit.Assert; >+import org.junit.Test; >+ >+public class TestJSONPostProcessor { >+ >+ private static final String VAR_NAME = "varName"; >+ >+ @Test >+ public void testBug59609() throws ParseException { >+ JMeterContext context = JMeterContextService.getContext(); >+ JSONPostProcessor processor = setupProcessor(context, "0", false); >+ >+ String innerValue = "{\"a\":\"one\",\"b\":\"two\"}"; >+ String data = "{\"context\":" + innerValue + "}"; >+ SampleResult result = new SampleResult(); >+ result.setResponseData(data.getBytes(StandardCharsets.UTF_8)); >+ >+ JMeterVariables vars = new JMeterVariables(); >+ context.setVariables(vars); >+ context.setPreviousResult(result); >+ >+ processor.setJsonPathExpressions("$.context"); >+ processor.process(); >+ >+ JSONParser parser = new JSONParser(0); >+ Object expectedValue = parser.parse(innerValue); >+ Assert.assertEquals(expectedValue, parser.parse(vars.get(VAR_NAME))); >+ Assert.assertEquals("1", vars.get(VAR_NAME + "_matchNr")); >+ } >+ >+ @Test >+ public void testExtractSimpleArrayElements() { >+ JMeterContext context = JMeterContextService.getContext(); >+ JSONPostProcessor processor = setupProcessor(context, "-1"); >+ String data = "[1,2,3]"; >+ SampleResult result = new SampleResult(); >+ result.setResponseData(data.getBytes(StandardCharsets.UTF_8)); >+ JMeterVariables vars = new JMeterVariables(); >+ context.setVariables(vars); >+ context.setPreviousResult(result); >+ >+ processor.setJsonPathExpressions("$[*]"); >+ processor.process(); >+ >+ Assert.assertEquals("1,2,3", vars.get(VAR_NAME+ "_ALL")); >+ for (int i = 1; i <= 3; i++) { >+ String v = Integer.toString(i); >+ Assert.assertEquals(v, vars.get(VAR_NAME + "_" + v)); >+ } >+ >+ Assert.assertEquals("3", vars.get(VAR_NAME + "_matchNr")); >+ } >+ >+ @Test >+ public void testExtractComplexElements() { >+ JMeterContext context = JMeterContextService.getContext(); >+ JSONPostProcessor processor = setupProcessor(context, "-1"); >+ String data = "[{\"a\":[1,{\"d\":2},3]},[\"b\",{\"h\":23}],3]"; >+ SampleResult result = new SampleResult(); >+ result.setResponseData(data.getBytes(StandardCharsets.UTF_8)); >+ JMeterVariables vars = new JMeterVariables(); >+ context.setVariables(vars); >+ context.setPreviousResult(result); >+ >+ processor.setJsonPathExpressions("$[*]"); >+ processor.process(); >+ >+ String jsonWithoutOuterParens = data.substring(1, data.length() - 1); >+ Assert.assertEquals(jsonWithoutOuterParens, vars.get(VAR_NAME + "_ALL")); >+ >+ Assert.assertEquals("{\"a\":[1,{\"d\":2},3]}", vars.get(VAR_NAME + "_1")); >+ Assert.assertEquals("[\"b\",{\"h\":23}]", vars.get(VAR_NAME + "_2")); >+ Assert.assertEquals("3", vars.get(VAR_NAME + "_3")); >+ >+ Assert.assertEquals("3", vars.get(VAR_NAME + "_matchNr")); >+ } >+ >+ private JSONPostProcessor setupProcessor(JMeterContext context, >+ String matchNumbers) { >+ return setupProcessor(context, matchNumbers, true); >+ } >+ >+ private JSONPostProcessor setupProcessor(JMeterContext context, >+ String matchNumbers, boolean computeConcatenation) { >+ JSONPostProcessor processor = new JSONPostProcessor(); >+ processor.setThreadContext(context); >+ processor.setRefNames(VAR_NAME); >+ processor.setMatchNumbers(matchNumbers); >+ processor.setComputeConcatenation(computeConcatenation); >+ return processor; >+ } >+ >+} View Attachment As Diff View Attachment As Raw

This is ASF Bugzilla: the Apache Software Foundation bug system. In case of problems with the functioning of ASF Bugzilla, please contact bugzilla-admin@apache.org. Please Note: this e-mail address is only for reporting problems with ASF Bugzilla. Mail about any other subject will be silently ignored.

• • Home
  • | New
  • | Browse
  • | Search
  • |
    [?]
  • | Reports
  • | Help
  • | New Account
  • | Log In
    Remember [x]
  • | Forgot Password
    Login: [x]