from plomtask.todos import Todo
from plomtask.days import Day
from plomtask.processes import Process
-from plomtask.exceptions import NotFoundException
+from plomtask.exceptions import (NotFoundException, BadFormatException,
+ HandledException)
class TestsWithDB(TestCaseWithDB):
self.assertEqual(Todo.by_date(self.db_conn, day2.date), [])
self.assertEqual(Todo.by_date(self.db_conn, 'foo'), [])
+ def test_Todo_children(self) -> None:
+ """Test Todo.children relations."""
+ day = Day('2024-01-01')
+ process = Process(None)
+ process.save_without_steps(self.db_conn)
+ todo_1 = Todo(None, process, False, day)
+ todo_2 = Todo(None, process, False, day)
+ with self.assertRaises(HandledException):
+ todo_1.add_child(todo_2)
+ todo_1.save(self.db_conn)
+ with self.assertRaises(HandledException):
+ todo_1.add_child(todo_2)
+ todo_2.save(self.db_conn)
+ todo_1.add_child(todo_2)
+ todo_1.save(self.db_conn)
+ todo_retrieved = Todo.by_id(self.db_conn, todo_1.id_)
+ self.assertEqual(todo_retrieved.children, [todo_2])
+ with self.assertRaises(BadFormatException):
+ todo_2.add_child(todo_1)
+
def test_Todo_singularity(self) -> None:
"""Test pointers made for single object keep pointing to it."""
day = Day('2024-01-01')
class TestsWithServer(TestCaseWithServer):
"""Tests against our HTTP server/handler (and database)."""
- def test_do_POST_todo(self) -> None:
+ def test_do_POST_day(self) -> None:
"""Test Todo posting of POST /day."""
form_data = {'title': '', 'description': '', 'effort': 1}
self.check_post(form_data, '/process?id=', 302, '/')
self.assertEqual(todo1.process.id_, process2.id_)
self.assertEqual(todo1.is_done, False)
+ def test_do_POST_todo(self) -> None:
+ """Test POST /todo."""
+ form_data = {'title': '', 'description': '', 'effort': 1}
+ self.check_post(form_data, '/process', 302, '/')
+ form_data = {'comment': '', 'new_todo': 1}
+ self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
+ form_data = {}
+ self.check_post(form_data, '/todo=', 404)
+ self.check_post(form_data, '/todo?id=', 404)
+ self.check_post(form_data, '/todo?id=FOO', 400)
+ self.check_post(form_data, '/todo?id=0', 404)
+ self.check_post(form_data, '/todo?id=1', 302, '/')
+ todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0]
+ self.assertEqual(todo1.children, [])
+ self.assertEqual(todo1.parents, [])
+ form_data = {'adopt': 'foo'}
+ self.check_post(form_data, '/todo?id=1', 400)
+ form_data = {'adopt': 1}
+ self.check_post(form_data, '/todo?id=1', 400)
+ form_data = {'adopt': 2}
+ self.check_post(form_data, '/todo?id=1', 404)
+ form_data = {'comment': '', 'new_todo': 1}
+ self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
+ form_data = {'adopt': 2}
+ self.check_post(form_data, '/todo?id=1', 302, '/')
+ self.db_conn.cached_todos = {}
+ todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0]
+ todo2 = Todo.by_date(self.db_conn, '2024-01-01')[1]
+ self.assertEqual(todo1.children, [todo2])
+ self.assertEqual(todo1.parents, [])
+ self.assertEqual(todo2.children, [])
+ self.assertEqual(todo2.parents, [todo1])
+ self.check_post(form_data, '/todo?id=1', 400, '/')
+
def test_do_GET_todo(self) -> None:
"""Test GET /todo response codes."""
form_data = {'title': '', 'description': '', 'effort': 1}