test: Implement post-test database cleanup in workflow history test

This commit is contained in:
Yunxiao Xu
2026-02-11 16:52:41 -08:00
parent 004f9fd9c9
commit 48924affa0

View File

@@ -8,12 +8,17 @@ from sqlalchemy import delete
def history_manager(): def history_manager():
settings = Settings() settings = Settings()
manager = HistoryManager(settings.history_db_url) manager = HistoryManager(settings.history_db_url)
def cleanup():
with manager.get_session() as session: with manager.get_session() as session:
session.execute(delete(Plot)) session.execute(delete(Plot))
session.execute(delete(Message)) session.execute(delete(Message))
session.execute(delete(Conversation)) session.execute(delete(Conversation))
session.execute(delete(User)) session.execute(delete(User))
return manager
cleanup()
yield manager
cleanup()
def test_full_history_workflow(history_manager): def test_full_history_workflow(history_manager):
# 1. Create and Authenticate User # 1. Create and Authenticate User