PySide6 실전 강좌 #7 테스트 — pytest-qt와 층별 테스트 전략

# tests/test_stats.py
from datetime import date

from daily.core.stats import completion_rate, current_streak


def test_오늘_체크했으면_오늘부터_연속():
    checked = {"2026-09-28", "2026-09-29", "2026-09-30"}
    assert current_streak(checked, today=date(2026, 9, 30)) == 3


def test_오늘_아직_안_했으면_어제까지로_계산():
    checked = {"2026-09-28", "2026-09-29"}
    assert current_streak(checked, today=date(2026, 9, 30)) == 2


def test_중간에_구멍이_있으면_거기서_끊긴다():
    checked = {"2026-09-27", "2026-09-29", "2026-09-30"}
    assert current_streak(checked, today=date(2026, 9, 30)) == 2


def test_달성률은_기간_전체_대비():
    checked = {"2026-09-29", "2026-09-30"}
    rate = completion_rate(checked, date(2026, 9, 27), date(2026, 9, 30))
    assert rate == 0.5

# tests/test_repository.py
from datetime import date

import pytest

from daily.data.repository import HabitRepository


@pytest.fixture
def repo(tmp_path):
    return HabitRepository(tmp_path / "test.db")


def test_습관_추가와_조회(repo):
    repo.add_habit("운동")
    habits = repo.active_habits()
    assert [h.name for h in habits] == ["운동"]


def test_같은_날_중복_체크는_한_번으로(repo):
    habit_id = repo.add_habit("독서")
    repo.set_checked(habit_id, date(2026, 9, 30), True)
    repo.set_checked(habit_id, date(2026, 9, 30), True)   # 중복
    assert repo.checked_days(habit_id) == {"2026-09-30"}


def test_보관된_습관은_목록에서_빠진다(repo):
    habit_id = repo.add_habit("명상")
    repo.archive_habit(habit_id)
    assert repo.active_habits() == []

# tests/test_today_model.py
from datetime import date

from PySide6.QtCore import Qt

from daily.data.repository import HabitRepository
from daily.ui.today_model import TodayModel


def make_model(tmp_path) -> TodayModel:
    repo = HabitRepository(tmp_path / "t.db")
    repo.add_habit("운동")
    return TodayModel(repo)


def test_모델은_습관_수만큼_행을_가진다(qtbot, tmp_path):
    model = make_model(tmp_path)
    assert model.rowCount() == 1
    index = model.index(0, 0)
    assert model.data(index, Qt.ItemDataRole.DisplayRole) == "운동"


def test_체크_토글이_DB까지_반영된다(qtbot, tmp_path):
    model = make_model(tmp_path)
    index = model.index(0, 0)

    with qtbot.waitSignal(model.dataChanged):     # 시그널 발생을 검증
        model.setData(index, Qt.CheckState.Checked.value,
                      Qt.ItemDataRole.CheckStateRole)

    assert model.data(index, Qt.ItemDataRole.CheckStateRole) == Qt.CheckState.Checked
    assert date.today().isoformat() in model._repo.checked_days(1)

def test_추가_버튼이_다이얼로그를_연다(qtbot, tmp_path, monkeypatch):
    from daily.ui.today_page import TodayPage
    from PySide6.QtWidgets import QInputDialog

    model = make_model(tmp_path)
    page = TodayPage(model)
    qtbot.addWidget(page)                          # 정리를 qtbot에 위임

    monkeypatch.setattr(QInputDialog, "getText",
                        staticmethod(lambda *a, **k: ("스트레칭", True)))
    qtbot.mouseClick(page.add_button, Qt.MouseButton.LeftButton)

    assert model.rowCount() == 2