改造为 SQLite 原生读写

2026-07-06 16:51:14 +08:00
parent 9884e91612
commit 73d7d766eb
23 changed files with 2974 additions and 893 deletions
@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+from pathlib import Path
+import sys
+import tempfile
+
+ROOT = Path(__file__).resolve().parents[1]
+sys.path.insert(0, str(ROOT))
+
+from app import repository
+from app.domain import Account, Payment, Teacher
+
+
+def assert_equal(name: str, actual, expected) -> None:
+    if actual != expected:
+        raise AssertionError(f"{name}: got {actual!r}, expected {expected!r}")
+
+
+def main() -> None:
+    with tempfile.TemporaryDirectory() as temp_dir:
+        repository.SQLITE_DB_PATH = Path(temp_dir) / "xsk_test.db"
+        repository.initialize_runtime_database()
+
+        student = repository.create_student_profile(
+            Account("", "甲", [Payment("2026-07-01", 10)], 0, "正常", 2020, "")
+        )
+        assert_equal("新增学生", student["account"]["student"], "甲")
+
+        teacher = repository.create_teacher(Teacher("", "王老师", "", ["英语"], "在岗", ""))
+        assert_equal("新增教师", teacher["teacher"]["name"], "王老师")
+
+        record = repository.register_class_record_lines(
+            line="2026.07.06-星期一-10:00-11:00-甲-1小时0分-王老师-英语"
+        )
+        assert_equal("登记上课", record["registered"], 1)
+
+        repository.register_payment_lines(line="甲-2026-07-02:2")
+        students = repository.students_payload()
+        assert_equal("学生数量", students["count"], 1)
+        assert_equal("余额重算", students["students"][0]["remaining"], 11)
+
+        queried = repository.query_public_records("甲7月英语")
+        assert_equal("查询记录", queried["total_records"], 1)
+        record_id = queried["records"][0]["record_id"]
+
+        correction = repository.submit_public_correction_tasks([{"record_id": record_id, "time": "11:00-12:00"}])
+        approved = repository.approve_admin_task(int(correction["items"][0]["id"]))
+        assert_equal("批准纠错", approved["task"]["status"], "approved")
+        assert_equal("纠错生效", repository.query_public_records("甲7月英语")["records"][0]["time"], "11:00-12:00")
+
+        supplement = repository.supplement_course_summary(
+            repository.query_public_records("甲7月英语")["records"][0]["record_id"],
+            "课堂表现很好",
+        )
+        assert_equal("补充小结", supplement["status"], "saved")
+        summaries = repository.query_course_summaries(q="课堂表现")
+        assert_equal("小结查询", summaries["count"], 1)
+        summary_id = summaries["items"][0]["id"]
+        repository.update_course_summary_body(summary_id, "更新后的课堂表现")
+        assert_equal("小结正文更新", repository.query_course_summaries(q="更新后")["count"], 1)
+
+        latest = repository.list_operation_logs()["items"][0]
+        rollback = repository.rollback_operation_log(str(latest["id"]))
+        assert_equal("撤回模式", rollback["rollback_mode"], "sqlite_snapshot")
+        assert_equal("撤回日志", repository.list_operation_logs()["items"][0]["operation"], "撤回操作")
+
+    print("db smoke test passed")
+
+
+if __name__ == "__main__":
+    main()