asgi-lifespan-middleware

使用简单的异步上下文管理器接口支持ASGI生命周期的ASGI中间件。

此中间件接受一个要包装的ASGI应用程序和一个异步上下文管理器生命周期。它将运行直接传递给它的生命周期以及ASGI应用程序的生命周期（如果包装的ASGI应用程序支持生命周期）。

示例（Starlette）

Starlette应用程序已经支持生命周期，所以我们只需使用TestClient对无操作的普通ASGI应用程序进行测试。

from contextlib import asynccontextmanager
from typing import AsyncIterator

from starlette.testclient import TestClient
from starlette.types import ASGIApp, Scope, Send, Receive

from asgi_lifespan_middleware import LifespanMiddleware

@asynccontextmanager
async def lifespan(
    # you'll get the wrapped app injected
    app: ASGIApp,
) -> AsyncIterator[None]:
    print("setup")
    yield
    print("teardown")


async def app(scope: Scope, receive: Receive, send: Send) -> None:
    ...  # do nothing


wrapped_app = LifespanMiddleware(
    app,
    lifespan=lifespan,
)

with TestClient(wrapped_app):
    pass