Fix head

Author: walkor

src/Request.php

@@ -324,8 +324,8 @@ public function doSend(): void
         $this->writeable = false;
 
         $body_size = $this->getBody()->getSize();
-        if ($body_size) {
-            $this->withHeaders(['Content-Length' => $body_size]);
+        if ($body_size !== null) {
+            $this->withHeader('Content-Length', (string)$body_size);
         }
 
         $package = str($this);
@@ -380,6 +380,10 @@ public function onMessage($connection, $receive_buffer): void
     protected function checkComplete($body): void
     {
         $status_code = $this->response->getStatusCode();
+        if (strtoupper($this->getMethod()) === 'HEAD') {
+            $this->emitSuccess();
+            return;
+        }
         $content_length = $this->response->getHeaderLine('Content-Length');
         if ($content_length === '0' || ($status_code >= 100 && $status_code < 200)
             || $status_code === 204 || $status_code === 304) {

tests/RequestTest.php

@@ -242,6 +242,39 @@ public function testProgressStreamHandling()
         $this->assertTrue($successCalled);
     }
 
+    public function testHeadRequestWithoutBody()
+    {
+        $successCalled = false;
+        $errorCalled = false;
+        $http = new Client();
+        $expectedBody = json_encode(['ok' => true]);
+
+        $options = [
+            'method' => 'HEAD',
+            'success' => function ($response) use (&$successCalled, $expectedBody) {
+                $this->assertInstanceOf(Response::class, $response);
+                $this->assertEquals(200, $response->getStatusCode());
+                $this->assertSame('', (string)$response->getBody());
+                $this->assertSame((string)strlen($expectedBody), $response->getHeaderLine('Content-Length'));
+                $successCalled = true;
+            },
+            'error' => function ($exception) use (&$errorCalled) {
+                $errorCalled = true;
+            }
+        ];
+
+        $http->request('http://127.0.0.1:7171/head', $options);
+
+        for ($i = 0; $i < 10; $i++) {
+            if ($successCalled || $errorCalled) {
+                break;
+            }
+            Timer::sleep(0.1);
+        }
+        $this->assertTrue($successCalled);
+        $this->assertFalse($errorCalled);
+    }
+
     /**
      * Test synchronous GET request
      */

tests/start.php

@@ -78,6 +78,15 @@ function create_test_worker(Closure $callable, $eventLoopClass): void
         case '/get':
             $connection->send(json_encode($request->get()));
             return;
+        case '/head':
+            $body = json_encode(['ok' => true]);
+            if (strtoupper($request->method()) === 'HEAD') {
+                $length = strlen($body);
+                $connection->send("HTTP/1.1 200 OK\r\nServer: workerman\r\nContent-Length: $length\r\nConnection: keep-alive\r\n\r\n", true);
+                return;
+            }
+            $connection->send($body);
+            return;
         case '/post':
             $connection->send(json_encode($request->post()));
             return;