home · contact · privacy
6ef8b091396b9d1047a1122ce8a446a7c995c14d
[plomtask] / tests / todos.py
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, BadFormatException,
7                                  HandledException)
8
9
10 class TestsWithDB(TestCaseWithDB):
11     """Tests requiring DB, but not server setup."""
12
13     def test_Todo_by_id(self) -> None:
14         """Test creation and findability of Todos."""
15         day = Day('2024-01-01')
16         process = Process(None)
17         todo = Todo(None, process, False, day)
18         with self.assertRaises(NotFoundException):
19             todo.save(self.db_conn)
20         process.save_without_steps(self.db_conn)
21         todo.save(self.db_conn)
22         self.assertEqual(Todo.by_id(self.db_conn, 1), todo)
23         with self.assertRaises(NotFoundException):
24             self.assertEqual(Todo.by_id(self.db_conn, 0), todo)
25         with self.assertRaises(NotFoundException):
26             self.assertEqual(Todo.by_id(self.db_conn, 2), todo)
27
28     def test_Todo_by_date(self) -> None:
29         """Test findability of Todos by date."""
30         day1 = Day('2024-01-01')
31         day2 = Day('2024-01-02')
32         process = Process(None)
33         process.save_without_steps(self.db_conn)
34         todo1 = Todo(None, process, False, day1)
35         todo1.save(self.db_conn)
36         todo2 = Todo(None, process, False, day1)
37         todo2.save(self.db_conn)
38         self.assertEqual(Todo.by_date(self.db_conn, day1.date), [todo1, todo2])
39         self.assertEqual(Todo.by_date(self.db_conn, day2.date), [])
40         self.assertEqual(Todo.by_date(self.db_conn, 'foo'), [])
41
42     def test_Todo_children(self) -> None:
43         """Test Todo.children relations."""
44         day = Day('2024-01-01')
45         process = Process(None)
46         process.save_without_steps(self.db_conn)
47         todo_1 = Todo(None, process, False, day)
48         todo_2 = Todo(None, process, False, day)
49         with self.assertRaises(HandledException):
50             todo_1.add_child(todo_2)
51         todo_1.save(self.db_conn)
52         with self.assertRaises(HandledException):
53             todo_1.add_child(todo_2)
54         todo_2.save(self.db_conn)
55         todo_1.add_child(todo_2)
56         todo_1.save(self.db_conn)
57         todo_retrieved = Todo.by_id(self.db_conn, todo_1.id_)
58         self.assertEqual(todo_retrieved.children, [todo_2])
59         with self.assertRaises(BadFormatException):
60             todo_2.add_child(todo_1)
61
62     def test_Todo_singularity(self) -> None:
63         """Test pointers made for single object keep pointing to it."""
64         day = Day('2024-01-01')
65         day.save(self.db_conn)
66         process = Process(None)
67         process.save_without_steps(self.db_conn)
68         todo = Todo(None, process, False, day)
69         todo.save(self.db_conn)
70         retrieved_todo = Todo.by_id(self.db_conn, 1)
71         todo.is_done = True
72         self.assertEqual(retrieved_todo.is_done, True)
73         retrieved_todo = Todo.by_date(self.db_conn, '2024-01-01')[0]
74         retrieved_todo.is_done = False
75         self.assertEqual(todo.is_done, False)
76
77
78 class TestsWithServer(TestCaseWithServer):
79     """Tests against our HTTP server/handler (and database)."""
80
81     def test_do_POST_day(self) -> None:
82         """Test Todo posting of POST /day."""
83         form_data = {'title': '', 'description': '', 'effort': 1}
84         self.check_post(form_data, '/process?id=', 302, '/')
85         self.check_post(form_data, '/process?id=', 302, '/')
86         process1 = Process.by_id(self.db_conn, 1)
87         process2 = Process.by_id(self.db_conn, 2)
88         form_data = {'comment': ''}
89         self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
90         self.assertEqual(Todo.by_date(self.db_conn, '2024-01-01'), [])
91         form_data['new_todo'] = str(process1.id_)
92         self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
93         todos = Todo.by_date(self.db_conn, '2024-01-01')
94         self.assertEqual(1, len(todos))
95         todo1 = todos[0]
96         self.assertEqual(todo1.id_, 1)
97         self.assertEqual(todo1.process.id_, process1.id_)
98         self.assertEqual(todo1.is_done, False)
99         form_data['new_todo'] = str(process2.id_)
100         self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
101         todos = Todo.by_date(self.db_conn, '2024-01-01')
102         todo1 = todos[1]
103         self.assertEqual(todo1.id_, 2)
104         self.assertEqual(todo1.process.id_, process2.id_)
105         self.assertEqual(todo1.is_done, False)
106
107     def test_do_POST_todo(self) -> None:
108         """Test POST /todo."""
109         form_data = {'title': '', 'description': '', 'effort': 1}
110         self.check_post(form_data, '/process', 302, '/')
111         form_data = {'comment': '', 'new_todo': 1}
112         self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
113         form_data = {}
114         self.check_post(form_data, '/todo=', 404)
115         self.check_post(form_data, '/todo?id=', 404)
116         self.check_post(form_data, '/todo?id=FOO', 400)
117         self.check_post(form_data, '/todo?id=0', 404)
118         self.check_post(form_data, '/todo?id=1', 302, '/')
119         todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0]
120         self.assertEqual(todo1.children, [])
121         self.assertEqual(todo1.parents, [])
122         self.assertEqual(todo1.is_done, False)
123         form_data = {'done': ''}
124         self.check_post(form_data, '/todo?id=1', 302, '/')
125         self.db_conn.cached_todos = {}
126         todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0]
127         self.assertEqual(todo1.is_done, True)
128         form_data = {'adopt': 'foo'}
129         self.check_post(form_data, '/todo?id=1', 400)
130         form_data = {'adopt': 1}
131         self.check_post(form_data, '/todo?id=1', 400)
132         form_data = {'adopt': 2}
133         self.check_post(form_data, '/todo?id=1', 404)
134         form_data = {'comment': '', 'new_todo': 1}
135         self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
136         form_data = {'adopt': 2}
137         self.check_post(form_data, '/todo?id=1', 302, '/')
138         self.db_conn.cached_todos = {}
139         todo1 = Todo.by_date(self.db_conn, '2024-01-01')[0]
140         todo2 = Todo.by_date(self.db_conn, '2024-01-01')[1]
141         self.assertEqual(todo1.children, [todo2])
142         self.assertEqual(todo1.parents, [])
143         self.assertEqual(todo2.children, [])
144         self.assertEqual(todo2.parents, [todo1])
145         self.check_post(form_data, '/todo?id=1', 400, '/')
146
147     def test_do_GET_todo(self) -> None:
148         """Test GET /todo response codes."""
149         form_data = {'title': '', 'description': '', 'effort': 1}
150         self.check_post(form_data, '/process?id=', 302, '/')
151         form_data = {'comment': '', 'new_todo': 1}
152         self.check_post(form_data, '/day?date=2024-01-01', 302, '/')
153         self.check_get('/todo', 404)
154         self.check_get('/todo?id=', 404)
155         self.check_get('/todo?id=foo', 400)
156         self.check_get('/todo?id=0', 404)
157         self.check_get('/todo?id=1', 200)