mirror of
https://github.com/kuhyx/diet-guard.git
synced 2026-11-11 22:53:51 +01:00
Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
233 lines
8.6 KiB
Python
233 lines
8.6 KiB
Python
"""Shared fixtures for diet_guard tests.
|
|
|
|
Three safety nets run for every test:
|
|
|
|
* ``_isolate_state`` redirects the food log, sealed budget, gate lock, and
|
|
sync token into ``tmp_path`` so a test can never read or clobber the real
|
|
``~/.local/share`` or ``~/.config/diet_guard``.
|
|
* ``_block_real_tk`` swaps ``tk`` and the ``GateRoot`` window class inside
|
|
``_gatelock`` for mocks, so no test can open a real fullscreen window or grab
|
|
the keyboard even if it forgets to.
|
|
* ``_block_real_vt`` makes ``gatelock``'s VT-switch disable a no-op, so a
|
|
prod-mode (``demo_mode=False``) gate built in a test never runs a real
|
|
``setxkbmap`` against the live X session.
|
|
|
|
The ``gate`` fixture and its supporting fakes (``FakeEntry``, ``_FAKE_TK``, ...)
|
|
build a demo :class:`~diet_guard._gatelock.MealGate` whose widgets
|
|
are functional in-memory stand-ins, shared by ``test_gatelock.py`` and
|
|
``test_gatelock_mealflow.py``.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
from contextlib import ExitStack
|
|
from typing import TYPE_CHECKING
|
|
from unittest.mock import patch
|
|
|
|
import freedays
|
|
import pytest
|
|
|
|
from diet_guard._estimator import Nutrition
|
|
|
|
# Importing these fixtures IS their registration: pytest discovers fixtures
|
|
# defined *or imported* in a conftest, and ``pytest_plugins`` is an error in a
|
|
# non-root conftest. They look unused and are not -- do not "tidy" them away.
|
|
from diet_guard.tests._gate_fixtures import (
|
|
_GATE_TK_MODULES,
|
|
FAKE_OUTPUTS,
|
|
TWO_OUTPUTS,
|
|
_block_real_tk,
|
|
_block_real_vt,
|
|
_hermetic_gatelock,
|
|
_hmac_key,
|
|
dual_output,
|
|
fake_tk,
|
|
gate,
|
|
)
|
|
from diet_guard.tests._tk_fakes import (
|
|
_FAKE_TK,
|
|
_FAKE_TTK,
|
|
FakeCanvas,
|
|
FakeEntry,
|
|
FakeListbox,
|
|
FakeNotebook,
|
|
FakeRadiobutton,
|
|
FakeScrollbar,
|
|
FakeStyle,
|
|
FakeText,
|
|
FakeVar,
|
|
FakeWidget,
|
|
_FakeTclError,
|
|
)
|
|
|
|
# Re-exported: the fake widgets moved into the package so this file stays
|
|
# under the 250-line cap, but tests import them from conftest by name.
|
|
__all__ = [
|
|
"FAKE_OUTPUTS",
|
|
"TWO_OUTPUTS",
|
|
"_FAKE_TK",
|
|
"_FAKE_TTK",
|
|
# The autouse fixtures: importing them here is what registers them with
|
|
# pytest, and naming them here is what stops a linter deleting the import.
|
|
"_GATE_TK_MODULES",
|
|
"FakeCanvas",
|
|
"FakeEntry",
|
|
"FakeListbox",
|
|
"FakeNotebook",
|
|
"FakeRadiobutton",
|
|
"FakeScrollbar",
|
|
"FakeStyle",
|
|
"FakeText",
|
|
"FakeVar",
|
|
"FakeWidget",
|
|
"_FakeTclError",
|
|
"_block_real_tk",
|
|
"_block_real_vt",
|
|
"_hermetic_gatelock",
|
|
"_hmac_key",
|
|
"dual_output",
|
|
"fake_tk",
|
|
"gate",
|
|
]
|
|
|
|
if TYPE_CHECKING:
|
|
from collections.abc import Iterator
|
|
from pathlib import Path
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def _isolate_state(tmp_path: Path) -> Iterator[None]:
|
|
"""Redirect all on-disk diet_guard state into a temp dir.
|
|
|
|
Built as a list fed through an ``ExitStack`` rather than one ``with``
|
|
tuple: the tuple form is a statically nested block per entry, and at
|
|
~20 redirects CPython refuses to compile it.
|
|
"""
|
|
redirects = [
|
|
# The shared free-day pool lives under ~/.local/share, outside every
|
|
# diet_guard path below. Without this a real free day on the
|
|
# developer's machine would make every "the gate is due" test here
|
|
# fail, for a reason that looks nothing like the cause.
|
|
patch(
|
|
"freedays._api.resolve_paths",
|
|
lambda paths: paths or freedays.Paths.under(tmp_path / "freedays"),
|
|
),
|
|
patch(
|
|
"diet_guard._budget.BUDGET_FILE",
|
|
tmp_path / ".budget",
|
|
),
|
|
patch(
|
|
"diet_guard._budget_history.BUDGET_HISTORY_FILE",
|
|
tmp_path / ".budget_history",
|
|
),
|
|
patch(
|
|
"diet_guard._meal_schedule_store.MEAL_SCHEDULE_FILE",
|
|
tmp_path / ".meal_schedule",
|
|
),
|
|
patch(
|
|
"diet_guard._state.FOOD_LOG_FILE",
|
|
tmp_path / "food_log.json",
|
|
),
|
|
patch(
|
|
"diet_guard._foodbank.FOOD_BANK_FILE",
|
|
tmp_path / "food_bank.json",
|
|
),
|
|
patch(
|
|
"diet_guard._foodbank_manual.MANUAL_BANK_FILE",
|
|
tmp_path / "food_bank_manual.json",
|
|
),
|
|
patch(
|
|
"diet_guard._gatelock_lockfile.GATE_LOCK_FILE",
|
|
tmp_path / ".gate.lock",
|
|
),
|
|
patch(
|
|
"diet_guard._sync_client.SYNC_TOKEN_FILE",
|
|
tmp_path / "sync_token",
|
|
),
|
|
patch(
|
|
"diet_guard._sync.SYNC_STATE_FILE",
|
|
tmp_path / "sync_state.json",
|
|
),
|
|
# Without this a test that syncs mints a uuid into the REAL
|
|
# ~/.local/share/diet_guard/.device_id, and this machine's live sync
|
|
# identity is decided by whichever test happened to run first.
|
|
patch(
|
|
"diet_guard._device.SYNC_DEVICE_ID_FILE",
|
|
tmp_path / ".device_id",
|
|
),
|
|
# `run_sync` reads this to decide whether to build a Firebase-primary
|
|
# mirror. On a developer machine the real file exists, so without this
|
|
# every sync test would sign in and push to the live database.
|
|
patch(
|
|
"diet_guard._sync_client.CONFIG_FILE",
|
|
tmp_path / "nonexistent-firebase.json",
|
|
),
|
|
# Every full tick ends by reading the phone's weigh-in through
|
|
# `wake_alarm._weight`, which is installed here and reads the live
|
|
# Firebase database. `test_sync_budget` asserts the call; nothing
|
|
# else may pay for it.
|
|
patch("diet_guard._sync.refresh_weight_from_phone", return_value=None),
|
|
# Logging a meal now publishes immediately (`_sync_events`), so every
|
|
# test that logs one would otherwise open a real connection through a
|
|
# path that has nothing to do with what it is asserting. Patched at the
|
|
# *call sites* rather than on `_sync_events`, so a test that imports
|
|
# `publish_after_log` directly still exercises the real helper.
|
|
patch("diet_guard._cli_log.publish_after_log_detached", return_value=None),
|
|
patch("diet_guard._cli_gate.publish_after_log", return_value=None),
|
|
patch("diet_guard._mcp.publish_after_log", return_value=None),
|
|
# The gate's pre-lock refresh is a *second*, independent network
|
|
# entry point (narrow peer-log pull, not the full tick). Without
|
|
# this every test that reaches `_should_lock` hits the real remote.
|
|
patch("diet_guard._cli_gate.pull_peer_logs", return_value=None),
|
|
# The catering credentials and its cached session cookie. Redirected
|
|
# for the same reason as `sync_token`: `_test_guard` raises on any
|
|
# write under the real ~/.config/diet_guard, and a test that logs in
|
|
# would otherwise clobber the live session. Deliberately NOT added to
|
|
# `test_state_redirect._REDIRECTED_CONSTANTS`, matching
|
|
# `SYNC_TOKEN_FILE` -- that check enforces a single naming module,
|
|
# which would forbid the CLI naming the path in its own setup text.
|
|
patch(
|
|
"diet_guard._kuchnia_config.KUCHNIA_CREDENTIALS_FILE",
|
|
tmp_path / "kuchnia_credentials",
|
|
),
|
|
patch(
|
|
"diet_guard._kuchnia_config.KUCHNIA_SESSION_FILE",
|
|
tmp_path / "kuchnia_session.json",
|
|
),
|
|
patch(
|
|
"diet_guard._kuchnia_credential_store.KUCHNIA_SYNCED_CREDENTIAL_FILE",
|
|
tmp_path / "kuchnia_synced_credential.json",
|
|
),
|
|
patch(
|
|
"diet_guard._kuchnia_config.KUCHNIA_LAST_IMPORT_FILE",
|
|
tmp_path / "kuchnia_last_import",
|
|
),
|
|
# The catering fetch is a third network entry point. Patched at each
|
|
# call site, so a test importing `refresh_delivery` directly still
|
|
# exercises the real helper.
|
|
patch("diet_guard._cli_kuchnia.refresh_delivery", return_value=([], None)),
|
|
# Logging a meal now warms the catering bank on a background
|
|
# thread. Without this every meal-logging test in the suite reaches
|
|
# the live panel.
|
|
patch(
|
|
"diet_guard._cli_log.refresh_delivery_once",
|
|
return_value=([], None),
|
|
),
|
|
# Logging a meal now warms the catering bank on a background
|
|
# thread. Without this every meal-logging test in the suite reaches
|
|
# the live panel.
|
|
patch(
|
|
"diet_guard._gatelock_delivery.refresh_delivery",
|
|
return_value=([], None),
|
|
),
|
|
]
|
|
with ExitStack() as stack:
|
|
for redirect in redirects:
|
|
stack.enter_context(redirect)
|
|
yield
|
|
|
|
|
|
def _nutrition(kcal: float = 100, grams: float = 100) -> Nutrition:
|
|
"""A simple reference nutrition for driving the gate form."""
|
|
return Nutrition(kcal, 10, 20, 5, grams, "food bank")
|