diff options
Diffstat (limited to 'http-client')
-rw-r--r-- | http-client/src/main/java/ai/vespa/hosted/client/MockHttpClient.java | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/http-client/src/main/java/ai/vespa/hosted/client/MockHttpClient.java b/http-client/src/main/java/ai/vespa/hosted/client/MockHttpClient.java index 6c2a882f990..97ef58ea76d 100644 --- a/http-client/src/main/java/ai/vespa/hosted/client/MockHttpClient.java +++ b/http-client/src/main/java/ai/vespa/hosted/client/MockHttpClient.java @@ -57,9 +57,11 @@ public class MockHttpClient extends AbstractHttpClient { public void expect(BiFunction<HttpURL, String, String> mapper, int status) { expect(request -> { try { - BasicClassicHttpResponse response = new BasicClassicHttpResponse(status); ByteArrayOutputStream buffer = new ByteArrayOutputStream(); - request.getEntity().writeTo(buffer); + if (request.getEntity() != null) + request.getEntity().writeTo(buffer); + + BasicClassicHttpResponse response = new BasicClassicHttpResponse(status); response.setEntity(HttpEntities.create(mapper.apply(HttpURL.from(request.getUri()), buffer.toString(UTF_8)), ContentType.APPLICATION_JSON)); return response; |