Merge pull request Azure#141 from jianghaolu/headfix

Fix HEAD response in azure builder
sima-zhu · Feb 10, 2017 · 1e3a2de · 1e3a2de
2 parents 8301c22 + ab07616
commit 1e3a2de
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/azure-client-runtime/src/main/java/com/microsoft/azure/AzureResponseBuilder.java b/azure-client-runtime/src/main/java/com/microsoft/azure/AzureResponseBuilder.java
@@ -84,7 +84,11 @@ public <THeader> ServiceResponseWithHeaders<T, THeader> buildWithHeaders(Respons
 
     @Override
     public <THeader> ServiceResponseWithHeaders<T, THeader> buildEmptyWithHeaders(Response<Void> response, Class<THeader> headerType) throws IOException {
-        return baseBuilder.buildEmptyWithHeaders(response, headerType);
+        ServiceResponse<T> bodyResponse = buildEmpty(response);
+        ServiceResponseWithHeaders<T, THeader> baseResponse = baseBuilder.buildEmptyWithHeaders(response, headerType);
+        ServiceResponseWithHeaders<T, THeader> serviceResponse = new ServiceResponseWithHeaders<>(baseResponse.headers(), bodyResponse.headResponse());
+        serviceResponse.withBody(bodyResponse.body());
+        return serviceResponse;
     }
 
     /**