plomlompom.com Git - plomtask/blob - tests/processes.py

   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
   7
   8
   9 class TestsSansDB(TestCase):
  10     """Module tests not requiring DB setup."""
  11
  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)
  17
  18
  19 class TestsWithDB(TestCaseWithDB):
  20     """Mdule tests not requiring DB setup."""
  21
  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):
  29             p_saved = Process(0)
  30         p_saved = Process(5)
  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)
  40
  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_)
  55
  56     def test_Process_all(self) -> None:
  57         """Test Process.all()."""
  58         p_1 = Process(None)
  59         p_1.save(self.db_conn)
  60         p_2 = Process(None)
  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)))
  64
  65
  66 class TestsWithServer(TestCaseWithServer):
  67     """Module tests against our HTTP server/handler (and database)."""
  68
  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)],
  90                          [retrieved.id_])
  91
  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)