fix invoke HEAD method failure (#952)

* fix invoke HEAD method failure Signed-off-by: MregXN <[email protected]> * add unit test Signed-off-by: MregXN <[email protected]> * remove unused package Signed-off-by: MregXN <[email protected]> --------- Signed-off-by: MregXN <[email protected]> Co-authored-by: Artur Souza <[email protected]>
dapr · Nov 27, 2023 · c91c410 · c91c410
1 parent e4cb03f
commit c91c410
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/sdk/src/main/java/io/dapr/client/DaprHttp.java b/sdk/src/main/java/io/dapr/client/DaprHttp.java
@@ -308,6 +308,8 @@ private CompletableFuture<Response> doInvokeApi(String method,
       requestBuilder.get();
     } else if (HttpMethods.DELETE.name().equals(method)) {
       requestBuilder.delete();
+    } else if (HttpMethods.HEAD.name().equals(method)) {
+      requestBuilder.head();  
     } else {
       requestBuilder.method(method, body);
     }

diff --git a/sdk/src/test/java/io/dapr/client/DaprHttpTest.java b/sdk/src/test/java/io/dapr/client/DaprHttpTest.java
@@ -30,14 +30,14 @@
 import uk.org.webcompere.systemstubs.jupiter.SystemStubsExtension;
 
 import java.io.IOException;
+import java.net.HttpURLConnection;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertNull;
-import static org.junit.jupiter.api.Assertions.assertThrows;
 import static org.junit.jupiter.api.Assertions.fail;
 
 @ExtendWith(SystemStubsExtension.class)
@@ -138,6 +138,16 @@ public void invokeDeleteMethod() throws IOException {
     assertEquals(EXPECTED_RESULT, body);
   }
 
+  @Test
+  public void invokeHEADMethod() throws IOException {
+    mockInterceptor.addRule().head("http://127.0.0.1:3500/v1.0/state").respond(HttpURLConnection.HTTP_OK);
+    DaprHttp daprHttp = new DaprHttp(Properties.SIDECAR_IP.get(), 3500, okHttpClient);
+    Mono<DaprHttp.Response> mono =
+        daprHttp.invokeApi("HEAD", "v1.0/state".split("/"), null, (String) null, null, Context.empty());
+    DaprHttp.Response response = mono.block();
+    assertEquals(HttpURLConnection.HTTP_OK, response.getStatusCode());
+  }
+
   @Test
   public void invokeGetMethod() throws IOException {
     mockInterceptor.addRule()