From 48924affa0fc4b1e66af4364f711f630d5de913d Mon Sep 17 00:00:00 2001 From: Yunxiao Xu Date: Wed, 11 Feb 2026 16:52:41 -0800 Subject: [PATCH] test: Implement post-test database cleanup in workflow history test --- tests/test_workflow_history.py | 17 +++++++++++------ 1 file changed, 11 insertions(+), 6 deletions(-) diff --git a/tests/test_workflow_history.py b/tests/test_workflow_history.py index f774f83..09e9068 100644 --- a/tests/test_workflow_history.py +++ b/tests/test_workflow_history.py @@ -8,12 +8,17 @@ from sqlalchemy import delete def history_manager(): settings = Settings() manager = HistoryManager(settings.history_db_url) - with manager.get_session() as session: - session.execute(delete(Plot)) - session.execute(delete(Message)) - session.execute(delete(Conversation)) - session.execute(delete(User)) - return manager + + def cleanup(): + with manager.get_session() as session: + session.execute(delete(Plot)) + session.execute(delete(Message)) + session.execute(delete(Conversation)) + session.execute(delete(User)) + + cleanup() + yield manager + cleanup() def test_full_history_workflow(history_manager): # 1. Create and Authenticate User