def test_document_on_session_destroyed(self):
doc = Document()
handler = bahd.DocumentLifecycleHandler()
def destroy(session_context):
assert doc is session_context._document
session_context.status = 'Destroyed'
doc.on_session_destroyed(destroy)
session_context = MockSessionContext(doc)
handler.on_session_destroyed(session_context)
assert session_context.status == 'Destroyed'
async def test_document_on_session_destroyed(self) -> None:
doc = Document()
handler = bahd.DocumentLifecycleHandler()
def destroy(session_context):
assert doc is session_context._document
session_context.status = 'Destroyed'
doc.on_session_destroyed(destroy)
session_context = MockSessionContext(doc)
await handler.on_session_destroyed(session_context)
assert session_context.status == 'Destroyed'
assert session_context._document.session_destroyed_callbacks == set()
async def test_document_on_session_destroyed_exceptions(self, caplog: pytest.LogCaptureFixture) -> None:
doc = Document()
def blowup(session_context):
raise ValueError("boom!")
doc.on_session_destroyed(blowup)
handler = bahd.DocumentLifecycleHandler()
session_context = MockSessionContext(doc)
with caplog.at_level(logging.WARN):
await handler.on_session_destroyed(session_context)
assert len(caplog.records) == 1
assert "boom!" in caplog.text
async def test_document_on_session_destroyed_calls_multiple(self) -> None:
doc = Document()
def increment(session_context):
session_context.counter += 1
doc.on_session_destroyed(increment)
def increment_by_two(session_context):
session_context.counter += 2
doc.on_session_destroyed(increment_by_two)
handler = bahd.DocumentLifecycleHandler()
session_context = MockSessionContext(doc)
await handler.on_session_destroyed(session_context)
assert session_context.counter == 3, 'DocumentLifecycleHandler did not call all callbacks'
