plomlompom.com Git - plomtask/blob - 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
   7
   8
   9 class TestsWithDB(TestCaseWithDB):
  10     """Tests requiring DB, but not server setup."""
  11
  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)
  26
  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'), [])
  40
  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)
  50         todo.is_done = True
  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)
  55
  56
  57 class TestsWithServer(TestCaseWithServer):
  58     """Tests against our HTTP server/handler (and database)."""
  59
  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))
  74         todo1 = todos[0]
  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')
  81         todo1 = todos[1]
  82         self.assertEqual(todo1.id_, 2)
  83         self.assertEqual(todo1.process.id_, process2.id_)
  84         self.assertEqual(todo1.is_done, False)
  85
  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)