migrace /remind na sqlite

This commit is contained in:
lachtan
2026-06-10 07:10:11 +02:00
parent be01fe7a07
commit b2d367e6a4
8 changed files with 1315 additions and 231 deletions

View File

@@ -0,0 +1,184 @@
import json
import os
import sqlite3
import sys
from pathlib import Path
import pytest
SCRIPTS = Path(__file__).parent.parent / "scripts"
sys.path.insert(0, str(SCRIPTS))
from db import get_db, init_db
import remind_edit
def _run(db_path, argv):
"""Run remind_edit main with a temporary DB path."""
original_db_path = remind_edit.DB_PATH
try:
remind_edit.DB_PATH = db_path
# Patch the module-level DB_PATH used by functions
sys.argv = ["remind_edit.py"] + argv
return remind_edit.main()
finally:
remind_edit.DB_PATH = original_db_path
def test_add_list(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
ret = _run(db_path, ["add", "--text", "drink water", "--cron", "0 9 * * *"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert data["added"]["text"] == "drink water"
assert len(data["added"]["cron"]) == 1
ret = _run(db_path, ["list"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert len(data["reminders"]) == 1
assert data["reminders"][0]["text"] == "drink water"
def test_add_at_validation(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
ret = _run(db_path, ["add", "--text", "x", "--at", "not-a-date"])
captured = capsys.readouterr()
assert ret == 1
assert "invalid --at datetime" in captured.err
def test_add_cron_validation(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
ret = _run(db_path, ["add", "--text", "x", "--cron", "bad"])
captured = capsys.readouterr()
assert ret == 1
assert "invalid cron expression" in captured.err
def test_remove(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
_run(db_path, ["add", "--text", "stretch", "--cron", "0 10 * * *"])
capsys.readouterr() # clear setup output
ret = _run(db_path, ["remove", "--keyword", "stretch"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert data["removed"]["text"] == "stretch"
ret = _run(db_path, ["list"])
captured = capsys.readouterr()
data = json.loads(captured.out)
assert len(data["reminders"]) == 0
def test_remove_no_match(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
ret = _run(db_path, ["remove", "--keyword", "nonexistent"])
captured = capsys.readouterr()
assert ret == 1
assert "no match" in captured.err
def test_remove_ambiguous(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
_run(db_path, ["add", "--text", "drink water", "--cron", "0 9 * * *"])
_run(db_path, ["add", "--text", "drink tea", "--cron", "0 10 * * *"])
ret = _run(db_path, ["remove", "--keyword", "drink"])
captured = capsys.readouterr()
assert ret == 1
assert "ambiguous" in captured.err
def test_edit_text(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
_run(db_path, ["add", "--text", "old text", "--cron", "0 9 * * *"])
capsys.readouterr() # clear setup output
ret = _run(db_path, ["edit", "--keyword", "old", "--text", "new text"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert data["edited"]["text"] == "new text"
def test_edit_replace_schedules(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
_run(db_path, ["add", "--text", "task", "--cron", "0 9 * * *"])
capsys.readouterr() # clear setup output
ret = _run(db_path, ["edit", "--keyword", "task", "--replace-schedules", "--at", "2026-06-15T10:00:00"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert len(data["edited"]["cron"]) == 0
assert len(data["edited"]["at"]) == 1
def test_enable_disable(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
_run(db_path, ["add", "--text", "toggle me", "--cron", "0 9 * * *"])
capsys.readouterr() # clear setup output
ret = _run(db_path, ["disable", "--keyword", "toggle"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert data["disabled"]["enabled"] == 0
ret = _run(db_path, ["enable", "--keyword", "toggle"])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert data["enabled"]["enabled"] == 1
def test_add_combined_schedules(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
ret = _run(db_path, [
"add", "--text", "multi",
"--at", "2026-06-15T10:00:00",
"--cron", "0 9 * * *",
"--random-times-per-day", "2",
"--random-window", "08:00-20:00",
])
captured = capsys.readouterr()
assert ret == 0
data = json.loads(captured.out)
assert len(data["added"]["at"]) == 1
assert len(data["added"]["cron"]) == 1
assert len(data["added"]["random"]) == 1
def test_add_random_validation(tmp_path, capsys):
db_path = tmp_path / "test.sqlite"
init_db(db_path)
ret = _run(db_path, [
"add", "--text", "bad",
"--random-times-per-day", "2",
"--random-window", "08:00-08:01",
])
captured = capsys.readouterr()
assert ret == 1
assert "window" in captured.err.lower() or "gap" in captured.err.lower()