From: Christian Heller Date: Fri, 12 Apr 2024 19:50:19 +0000 (+0200) Subject: Add most basic Todo infrastructure. X-Git-Url: https://plomlompom.com/repos/?a=commitdiff_plain;h=a4ca74f81ae42abe27cf6dbab7ef18c850db72c2;p=plomtask Add most basic Todo infrastructure. --- diff --git a/plomtask/todos.py b/plomtask/todos.py new file mode 100644 index 0000000..1b6b740 --- /dev/null +++ b/plomtask/todos.py @@ -0,0 +1,46 @@ +"""Actionables.""" +from __future__ import annotations +from sqlite3 import Row +from plomtask.db import DatabaseConnection +from plomtask.days import Day +from plomtask.processes import Process +from plomtask.exceptions import NotFoundException + + +class Todo: + """Individual actionable.""" + + def __init__(self, id_: int | None, process: Process, + is_done: bool, day: Day) -> None: + self.id_ = id_ + self.process = process + self.is_done = is_done + self.day = day + + def __eq__(self, other: object) -> bool: + return isinstance(other, self.__class__) and self.id_ == other.id_ + + @classmethod + def from_table_row(cls, row: Row, db_conn: DatabaseConnection) -> Todo: + """Make Todo from database row.""" + return cls(id_=row[0], + process=Process.by_id(db_conn, row[1]), + is_done=row[2], + day=Day.by_date(db_conn, row[3])) + + @classmethod + def by_date(cls, db_conn: DatabaseConnection, date: str) -> list[Todo]: + """Collect all Todos for Day of date.""" + todos = [] + for row in db_conn.exec('SELECT * FROM todos WHERE day = ?', (date,)): + todos += [cls.from_table_row(row, db_conn)] + return todos + + def save(self, db_conn: DatabaseConnection) -> None: + """Write self to DB.""" + if self.process.id_ is None: + raise NotFoundException('Process of Todo without ID (not saved?)') + cursor = db_conn.exec('REPLACE INTO todos VALUES (?,?,?,?)', + (self.id_, self.process.id_, + self.is_done, self.day.date)) + self.id_ = cursor.lastrowid diff --git a/tests/todos.py b/tests/todos.py new file mode 100644 index 0000000..93b34d1 --- /dev/null +++ b/tests/todos.py @@ -0,0 +1,60 @@ +"""Test Todos module.""" +from tests.utils import TestCaseWithDB, TestCaseWithServer +from plomtask.todos import Todo +from plomtask.days import Day +from plomtask.processes import Process +from plomtask.exceptions import NotFoundException + + +class TestsWithDB(TestCaseWithDB): + """Tests not requiring DB setup.""" + + def test_Todo_by_date(self) -> None: + """Test creation and findability of Todos.""" + day1 = Day('2024-01-01') + day2 = Day('2024-01-02') + process1 = Process(None) + todo1 = Todo(None, process1, False, day1) + with self.assertRaises(NotFoundException): + todo1.save(self.db_conn) + process1.save_without_steps(self.db_conn) + todo1.save(self.db_conn) + todo2 = Todo(None, process1, False, day1) + todo2.save(self.db_conn) + with self.assertRaises(NotFoundException): + Todo.by_date(self.db_conn, day1.date), + day1.save(self.db_conn) + day2.save(self.db_conn) + self.assertEqual(Todo.by_date(self.db_conn, day1.date), [todo1, todo2]) + self.assertEqual(Todo.by_date(self.db_conn, day2.date), []) + self.assertEqual(Todo.by_date(self.db_conn, 'foo'), []) + + +class TestsWithServer(TestCaseWithServer): + """Tests against our HTTP server/handler (and database).""" + + def test_do_POST_todo(self) -> None: + """Test Todo posting of POST /day.""" + form_data = {'title': '', 'description': '', 'effort': 1} + self.check_post(form_data, '/process?id=', 302, '/') + self.check_post(form_data, '/process?id=', 302, '/') + process1 = Process.by_id(self.db_conn, 1) + process2 = Process.by_id(self.db_conn, 2) + form_data = {'comment': ''} + self.check_post(form_data, '/day?date=2024-01-01', 302, '/') + self.assertEqual(Todo.by_date(self.db_conn, '2024-01-01'), []) + form_data['new_todo'] = str(process1.id_) + self.check_post(form_data, '/day?date=2024-01-01', 302, '/') + todos = Todo.by_date(self.db_conn, '2024-01-01') + self.assertEqual(1, len(todos)) + todo1 = todos[0] + self.assertEqual(todo1.id_, 1) + self.assertEqual(todo1.process.id_, process1.id_) + self.assertEqual(todo1.is_done, False) + form_data['new_todo'] = str(process2.id_) + self.check_post(form_data, '/day?date=2024-01-01', 302, '/') + todos = Todo.by_date(self.db_conn, '2024-01-01') + todo1 = todos[1] + self.assertEqual(todo1.id_, 2) + self.assertEqual(todo1.process.id_, process2.id_) + self.assertEqual(todo1.is_done, False)