more UT on Runner

manatlan · Jun 8, 2024 · 3d310a6 · 3d310a6
1 parent e26a58e
commit 3d310a6
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -97,8 +97,10 @@ IMPORTANT : the "tag.update" feature doesn't reset the inactivity timeout !
 Example to add a static endpoint :
 
 ```python
+from starlette.responses import PlainTextResponse
+
 async def serve(req):
-    return open("...","r").read()
+    return PlainTextResponse("body {}")
 
 app=Runner( App, debug=False, ssl=True )
 app.add_route("/my.css", serve)

diff --git a/test_runner.py b/test_runner.py
@@ -3,6 +3,7 @@
 from htagweb import Runner
 import sys,json
 from starlette.testclient import TestClient
+from starlette.responses import PlainTextResponse
 
 import bs4
 
@@ -37,7 +38,6 @@ def test_runner_http_mode():
     def do_tests(client):
         response = client.get('/')
 
-        # assert that get bootstrap page
         assert response.status_code == 200
         assert "<!DOCTYPE html>" in response.text
 
@@ -51,3 +51,45 @@ def do_tests(client):
     app=Runner("examples.simple.App",http_only=True)
     with TestClient(app) as client:
         do_tests(client)
+
+
+def test_runner_route_static():
+
+    def do_tests(client):
+        # assert nothing is at "/"
+        response = client.get('/')
+        assert response.status_code == 404
+
+        # but something is at "/hello"
+        response = client.get('/hello')
+        assert response.status_code == 200
+        assert response.text == "world"
+
+
+    async def serve(req): return PlainTextResponse("world")
+
+    app=Runner()
+    app.add_route("/hello", serve)
+    with TestClient(app) as client:
+        do_tests(client)
+
+
+def test_runner_route_app():
+
+    def do_tests(client):
+        # assert nothing is at "/"
+        response = client.get('/')
+        assert response.status_code == 404
+
+        # but something is at "/hello"
+        response = client.get('/an_app')
+        assert response.status_code == 200
+        assert "<!DOCTYPE html>" in response.text
+
+    async def serve_an_app(req):
+        return await req.app.handle(req,"examples.simple.App") 
+
+    app=Runner()
+    app.add_route("/an_app", serve_an_app)
+    with TestClient(app) as client:
+        do_tests(client)