test: align scoring fixtures and upgrade checks with current contracts

2026-10-17 20:50:48 +08:00
parent a5cb40f83d
commit f7db0a47ac
3 changed files with 5 additions and 3 deletions
@@ -388,7 +388,8 @@ def test_upgrade_from_legacy_0003_migrates_pages_and_citations(session_factory,
    config.set_main_option('script_location', str(root / 'migrations'))
    command.upgrade(config, 'head')
    with session_factory() as db:
-        assert db.scalar(text('SELECT version_num FROM alembic_version')) == '0006'
+        from alembic.script import ScriptDirectory
+        assert db.scalar(text('SELECT version_num FROM alembic_version')) == ScriptDirectory.from_config(config).get_current_head()
        assert db.get(Conversation, conversation_id).title == '旧记录'
        assert db.scalar(select(ChatMessage.content).where(ChatMessage.conversation_id == conversation_id)) == '旧问题'
        run = db.scalar(select(AgentRun).where(AgentRun.conversation_id == conversation_id))