io.restassured.internal.ContentParser.groovy Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of rest-assured Show documentation
Show all versions of rest-assured Show documentation
Java DSL for easy testing of REST services
/*
* Copyright 2019 the original author or authors.
*
* Licensed 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 io.restassured.internal
import groovy.util.slurpersupport.GPathResult
import io.restassured.config.RestAssuredConfig
import io.restassured.config.XmlConfig
import io.restassured.internal.http.CharsetExtractor
import io.restassured.internal.path.json.ConfigurableJsonSlurper
import io.restassured.parsing.Parser
import io.restassured.response.Response
import io.restassured.response.ResponseBodyExtractionOptions
import static io.restassured.parsing.Parser.*
class ContentParser {
def parse(Response response, ResponseParserRegistrar rpr, RestAssuredConfig config, boolean parseAsString) {
Parser parser = rpr.getParser(response.contentType())
def content;
if (parser == null) {
content = response.asInputStream()
} else {
switch (parser) {
case JSON:
def slurper = new ConfigurableJsonSlurper(config.getJsonConfig().numberReturnType())
if (parseAsString) {
content = slurper.parseText(response.asString())
} else {
def charset = CharsetExtractor.getCharsetFromContentType(response.getContentType()) ?: config.getDecoderConfig().defaultCharsetForContentType(response.getContentType());
content = slurper.parse(new InputStreamReader(new BufferedInputStream(response.asInputStream()), charset))
}
break;
case XML:
def xmlConfig = config.getXmlConfig()
def slurper = configureXmlSlurper(new XmlSlurper(xmlConfig.isValidating(), xmlConfig.isNamespaceAware(), xmlConfig.isAllowDocTypeDeclaration()), xmlConfig)
content = declareNamespacesIfNeeded(parseXml(slurper, response, parseAsString), xmlConfig)
break
case HTML:
def xmlConfig = config.getXmlConfig()
def slurper = configureXmlSlurper(new XmlSlurper(new org.ccil.cowan.tagsoup.Parser()), xmlConfig)
content = declareNamespacesIfNeeded(parseXml(slurper, response, parseAsString), xmlConfig)
break
case TEXT:
default:
content = response.asInputStream()
}
}
content
}
def private static GPathResult parseXml(XmlSlurper xmlSlurper, ResponseBodyExtractionOptions response, boolean parseAsString) {
if (parseAsString) {
// We force default charset to be backward compatible with "InputStream charset"
xmlSlurper.parseText(response.asString(true))
} else {
xmlSlurper.parse(response.asInputStream())
}
}
def private static GPathResult declareNamespacesIfNeeded(GPathResult gPathResult, XmlConfig xmlConfig) {
if (xmlConfig.isNamespaceAware()) {
gPathResult.declareNamespace(xmlConfig.declaredNamespaces())
}
gPathResult
}
def private static XmlSlurper configureXmlSlurper(XmlSlurper xmlSlurper, XmlConfig xmlConfig) {
def features = xmlConfig.features();
def properties = xmlConfig.properties()
properties.each { name, value ->
xmlSlurper.setProperty(name, value)
}
features.each { name, isEnabled ->
xmlSlurper.setFeature(name, isEnabled)
}
xmlSlurper
}
}