1 """Test Todos module."""
2 from tests.utils import TestCaseWithDB, TestCaseWithServer
3 from plomtask.todos import Todo
4 from plomtask.days import Day
5 from plomtask.processes import Process
6 from plomtask.exceptions import NotFoundException
9 class TestsWithDB(TestCaseWithDB):
10 """Tests requiring DB, but not server setup."""
12 def test_Todo_by_id(self) -> None:
13 """Test creation and findability of Todos."""
14 day = Day('2024-01-01')
15 process = Process(None)
16 todo = Todo(None, process, False, day)
17 with self.assertRaises(NotFoundException):
18 todo.save(self.db_conn)
19 process.save_without_steps(self.db_conn)
20 todo.save(self.db_conn)
21 self.assertEqual(Todo.by_id(self.db_conn, 1), todo)
22 with self.assertRaises(NotFoundException):
23 self.assertEqual(Todo.by_id(self.db_conn, 0), todo)
24 with self.assertRaises(NotFoundException):
25 self.assertEqual(Todo.by_id(self.db_conn, 2), todo)
27 def test_Todo_by_date(self) -> None:
28 """Test findability of Todos by date."""
29 day1 = Day('2024-01-01')
30 day2 = Day('2024-01-02')
31 process = Process(None)
32 process.save_without_steps(self.db_conn)
33 todo1 = Todo(None, process, False, day1)
34 todo1.save(self.db_conn)
35 todo2 = Todo(None, process, False, day1)
36 todo2.save(self.db_conn)
37 self.assertEqual(Todo.by_date(self.db_conn, day1.date), [todo1, todo2])
38 self.assertEqual(Todo.by_date(self.db_conn, day2.date), [])
39 self.assertEqual(Todo.by_date(self.db_conn, 'foo'), [])
41 def test_Todo_singularity(self) -> None:
42 """Test pointers made for single object keep pointing to it."""
43 day = Day('2024-01-01')
44 day.save(self.db_conn)
45 process = Process(None)
46 process.save_without_steps(self.db_conn)
47 todo = Todo(None, process, False, day)
48 todo.save(self.db_conn)
49 retrieved_todo = Todo.by_id(self.db_conn, 1)
51 self.assertEqual(retrieved_todo.is_done, True)
52 retrieved_todo = Todo.by_date(self.db_conn, '2024-01-01')[0]
53 retrieved_todo.is_done = False
54 self.assertEqual(todo.is_done, False)
57 class TestsWithServer(TestCaseWithServer):
58 """Tests against our HTTP server/handler (and database)."""
60 def test_do_POST_todo(self) -> None:
61 """Test Todo posting of POST /day."""
62 form_data = {'title': '', 'description': '', 'effort': 1}
63 self.check_post(form_data, '/process?id=', 302, '/')
64 self.check_post(form_data, '/process?id=', 302, '/')
65 process1 = Process.by_id(self.db_conn, 1)
66 process2 = Process.by_id(self.db_conn, 2)
67 form_data = {'comment': ''}
68 self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
69 self.assertEqual(Todo.by_date(self.db_conn, '2024-01-01'), [])
70 form_data['new_todo'] = str(process1.id_)
71 self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
72 todos = Todo.by_date(self.db_conn, '2024-01-01')
73 self.assertEqual(1, len(todos))
75 self.assertEqual(todo1.id_, 1)
76 self.assertEqual(todo1.process.id_, process1.id_)
77 self.assertEqual(todo1.is_done, False)
78 form_data['new_todo'] = str(process2.id_)
79 self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
80 todos = Todo.by_date(self.db_conn, '2024-01-01')
82 self.assertEqual(todo1.id_, 2)
83 self.assertEqual(todo1.process.id_, process2.id_)
84 self.assertEqual(todo1.is_done, False)
86 def test_do_GET_todo(self) -> None:
87 """Test GET /todo response codes."""
88 form_data = {'title': '', 'description': '', 'effort': 1}
89 self.check_post(form_data, '/process?id=', 302, '/')
90 form_data = {'comment': '', 'new_todo': 1}
91 self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
92 self.check_get('/todo', 404)
93 self.check_get('/todo?id=', 404)
94 self.check_get('/todo?id=foo', 400)
95 self.check_get('/todo?id=0', 404)
96 self.check_get('/todo?id=1', 200)