mirror of
https://github.com/kuhyx/diet-guard.git
synced 2026-11-10 19:27:14 +01:00
230 lines
8.9 KiB
Python
230 lines
8.9 KiB
Python
"""Tests for the nutrition model, lookup, and meal-building flow of MealGate.
|
|
|
|
Covers :mod:`._gatelock_nutrition` (reference -> total maths, suggestions,
|
|
unit toggling) and :mod:`._gatelock_mealflow` (submit/lookup/record, the
|
|
dashboard, and multi-item meals). The functional fake ``tk`` widgets and the
|
|
``gate`` fixture live in ``conftest.py`` and are shared with
|
|
:mod:`test_gatelock`.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING, cast
|
|
from unittest.mock import MagicMock, patch
|
|
|
|
from diet_guard import _gatelock_delivery
|
|
from diet_guard import _gatelock_glyphs as g
|
|
from diet_guard._budget import write_budget
|
|
from diet_guard._state import log_meal
|
|
from diet_guard.tests.conftest import _nutrition
|
|
|
|
if TYPE_CHECKING:
|
|
from diet_guard._gatelock import MealGate
|
|
|
|
|
|
class TestDashboard:
|
|
"""The running calorie/macro panel."""
|
|
|
|
def test_headline_with_budget(self, gate: MealGate) -> None:
|
|
"""A sealed budget shows consumed/target/remaining."""
|
|
write_budget(2000)
|
|
gate._refresh_dashboard()
|
|
assert gate._vars.cal_headline.get() == f"0 / 2000 kcal {g.BUDGET} 2000"
|
|
|
|
def test_headline_without_budget(self, gate: MealGate) -> None:
|
|
"""With no budget, only today's total is shown."""
|
|
gate._refresh_dashboard()
|
|
assert gate._vars.cal_headline.get() == "0 kcal"
|
|
|
|
def test_dashboard_lists_entries(self, gate: MealGate) -> None:
|
|
"""Logged entries appear in the detail panel."""
|
|
write_budget(2000, weight_kg=80)
|
|
log_meal("apple", _nutrition(95, 100), 8)
|
|
gate._refresh_dashboard()
|
|
text = gate._vars.dashboard.get()
|
|
assert "apple" in text
|
|
assert f"{g.PROTEIN} 10 /" in text
|
|
|
|
def test_dashboard_empty(self, gate: MealGate) -> None:
|
|
"""With nothing logged, the panel says so."""
|
|
gate._refresh_dashboard()
|
|
assert g.EMPTY in gate._vars.dashboard.get()
|
|
|
|
def test_slot_header_variants(self, gate: MealGate) -> None:
|
|
"""The header covers none / one / several pending slots."""
|
|
gate._pending = []
|
|
gate._refresh_slot_header()
|
|
assert gate._vars.slot_header.get() == g.OK
|
|
gate._pending = [480]
|
|
gate._refresh_slot_header()
|
|
assert gate._vars.slot_header.get() == f"{g.CLOCK} 08:00"
|
|
gate._pending = [480, 720]
|
|
gate._refresh_slot_header()
|
|
assert gate._vars.slot_header.get() == f"{g.CLOCK} 08:00 1/2"
|
|
|
|
def test_projection_with_budget(self, gate: MealGate) -> None:
|
|
"""The projection shows the after-this-item remaining when priced."""
|
|
write_budget(2000)
|
|
gate._widgets.macros.kcal.insert(0, "300")
|
|
gate._refresh_projection()
|
|
assert gate._vars.projection.get().endswith(f"→ {g.BUDGET} 1700")
|
|
|
|
|
|
class TestSlotWalk:
|
|
"""Slot tagging and the per-slot input reset."""
|
|
|
|
def test_slot_for_log_demo_is_none(self, gate: MealGate) -> None:
|
|
"""A demo gate tags logs with no real slot."""
|
|
gate._pending = [480]
|
|
assert gate._slot_for_log() is None
|
|
|
|
def test_slot_for_log_production_is_slot(self, gate: MealGate) -> None:
|
|
"""A production gate tags logs with the current slot."""
|
|
gate.demo_mode = False
|
|
gate._pending = [720]
|
|
assert gate._slot_for_log() == 720
|
|
|
|
def test_clear_inputs_resets_the_form(self, gate: MealGate) -> None:
|
|
"""Clearing between slots empties the description and the macros."""
|
|
gate._set_desc("salad")
|
|
gate._widgets.macros.kcal.insert(0, "80")
|
|
gate._clear_inputs()
|
|
assert gate._get_desc() == ""
|
|
assert gate._widgets.macros.kcal.get() == ""
|
|
|
|
def test_finish_slot_unlocks_on_last(self, gate: MealGate) -> None:
|
|
"""Finishing the final slot triggers unlock."""
|
|
gate._pending = [1200]
|
|
with patch.object(gate, "_unlock") as unlock:
|
|
gate._finish_slot("done")
|
|
unlock.assert_called_once()
|
|
|
|
|
|
def _drive_fetch(gate: MealGate) -> None:
|
|
"""Run a started fetch to completion, deterministically.
|
|
|
|
``_on_fetch_sync`` now hands the pull to a worker thread and schedules a
|
|
poll via ``root.after``; under test ``after`` is a MagicMock, so nothing
|
|
would ever pick the result up. Joining on the queue and calling the poll
|
|
directly keeps these assertions synchronous without a sleep.
|
|
"""
|
|
result = gate._fetch_result
|
|
assert result is not None, "no fetch was started"
|
|
result.put(result.get(timeout=5))
|
|
gate._poll_fetch()
|
|
|
|
|
|
class TestFetchFromSync:
|
|
"""The manual "Fetch from sync" button on the lock screen."""
|
|
|
|
def test_demo_mode_does_not_sync(self, gate: MealGate) -> None:
|
|
"""In demo the button is inert and never touches the network."""
|
|
with patch.object(_gatelock_delivery, "pull_peer_logs") as pull:
|
|
gate._on_fetch_sync()
|
|
pull.assert_not_called()
|
|
assert gate._vars.status.get() == f"{g.DEMO_ONLY} {g.SYNC}"
|
|
|
|
def test_pull_failure_keeps_lock(self, gate: MealGate) -> None:
|
|
"""A failed pull shows the reason and leaves pending slots intact."""
|
|
gate.demo_mode = False
|
|
gate._pending = [480, 720]
|
|
with patch.object(
|
|
_gatelock_delivery,
|
|
"pull_peer_logs",
|
|
return_value="sync unavailable (x)",
|
|
):
|
|
gate._on_fetch_sync()
|
|
_drive_fetch(gate)
|
|
assert gate._pending == [480, 720]
|
|
assert gate._vars.status.get() == f"{g.ERROR} {g.SYNC}"
|
|
|
|
def test_no_new_meals_keeps_all_slots(self, gate: MealGate) -> None:
|
|
"""A clean pull that satisfies nothing reports so and keeps the slots."""
|
|
gate.demo_mode = False
|
|
gate._pending = [480, 720]
|
|
with (
|
|
patch.object(_gatelock_delivery, "pull_peer_logs", return_value=None),
|
|
patch.object(_gatelock_delivery, "due_slots", return_value=(480, 720)),
|
|
):
|
|
gate._on_fetch_sync()
|
|
_drive_fetch(gate)
|
|
assert gate._pending == [480, 720]
|
|
assert gate._vars.status.get() == f"{g.EMPTY} {g.SYNC}"
|
|
|
|
def test_partial_advances_to_next_slot(self, gate: MealGate) -> None:
|
|
"""One slot pulled in leaves the rest; the window advances (singular)."""
|
|
gate.demo_mode = False
|
|
gate._pending = [480, 720]
|
|
with (
|
|
patch.object(_gatelock_delivery, "pull_peer_logs", return_value=None),
|
|
patch.object(_gatelock_delivery, "due_slots", return_value=(720,)),
|
|
):
|
|
gate._on_fetch_sync()
|
|
_drive_fetch(gate)
|
|
assert gate._pending == [720]
|
|
assert gate._vars.status.get() == f"{g.OK} {g.SYNC} 1"
|
|
|
|
def test_partial_shows_the_count(self, gate: MealGate) -> None:
|
|
"""Two slots pulled in shows the count 2."""
|
|
gate.demo_mode = False
|
|
gate._pending = [480, 720, 960]
|
|
with (
|
|
patch.object(_gatelock_delivery, "pull_peer_logs", return_value=None),
|
|
patch.object(_gatelock_delivery, "due_slots", return_value=(960,)),
|
|
):
|
|
gate._on_fetch_sync()
|
|
_drive_fetch(gate)
|
|
assert gate._pending == [960]
|
|
assert gate._vars.status.get() == f"{g.OK} {g.SYNC} 2"
|
|
|
|
def test_all_satisfied_unlocks(self, gate: MealGate) -> None:
|
|
"""When the pull satisfies every pending slot, the gate unlocks."""
|
|
gate.demo_mode = False
|
|
gate._pending = [480, 720]
|
|
with (
|
|
patch.object(_gatelock_delivery, "pull_peer_logs", return_value=None),
|
|
patch.object(_gatelock_delivery, "due_slots", return_value=()),
|
|
):
|
|
gate._on_fetch_sync()
|
|
_drive_fetch(gate)
|
|
assert gate._pending == []
|
|
assert g.UNLOCKING in gate._vars.status.get()
|
|
|
|
def test_poll_without_a_fetch_is_inert(self, gate: MealGate) -> None:
|
|
"""A stray poll after completion must stop, not block on an empty queue."""
|
|
gate._fetch_result = None
|
|
|
|
gate._poll_fetch() # must simply return
|
|
|
|
assert gate._fetch_result is None
|
|
|
|
def test_poll_rearms_while_the_worker_is_still_running(
|
|
self, gate: MealGate
|
|
) -> None:
|
|
"""An empty queue means "not finished yet" -- reschedule, do not block."""
|
|
import queue as _queue
|
|
|
|
gate.demo_mode = False
|
|
gate._fetch_result = _queue.Queue()
|
|
|
|
gate._poll_fetch()
|
|
|
|
after = cast("MagicMock", gate.root.after)
|
|
after.assert_called_with(_gatelock_delivery.FETCH_POLL_MS, gate._poll_fetch)
|
|
assert gate._fetch_result is not None
|
|
|
|
def test_a_second_click_while_fetching_is_ignored(self, gate: MealGate) -> None:
|
|
"""Two workers would race two log writes and orphan one result."""
|
|
gate.demo_mode = False
|
|
with patch.object(
|
|
_gatelock_delivery, "pull_peer_logs", return_value=None
|
|
) as pull:
|
|
gate._on_fetch_sync()
|
|
first = gate._fetch_result
|
|
gate._on_fetch_sync() # second click, still in flight
|
|
|
|
assert gate._fetch_result is first
|
|
_drive_fetch(gate)
|
|
|
|
assert pull.call_count == 1
|