Skip to content

Commit b1cbf2e

Browse files
committed
helpers
1 parent 6d10d46 commit b1cbf2e

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

packages/service-library/tests/fastapi/test_lifespan_utils.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -110,7 +110,7 @@ async def setup_database(app: FastAPI, state: State) -> AsyncIterator[State]:
110110

111111
app.state.database_engine = state["postgres"]["engine"]
112112

113-
yield {}
113+
yield {} # no update
114114

115115
# tear-down stage
116116
assert app.state.database_engine
@@ -145,14 +145,17 @@ async def setup_rpc_server(app: FastAPI, state: State) -> AsyncIterator[State]:
145145
assert app.state.database_engine
146146
assert app.state.rpc_server
147147

148+
# NOTE: these are different states!
149+
assert app.state._state != asgi_manager._state # noqa: SLF001
150+
148151
# Logs shows lifespan execution:
149152
# -> postgres_sync_engine starting ...
150153
# -> postgres_async_engine starting ...
151154
# -> app database starting ...
152155
# -> rabbitmq starting ...
153156
# -> app rpc-server starting ...
154-
# <- app rpc-server done ()
155-
# <- rabbitmq done ()
157+
# <- app rpc-server done (<1ms)
158+
# <- rabbitmq done (<1ms)
156159
# <- app database done (1ms)
157160
# <- postgres_async_engine done (1ms)
158161
# <- postgres_sync_engine done (1ms)

0 commit comments

Comments
 (0)