lifespan handling for test client

[PaulRudin]

import contextlib
from typing import TypedDict
from starlette.applications import Starlette
from starlette.testclient import TestClient
from starlette.routing import Route
from starlette.responses import PlainTextResponse


class State(TypedDict):
    foo: int


@contextlib.asynccontextmanager
async def lifespan(app: Starlette) -> State:
    yield {"foo": "42"}


async def hello(request):
    return PlainTextResponse(request.app.state.foo)


app = Starlette(
    lifespan=lifespan,
    routes=[Route("/", hello)]
)


def test_client():
    with TestClient(app) as client:
        client.get("/")


if __name__ == "__main__":
    test_client()

running this gives: AttributeError: 'State' object has no attribute 'foo'

Shouldn't state have a foo attribute here?

[Kludex]

It should be request.state.foo, and not request.app.state.foo. They are different.

[PaulRudin]

Ah, hmm - I see. ty.

[PaulRudin]

So... if I want to patch state.foo in a test where can I patch it?

[adriangb]

Replied here: #2088 (comment)