| from __future__ import annotations |
|
|
| import tempfile |
| import unittest |
| from datetime import datetime, timedelta |
| from pathlib import Path |
|
|
| from mr_reachy.medication import MedicationMemory, build_plan |
| from mr_reachy.reminders import confirm_due_dose, process_due_reminders |
| from mr_reachy.storage import HybridMedicationStore, LocalMedicationStore |
|
|
|
|
| def _store_with_due_plan(tmp: str) -> tuple[HybridMedicationStore, datetime]: |
| now = datetime(2026, 5, 28, 9, 5).astimezone() |
| plan = build_plan( |
| raw_instruction="Take metformin once a day for one day.", |
| medication_name="metformin", |
| frequency_per_day=1, |
| duration_days=1, |
| dose_times=["09:00"], |
| now=now, |
| ) |
| store = HybridMedicationStore(LocalMedicationStore(Path(tmp) / "memory.json"), auto_sync=False) |
| store.save(MedicationMemory(plans=[plan])) |
| return store, now |
|
|
|
|
| class ReminderTest(unittest.TestCase): |
| def test_due_dose_triggers_reminder(self) -> None: |
| with tempfile.TemporaryDirectory() as tmp: |
| store, now = _store_with_due_plan(tmp) |
| replies = [] |
|
|
| changed = process_due_reminders(store=store, notify=replies.append, now=now) |
| memory = store.load() |
|
|
| self.assertTrue(changed) |
| self.assertEqual(len(replies), 1) |
| self.assertIn("metformin", replies[0].speech) |
| self.assertEqual(memory.plans[0].doses[0].reminder_count, 1) |
|
|
| def test_confirmation_marks_due_dose_complete(self) -> None: |
| with tempfile.TemporaryDirectory() as tmp: |
| store, now = _store_with_due_plan(tmp) |
|
|
| confirmed, message = confirm_due_dose(store, now=now) |
| memory = store.load() |
|
|
| self.assertTrue(confirmed) |
| self.assertIn("metformin", message) |
| self.assertEqual(memory.plans[0].doses[0].status, "confirmed") |
|
|
| def test_missed_dose_after_three_reminders(self) -> None: |
| with tempfile.TemporaryDirectory() as tmp: |
| store, now = _store_with_due_plan(tmp) |
| memory = store.load() |
| dose = memory.plans[0].doses[0] |
| dose.reminder_count = 3 |
| dose.last_reminded_at = (now - timedelta(minutes=10)).isoformat() |
| store.save(memory) |
| replies = [] |
|
|
| changed = process_due_reminders(store=store, notify=replies.append, now=now) |
| updated = store.load() |
|
|
| self.assertTrue(changed) |
| self.assertEqual(updated.plans[0].doses[0].status, "missed") |
| self.assertEqual(replies[0].emotion, "sad") |
|
|
|
|
| if __name__ == "__main__": |
| unittest.main() |
|
|
