1 """Test Processes module."""
2 from unittest import TestCase
3 from urllib.parse import urlencode
4 from tests.utils import TestCaseWithDB, TestCaseWithServer
5 from plomtask.processes import Process
6 from plomtask.exceptions import NotFoundException, BadFormatException
9 class TestsSansDB(TestCase):
10 """Module tests not requiring DB setup."""
12 def test_Process_versioned_defaults(self) -> None:
13 """Test defaults of Process' VersionedAttributes."""
14 self.assertEqual(Process(None).title.newest, 'UNNAMED')
15 self.assertEqual(Process(None).description.newest, '')
16 self.assertEqual(Process(None).effort.newest, 1.0)
19 class TestsWithDB(TestCaseWithDB):
20 """Mdule tests not requiring DB setup."""
22 def test_Process_save(self) -> None:
23 """Test Process.save()."""
24 p_saved = Process(None)
25 p_saved.save(self.db_conn)
26 self.assertEqual(p_saved.id_,
27 Process.by_id(self.db_conn, 1, create=False).id_)
28 with self.assertRaises(BadFormatException):
31 p_saved.save(self.db_conn)
32 self.assertEqual(p_saved.id_,
33 Process.by_id(self.db_conn, 5, create=False).id_)
34 p_saved.title.set('named')
35 p_loaded = Process.by_id(self.db_conn, p_saved.id_)
36 self.assertNotEqual(p_saved.title.history, p_loaded.title.history)
37 p_saved.save(self.db_conn)
38 p_loaded = Process.by_id(self.db_conn, p_saved.id_)
39 self.assertEqual(p_saved.title.history, p_loaded.title.history)
41 def test_Process_by_id(self) -> None:
42 """Test Process.by_id()."""
43 with self.assertRaises(NotFoundException):
44 Process.by_id(self.db_conn, None, create=False)
45 with self.assertRaises(NotFoundException):
46 Process.by_id(self.db_conn, 0, create=False)
47 with self.assertRaises(NotFoundException):
48 Process.by_id(self.db_conn, 1, create=False)
49 self.assertNotEqual(Process(1).id_,
50 Process.by_id(self.db_conn, None, create=True).id_)
51 self.assertEqual(Process(1).id_,
52 Process.by_id(self.db_conn, 1, create=True).id_)
53 self.assertEqual(Process(2).id_,
54 Process.by_id(self.db_conn, 2, create=True).id_)
56 def test_Process_all(self) -> None:
57 """Test Process.all()."""
59 p_1.save(self.db_conn)
61 p_2.save(self.db_conn)
62 self.assertEqual({p_1.id_, p_2.id_},
63 set(p.id_ for p in Process.all(self.db_conn)))
66 class TestsWithServer(TestCaseWithServer):
67 """Module tests against our HTTP server/handler (and database)."""
69 def test_do_POST_process(self) -> None:
70 """Test POST /process and its effect on the database."""
71 def post_data_to_expect(form_data: dict[str, object],
72 to_: str, expect: int) -> None:
73 encoded_form_data = urlencode(form_data).encode('utf-8')
74 headers = {'Content-Type': 'application/x-www-form-urlencoded',
75 'Content-Length': str(len(encoded_form_data))}
76 self.conn.request('POST', to_,
77 body=encoded_form_data, headers=headers)
78 self.assertEqual(self.conn.getresponse().status, expect)
79 form_data = {'title': 'foo', 'description': 'foo', 'effort': 1.0}
80 post_data_to_expect(form_data, '/process?id=FOO', 400)
81 form_data['effort'] = 'foo'
82 post_data_to_expect(form_data, '/process?id=', 400)
83 form_data['effort'] = None
84 post_data_to_expect(form_data, '/process?id=', 400)
85 form_data = {'title': None, 'description': 1, 'effort': 1.0}
86 post_data_to_expect(form_data, '/process?id=', 302)
87 retrieved = Process.by_id(self.db_conn, 1)
88 self.assertEqual(retrieved.title.newest, 'None')
89 self.assertEqual([p.id_ for p in Process.all(self.db_conn)],
92 def test_do_GET(self) -> None:
93 """Test /process and /processes response codes."""
94 self.conn.request('GET', '/process')
95 self.assertEqual(self.conn.getresponse().status, 200)
96 self.conn.request('GET', '/process?id=')
97 self.assertEqual(self.conn.getresponse().status, 200)
98 self.conn.request('GET', '/process?id=0')
99 self.assertEqual(self.conn.getresponse().status, 400)
100 self.conn.request('GET', '/process?id=FOO')
101 self.assertEqual(self.conn.getresponse().status, 400)
102 self.conn.request('GET', '/processes')
103 self.assertEqual(self.conn.getresponse().status, 200)