fix: sanic example raises no such table (#1917)

* fix: sanic example not work

* Fix sanic testing manager not generate schemas

Author: waketzheng

.github/workflows/ci.yml

@@ -73,9 +73,11 @@ jobs:
         run: |
           PYTHONPATH=$DEST_FASTAPI pytest $PYTEST_ARGS $DEST_FASTAPI/_tests.py
           PYTHONPATH=$DEST_BLACKSHEEP pytest $PYTEST_ARGS $DEST_BLACKSHEEP/_tests.py
+          PYTHONPATH=$DEST_SANIC pytest $PYTEST_ARGS $DEST_SANIC/_tests.py
         env:
           DEST_FASTAPI: examples/fastapi
           DEST_BLACKSHEEP: examples/blacksheep
+          DEST_SANIC: examples/sanic
           PYTHONDEVMODE: 1
           PYTEST_ARGS: "-n auto --cov=tortoise --cov-append --cov-branch --tb=native -q"
       - name: Upload Coverage

.gitignore

@@ -53,6 +53,7 @@ coverage.xml
 .hypothesis/
 .vscode/
 examples/*.sqlite3
+examples/*/*.sqlite3*
 
 # Translations
 *.mo
@@ -108,4 +109,4 @@ ENV/
 .mypy_cache/
 
 # macos
-.DS_Store
+.DS_Store

examples/sanic/_tests.py

@@ -0,0 +1,38 @@
+import re
+from pathlib import Path
+
+import pytest
+from sanic_testing import TestManager
+
+try:
+    import main
+except ImportError:
+    if (cwd := Path.cwd()) == (parent := Path(__file__).parent):
+        dirpath = "."
+    else:
+        dirpath = str(parent.relative_to(cwd))
+    print(f"You may need to explicitly declare python path:\n\nexport PYTHONPATH={dirpath}\n")
+    raise
+
+
+@pytest.fixture(scope="module")
+def anyio_backend() -> str:
+    return "asyncio"
+
+
+@pytest.fixture
+def app():
+    sanic_app = main.app
+    TestManager(sanic_app)
+    return sanic_app
+
+
+@pytest.mark.anyio
+async def test_basic_asgi_client(app):
+    request, response = await app.asgi_client.get("/")
+    assert response.status == 200
+    assert b'{"users":[' in response.body
+
+    request, response = await app.asgi_client.post("/user")
+    assert response.status == 200
+    assert re.match(rb'{"user":"User \d+: New User"}$', response.body)

examples/sanic/main.py

@@ -17,16 +17,16 @@ async def list_all(request):
     return response.json({"users": [str(user) for user in users]})
 
 
-@app.route("/user")
+@app.post("/user")
 async def add_user(request):
     user = await Users.create(name="New User")
     return response.json({"user": str(user)})
 
 
 register_tortoise(
-    app, db_url="sqlite://:memory:", modules={"models": ["models"]}, generate_schemas=True
+    app, db_url="sqlite://db.sqlite3", modules={"models": ["models"]}, generate_schemas=True
 )
 
 
 if __name__ == "__main__":
-    app.run(port=5000)
+    app.run(port=5000, debug=True)

poetry.lock

@@ -81,6 +81,7 @@ twine = "*"
 quart = "*"
 # Sample integration - Sanic
 sanic = "*"
+sanic-testing = "*"
 # Sample integration - Starlette
 starlette = "*"
 # Pydantic support

pyproject.toml

@@ -87,17 +87,20 @@ async def tortoise_init() -> None:
 
     if generate_schemas:
 
-        @app.listener("main_process_start")
-        async def init_orm_main(app, loop):  # pylint: disable=W0612
+        @app.main_process_start
+        async def init_orm_main(app):  # pylint: disable=W0612
             await tortoise_init()
             logger.info("Tortoise-ORM generating schema")
             await Tortoise.generate_schemas()
 
-    @app.listener("before_server_start")
-    async def init_orm(app, loop):  # pylint: disable=W0612
+    @app.before_server_start
+    async def init_orm(app):
         await tortoise_init()
+        if generate_schemas and getattr(app, "_test_manager", None):
+            # Running by sanic-testing
+            await Tortoise.generate_schemas()
 
-    @app.listener("after_server_stop")
-    async def close_orm(app, loop):  # pylint: disable=W0612
+    @app.after_server_stop
+    async def close_orm(app):  # pylint: disable=W0612
         await connections.close_all()
         logger.info("Tortoise-ORM shutdown")