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):
- """Tests not requiring DB setup."""
+ """Tests requiring DB, but not server setup."""
- def test_Todo_by_date(self) -> None:
+ def test_Todo_by_id(self) -> None:
"""Test creation and findability of Todos."""
+ day = Day('2024-01-01')
+ process = Process(None)
+ todo = Todo(None, process, False, day)
+ with self.assertRaises(NotFoundException):
+ todo.save(self.db_conn)
+ process.save_without_steps(self.db_conn)
+ todo.save(self.db_conn)
+ self.assertEqual(Todo.by_id(self.db_conn, 1), todo)
+ with self.assertRaises(NotFoundException):
+ self.assertEqual(Todo.by_id(self.db_conn, 0), todo)
+ with self.assertRaises(NotFoundException):
+ self.assertEqual(Todo.by_id(self.db_conn, 2), todo)
+
+ def test_Todo_by_date(self) -> None:
+ """Test findability of Todos by date."""
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)
+ process = Process(None)
+ process.save_without_steps(self.db_conn)
+ todo1 = Todo(None, process, False, day1)
todo1.save(self.db_conn)
- todo2 = Todo(None, process1, False, day1)
+ todo2 = Todo(None, process, 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'), [])
+ 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')
+ day.save(self.db_conn)
+ process = Process(None)
+ process.save_without_steps(self.db_conn)
+ todo = Todo(None, process, False, day)
+ todo.save(self.db_conn)
+ retrieved_todo = Todo.by_id(self.db_conn, 1)
+ todo.is_done = True
+ self.assertEqual(retrieved_todo.is_done, True)
+ retrieved_todo = Todo.by_date(self.db_conn, '2024-01-01')[0]
+ retrieved_todo.is_done = False
+ self.assertEqual(todo.is_done, False)
+
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.id_, 2)
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, [])
+ 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.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}
+ self.check_post(form_data, '/process?id=', 302, '/')
+ form_data = {'comment': '', 'new_todo': 1}
+ self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
+ self.check_get('/todo', 404)
+ self.check_get('/todo?id=', 404)
+ self.check_get('/todo?id=foo', 400)
+ self.check_get('/todo?id=0', 404)
+ self.check_get('/todo?id=1', 200)