66 lines
2.7 KiB
Python
66 lines
2.7 KiB
Python
|
"""Exercise coach.py: preferences flip between like and dislike, the log feeds hours since."""
|
||
|
|
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
import json
|
||
|
|
import subprocess
|
||
|
|
import sys
|
||
|
|
import tempfile
|
||
|
|
import unittest
|
||
|
|
from pathlib import Path
|
||
|
|
|
||
|
|
REPO_ROOT = Path(__file__).resolve().parents[1]
|
||
|
|
SCRIPT = REPO_ROOT / "skills" / "kvb-coach" / "scripts" / "coach.py"
|
||
|
|
|
||
|
|
|
||
|
|
class TestCoach(unittest.TestCase):
|
||
|
|
def setUp(self):
|
||
|
|
self.tmp = tempfile.TemporaryDirectory()
|
||
|
|
self.addCleanup(self.tmp.cleanup)
|
||
|
|
self.profile = Path(self.tmp.name) / "state" / "coach.json"
|
||
|
|
|
||
|
|
def _run(self, *args: str) -> subprocess.CompletedProcess:
|
||
|
|
return subprocess.run(
|
||
|
|
[sys.executable, str(SCRIPT), "--profile", str(self.profile), *args],
|
||
|
|
check=False,
|
||
|
|
capture_output=True,
|
||
|
|
text=True,
|
||
|
|
)
|
||
|
|
|
||
|
|
def _show(self, now: str) -> dict:
|
||
|
|
result = self._run("show", "--now", now, "--json")
|
||
|
|
self.assertEqual(result.returncode, 0, result.stderr)
|
||
|
|
return {row["area"]: row for row in json.loads(result.stdout)}
|
||
|
|
|
||
|
|
def test_a_dislike_replaces_a_like_and_forget_drops_it(self):
|
||
|
|
self._run("like", "body", "Laufen")
|
||
|
|
self._run("like", "body", "Laufen")
|
||
|
|
self._run("like", "body", "Yoga")
|
||
|
|
self._run("dislike", "body", "Laufen")
|
||
|
|
body = self._show("2026-11-03T18:00:00+00:00")["body"]
|
||
|
|
self.assertEqual(body["likes"], ["Yoga"])
|
||
|
|
self.assertEqual(body["dislikes"], ["Laufen"])
|
||
|
|
self._run("forget", "body", "Laufen")
|
||
|
|
self.assertEqual(self._show("2026-11-03T18:00:00+00:00")["body"]["dislikes"], [])
|
||
|
|
|
||
|
|
def test_the_latest_done_per_area_gives_the_hours_since(self):
|
||
|
|
self._run("done", "food", "Frühstück", "--at", "2026-11-03T08:00:00+00:00")
|
||
|
|
self._run("done", "food", "Mittagessen", "--at", "2026-11-03T12:30:00+00:00")
|
||
|
|
self._run("done", "sleep", "Nacht", "--at", "2026-11-02T23:00:00+00:00")
|
||
|
|
rows = self._show("2026-11-03T18:00:00+00:00")
|
||
|
|
self.assertEqual((rows["food"]["last"], rows["food"]["hours_ago"]), ("Mittagessen", 5.5))
|
||
|
|
self.assertEqual(rows["sleep"]["hours_ago"], 19.0)
|
||
|
|
self.assertIsNone(rows["psyche"]["last"])
|
||
|
|
|
||
|
|
def test_invalid_input_exits_2_and_keeps_the_profile(self):
|
||
|
|
self.assertEqual(self._run("like", "work", "x").returncode, 2)
|
||
|
|
self.assertEqual(self._run("done", "body", "x", "--at", "gestern").returncode, 2)
|
||
|
|
self.assertFalse(self.profile.exists())
|
||
|
|
self.profile.parent.mkdir(parents=True)
|
||
|
|
self.profile.write_text("{broken", encoding="utf-8")
|
||
|
|
self.assertEqual(self._run("show").returncode, 2)
|
||
|
|
self.assertEqual(self.profile.read_text(encoding="utf-8"), "{broken")
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
unittest.main()
|