quick test for client

bentoml · Aug 23, 2023 · e21c6de · e21c6de
1 parent 72cfc8b
commit e21c6de
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/tests/e2e/bento_server_http/service.py b/tests/e2e/bento_server_http/service.py
@@ -217,6 +217,14 @@ async def predict_text_stream(inp: str) -> t.AsyncGenerator[str, None]:
     return stream_runner.count_text_stream.async_stream(inp)
 
 
+@svc.api(
+    input=Text(),
+    output=Text(),
+)
+def yo(inp: str) -> str:
+    return f"yo {inp}"
+
+
 # customise the service
 class AllowPingMiddleware:
     def __init__(

diff --git a/tests/e2e/bento_server_http/tests/test_client.py b/tests/e2e/bento_server_http/tests/test_client.py
@@ -16,3 +16,10 @@ def test_health(host: str) -> None:
     resp = http_client.health()
 
     assert resp.status == 200
+
+
+def test_text_endpoint(host: str) -> None:
+    http_client = HTTPClient.from_url(f"http://{host}")
+    resp = http_client.yo("Bob")
+
+    assert resp == "yo Bob"