chore: http client healthcheck

src/bentoml/_internal/client/http.py

@@ -4,6 +4,7 @@
 import time
 import socket
 import typing as t
+import asyncio
 import logging
 import urllib.error
 import urllib.request
@@ -77,6 +78,14 @@ def wait_until_server_ready(
             logger.error(err)
             raise
 
+    async def async_health(self) -> t.Any:
+        async with aiohttp.ClientSession(self.server_url) as sess:
+            async with sess.get("/readyz") as resp:
+                return resp
+
+    def health(self) -> t.Any:
+        return asyncio.run(self.async_health())
+
     @classmethod
     def from_url(cls, server_url: str, **kwargs: t.Any) -> HTTPClient:
         server_url = server_url if "://" in server_url else "http://" + server_url

tests/e2e/bento_server_http/tests/test_client.py

@@ -0,0 +1,24 @@
+import pytest
+
+import bentoml
+from bentoml.client import HTTPClient
+
+
+@pytest.fixture(scope="session")
+def http_client(host: str) -> HTTPClient:
+    service = bentoml.load(bento_identifier="service:svc")
+
+    return HTTPClient(service, f"http://{host}")
+
+
+@pytest.mark.asyncio
+async def test_async_health(http_client: HTTPClient) -> None:
+    resp = await http_client.async_health()
+
+    assert resp.status == 200
+
+
+def test_health(http_client: HTTPClient) -> None:
+    resp = http_client.health()
+
+    assert resp.status == 200