Check for null entity

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;