from datetime import date, timedelta from app.models import DailyRecordInput, MetricValue, TaskInput from app.repository import ConflictError, JsonRepository def task_payload(): return TaskInput.model_validate({ "name": "学习计划", "startDate": "2026-08-01", "priority": 2, "metrics": [ {"name": "打卡", "kind": "checkin", "period": "day"}, {"name": "刷题", "kind": "quantity", "period": "lifetime", "target": 100, "unit": "题"}, {"name": "专注", "kind": "duration", "period": "day", "target": 3600}, ], }) def test_crud_revision_soft_delete_and_purge(tmp_path): repo = JsonRepository(tmp_path) task = repo.create_task(task_payload()) assert len(task["metrics"]) == 3 updated = repo.update_task(task["taskId"], task_payload().model_copy(update={"name": "改名后的计划"}), task["version"]) assert updated["taskId"] == task["taskId"] and updated["revision"] == 2 assert repo.history(task["taskId"], date(2026, 8, 1), date(2026, 8, 1))["revisions"] deleted = repo.soft_delete_task(task["taskId"], updated["version"]) assert repo.list_tasks()["items"] == [] repo.restore_task(task["taskId"], deleted["version"]) repo.soft_delete_task(task["taskId"], deleted["version"] + 1) repo.purge_task(task["taskId"], task["taskId"]) assert repo.list_tasks(include_deleted=True)["items"] == [] def test_daily_mixed_metrics_history_and_idempotent_archive(tmp_path): repo = JsonRepository(tmp_path) task = repo.create_task(task_payload()) values = [MetricValue(metricId=item["metricId"], value=1 if item["kind"] == "checkin" else 120) for item in task["metrics"]] record = repo.upsert_daily(task["taskId"], date(2026, 8, 10), DailyRecordInput(note="不错", values=values)) assert record["version"] == 1 repo.close_day(date(2026, 8, 10)); repo.close_day(date(2026, 8, 10)) history = repo.history(task["taskId"], date(2026, 8, 10), date(2026, 8, 10)) assert history["records"][0]["note"] == "不错" try: repo.upsert_daily(task["taskId"], date(2026, 8, 10), DailyRecordInput(expectedVersion=0, values=[])) except ConflictError: pass else: raise AssertionError("expected version conflict")