3db196e49d
基于 skill-template 脚手架开发，包含： - payroll_config（费率配置，可编辑）+ payslip_snapshots（解析历史，只读）2 张业务表 - 全国默认 + 北上广 4 套费率（社保/公积金/个税起征点/7 级税率表） - 5 个业务 Action：calculate-payslip-simple / calculate-payslip-full / explain-term / view-config / edit-config - 个税累计预扣法计算，支持多月历史累加 - 20 个薪资名词库（社保/公积金/个税/全勤/加班费等） - 完整市场文档（README/TUTORIAL/DEMO/CHANGELOG） - 296 测试全部通过
67 lines
1.9 KiB
Plaintext
67 lines
1.9 KiB
Plaintext
# -*- coding: utf-8 -*-
|
||
"""旧库废弃表迁移样例（复制到 tests/ 并按技能改写）。
|
||
|
||
验收：给定旧库含废弃物理表 + 元数据 + 历史行 → init_db() 后表与元数据均清除。
|
||
见 references/SCHEMA.md「表增删改与退役」。
|
||
"""
|
||
from __future__ import annotations
|
||
|
||
import unittest
|
||
|
||
from _support import IsolatedDataRoot
|
||
|
||
|
||
class TestTableDeprecationMigration(unittest.TestCase):
|
||
def test_init_db_drops_legacy_task_logs(self) -> None:
|
||
with IsolatedDataRoot():
|
||
from db.connection import get_conn, init_db
|
||
from db.display_metadata import iter_user_tables
|
||
|
||
# 1) 模拟升级前的旧库
|
||
init_db()
|
||
conn = get_conn()
|
||
try:
|
||
cur = conn.cursor()
|
||
cur.execute(
|
||
"""
|
||
CREATE TABLE task_logs (
|
||
id INTEGER PRIMARY KEY,
|
||
task_type TEXT,
|
||
status TEXT
|
||
)
|
||
"""
|
||
)
|
||
cur.execute(
|
||
"""
|
||
INSERT INTO _jiangchang_tables
|
||
(table_name, display_name, visible, readonly)
|
||
VALUES ('task_logs', '任务日志', 1, 0)
|
||
"""
|
||
)
|
||
cur.execute("INSERT INTO task_logs (task_type, status) VALUES ('run', 'ok')")
|
||
conn.commit()
|
||
finally:
|
||
conn.close()
|
||
|
||
# 2) 新代码 init_db() 应幂等清除（在 connection.init_db 中调用 drop_deprecated_user_table）
|
||
init_db()
|
||
|
||
conn = get_conn()
|
||
try:
|
||
cur = conn.cursor()
|
||
cur.execute(
|
||
"SELECT 1 FROM sqlite_master WHERE type='table' AND name='task_logs'"
|
||
)
|
||
self.assertIsNone(cur.fetchone())
|
||
cur.execute(
|
||
"SELECT COUNT(*) FROM _jiangchang_tables WHERE table_name='task_logs'"
|
||
)
|
||
self.assertEqual(int(cur.fetchone()[0]), 0)
|
||
self.assertNotIn("task_logs", list(iter_user_tables(cur)))
|
||
finally:
|
||
conn.close()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
unittest.main()
|