"""数据导出/导入测试"""

from datetime import date
import json

from src.vitals.core import database as db
from src.vitals.core.export import export_all_data_json, export_to_csv, import_from_json
from src.vitals.core.models import Exercise, Meal, Sleep, Weight, UserConfig


def test_export_all_data_json(tmp_path):
    db.save_config(UserConfig(age=28, gender="male", height=175.0, weight=72.0))
    db.add_exercise(Exercise(date=date(2026, 1, 18), type="跑步", duration=30, calories=240))
    db.add_meal(Meal(date=date(2026, 1, 18), meal_type="午餐", description="米饭+鸡蛋", calories=400))
    db.add_sleep(Sleep(date=date(2026, 1, 18), duration=7.5, quality=4))
    db.add_weight(Weight(date=date(2026, 1, 18), weight_kg=72.5))

    out = tmp_path / "export.json"
    export_all_data_json(out)

    assert out.exists()
    data = json.loads(out.read_text(encoding="utf-8"))
    assert data["version"] == "1.0"
    assert "exercises" in data and len(data["exercises"]) == 1
    assert "meals" in data and len(data["meals"]) == 1
    assert "sleep" in data and len(data["sleep"]) == 1
    assert "weight" in data and len(data["weight"]) == 1
    assert data["config"]["age"] == 28


def test_export_to_csv_exercise(tmp_path):
    db.add_exercise(Exercise(date=date(2026, 1, 18), type="跑步", duration=30, calories=240))

    out = tmp_path / "exercise.csv"
    export_to_csv("exercise", out)

    assert out.exists()
    content = out.read_text(encoding="utf-8")
    assert "date,type,duration,calories" in content
    assert "2026-01-18" in content


def test_import_from_json_roundtrip(tmp_path):
    # 先写入数据并导出
    db.save_config(UserConfig(age=28, gender="male", height=175.0, weight=72.0))
    db.add_exercise(Exercise(date=date(2026, 1, 18), type="跑步", duration=30, calories=240))
    db.add_weight(Weight(date=date(2026, 1, 18), weight_kg=72.5))

    out = tmp_path / "export.json"
    export_all_data_json(out)

    # 新的临时 DB 会在 conftest.py 中自动创建，这里直接导入并验证统计
    stats = import_from_json(out)
    assert stats["exercise"] >= 1
    assert stats["weight"] >= 1