starred/WireMock.Net-wiremock
1
0
Fork 0
mirror of https://github.com/wiremock/WireMock.Net.git synced 2026-03-20 00:03:48 +01:00
Code Issues 32 Packages Projects Releases 10 Wiki Activity

SetBody (#81)

Browse Source
This commit is contained in:
Stef Heyenrath
2018-02-02 15:42:25 +01:00
parent 66b2ff16de
commit e5b2ad0543
1 changed files with 38 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

64
src/WireMock.Net/Http/HttpClientHelper.cs
View File

@@ -92,34 +92,10 @@ namespace WireMock.Http
// Set both content and response headers, replacing URLs in values
var headers = (httpResponseMessage.Content?.Headers.Union(httpResponseMessage.Headers) ?? Enumerable.Empty<KeyValuePair<string, IEnumerable<string>>>()).ToArray();
// In case the Content-Type header is application/json, try to set BodyAsJson, else set Body and BodyAsBytes.
bool bodyAsJson = false;
var contentTypeHeader = headers.FirstOrDefault(header => string.Equals(header.Key, HttpKnownHeaderNames.ContentType, StringComparison.OrdinalIgnoreCase));
if (!contentTypeHeader.Equals(default(KeyValuePair<string, IEnumerable<string>>)) &&
contentTypeHeader.Value.Any(value => value != null && value.StartsWith("application/json", StringComparison.OrdinalIgnoreCase)))
if (httpResponseMessage.Content != null)
{
if (httpResponseMessage.Content != null)
{
string content = await httpResponseMessage.Content.ReadAsStringAsync();
try
{
responseMessage.BodyAsJson = JsonConvert.DeserializeObject(content, new JsonSerializerSettings { Formatting = Formatting.Indented });
bodyAsJson = true;
}
catch
{
}
}
}
if (!bodyAsJson)
{
if (httpResponseMessage.Content != null)
{
responseMessage.BodyAsBytes = await httpResponseMessage.Content.ReadAsByteArrayAsync();
responseMessage.Body = await httpResponseMessage.Content.ReadAsStringAsync();
}
SetBody(httpResponseMessage.Content, contentTypeHeader, responseMessage);
}
foreach (var header in headers)
@@ -141,5 +117,41 @@ namespace WireMock.Http
return responseMessage;
}
private static async void SetBody(HttpContent content, KeyValuePair<string, IEnumerable<string>> contentTypeHeader, ResponseMessage responseMessage)
{
bool contentTypeIsDefault = contentTypeHeader.Equals(default(KeyValuePair<string, IEnumerable<string>>));
string[] textContentTypes = { "text/", "application/xml", "application/javascript", "application/typescript", "application/xhtml+xml" };
if (!contentTypeIsDefault && contentTypeHeader.Value.Any(value => textContentTypes.Any(t => value != null && value.StartsWith(t, StringComparison.OrdinalIgnoreCase))))
{
try
{
responseMessage.Body = await content.ReadAsStringAsync();
}
catch
{
// Reading as string failed, just get the ByteArray.
responseMessage.BodyAsBytes = await content.ReadAsByteArrayAsync();
}
}
else if (!contentTypeIsDefault && contentTypeHeader.Value.Any(value => value != null && value.StartsWith("application/json", StringComparison.OrdinalIgnoreCase)))
{
string stringContent = await content.ReadAsStringAsync();
try
{
responseMessage.BodyAsJson = JsonConvert.DeserializeObject(stringContent, new JsonSerializerSettings { Formatting = Formatting.Indented });
}
catch
{
// JsonConvert failed, just set the Body as string.
responseMessage.Body = stringContent;
}
}
else
{
responseMessage.BodyAsBytes = await content.ReadAsByteArrayAsync();
}
}
}
}