"""Tests for RunnerUpVerificationMixin in _runnerup_verification.py.""" from __future__ import annotations import json from typing import TYPE_CHECKING, Any from unittest.mock import MagicMock, patch from screen_locker.tests.conftest import create_locker if TYPE_CHECKING: from pathlib import Path # Minimal valid TCX XML for a 40-minute, 6-km run. _TCX_RUNNING = """\ 2400.0 6000.0 """ # TCX with an unrecognised sport tag (not in RUNNERUP_ACCEPTED_SPORTS). _TCX_GYM = """\ 3600.0 0.0 """ # Two laps that together make a valid run. _TCX_MULTI_LAP = """\ 1200.0 3000.0 1200.0 3000.0 """ # --------------------------------------------------------------------------- # Helpers # --------------------------------------------------------------------------- def _write_tcx(tmp_path: Path, content: str, name: str = "activity.tcx") -> str: """Write TCX content to a temp file and return the path string.""" p = tmp_path / name p.write_text(content) return str(p) # --------------------------------------------------------------------------- # _validate_runnerup_data # --------------------------------------------------------------------------- class TestScanAndFillWeekRunnerup: """Tests for _scan_and_fill_week_runnerup (lines 186-248).""" def test_returns_zero_when_no_phone( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """No ADB device → 0 filled.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("{}") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=False)) assert locker._scan_and_fill_week_runnerup(log_file) == 0 def test_returns_zero_when_all_days_already_logged( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """All days this week already have counted workouts → 0 new fills.""" from datetime import date, timedelta locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" # Fill Mon-today with phone_verified entries. today = date.today() monday = today - timedelta(days=today.weekday()) logs: dict[str, Any] = {} cur = monday while cur <= today: logs[cur.strftime("%Y-%m-%d")] = { "workout_data": {"type": "phone_verified"} } cur += timedelta(days=1) log_file.write_text(json.dumps(logs)) object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=[]) ) assert locker._scan_and_fill_week_runnerup(log_file) == 0 def test_fills_gap_for_unlogged_day( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """Gap in log + exports found + validated → entry written, count > 0.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("{}") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=["/sdcard/run.tcx"]), ) object.__setattr__( locker, "_pull_and_parse_tcx", MagicMock( return_value={"sport": 0, "duration_seconds": 2400, "distance_m": 6000} ), ) object.__setattr__( locker, "_validate_runnerup_data", MagicMock(return_value=("verified", "Running: 6.0 km in 40 min")), ) with patch( "screen_locker._runnerup_verification.compute_entry_hmac", return_value="sig", ): result = locker._scan_and_fill_week_runnerup(log_file) assert result > 0 logs = json.loads(log_file.read_text()) # At least one date should have been filled. types = [ v.get("workout_data", {}).get("type") for v in logs.values() if isinstance(v, dict) ] assert "runnerup_verified" in types def test_skips_date_when_no_exports( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """No exports for a date → date skipped, count stays 0.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("{}") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=[]), ) assert locker._scan_and_fill_week_runnerup(log_file) == 0 def test_skips_unreadable_tcx( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """_pull_and_parse_tcx returns None → remote skipped, not filled.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("{}") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=["/sdcard/run.tcx"]), ) object.__setattr__(locker, "_pull_and_parse_tcx", MagicMock(return_value=None)) assert locker._scan_and_fill_week_runnerup(log_file) == 0 def test_skips_not_verified_export( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """_validate_runnerup_data returns not-verified → date not filled.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("{}") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=["/sdcard/run.tcx"]), ) object.__setattr__( locker, "_pull_and_parse_tcx", MagicMock( return_value={"sport": 0, "duration_seconds": 60, "distance_m": 6000} ), ) assert locker._scan_and_fill_week_runnerup(log_file) == 0 def test_returns_zero_on_write_error( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """OSError writing log after fill → returns 0 (lines 241-246).""" locker = create_locker(mock_tk, tmp_path) # Can't patch PosixPath.open (read-only slot), so wrap it in a # tiny class that delegates reads but raises on writes. real_log = tmp_path / "log.json" real_log.write_text("{}") class _FailWrite: def open(self, mode: str = "r", **kw): if mode == "w": msg = "disk full" raise OSError(msg) return real_log.open(mode, **kw) fail_log = _FailWrite() object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=["/sdcard/run.tcx"]), ) object.__setattr__( locker, "_pull_and_parse_tcx", MagicMock( return_value={"sport": 0, "duration_seconds": 2400, "distance_m": 6000} ), ) object.__setattr__( locker, "_validate_runnerup_data", MagicMock(return_value=("verified", "ok")), ) with patch( "screen_locker._runnerup_verification.compute_entry_hmac", return_value="sig", ): result = locker._scan_and_fill_week_runnerup(fail_log) assert result == 0 def test_handles_corrupt_log_file( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """Corrupt log JSON → starts with empty dict, still works.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("not-json") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=[]), ) # Should not raise; returns 0 (no exports found). assert locker._scan_and_fill_week_runnerup(log_file) == 0 def test_hmac_none_still_fills_entry( self, mock_tk: MagicMock, mock_sys_exit: MagicMock, tmp_path: Path, ) -> None: """HMAC key absent (compute_entry_hmac returns None) → entry still written.""" locker = create_locker(mock_tk, tmp_path) log_file = tmp_path / "log.json" log_file.write_text("{}") object.__setattr__(locker, "_has_adb_device", MagicMock(return_value=True)) object.__setattr__( locker, "_find_runnerup_exports_for_date", MagicMock(return_value=["/sdcard/run.tcx"]), ) object.__setattr__( locker, "_pull_and_parse_tcx", MagicMock( return_value={"sport": 0, "duration_seconds": 2400, "distance_m": 6000} ), ) object.__setattr__( locker, "_validate_runnerup_data", MagicMock(return_value=("verified", "Running: 6.0 km in 40 min")), ) with patch( "screen_locker._runnerup_verification.compute_entry_hmac", return_value=None, ): result = locker._scan_and_fill_week_runnerup(log_file) assert result > 0 # No "hmac" key when signature is None. logs = json.loads(log_file.read_text()) for entry in logs.values(): assert "hmac" not in entry # --------------------------------------------------------------------------- # _find_runnerup_package # ---------------------------------------------------------------------------