Add POST /day tests.

[plomtask] / tests / test_days.py

diff --git a/tests/test_days.py b/tests/test_days.py
index be2719318140fa52e5c53613e854a727b76d1887..7daf7498214375914493cd1bab399326159772f0 100644 (file)
--- a/tests/test_days.py
+++ b/tests/test_days.py
@@ -1,7 +1,11 @@
 """Test days module."""
 from unittest import TestCase
 """Test days module."""
 from unittest import TestCase

+from threading import Thread
+from http.client import HTTPConnection
+from urllib.parse import urlencode

 from datetime import datetime
 from os import remove as remove_file
 from datetime import datetime
 from os import remove as remove_file

+from plomtask.http import TaskHandler, TaskServer

 from plomtask.days import Day
 from plomtask.misc import HandledException
 from plomtask.db import DatabaseFile, DatabaseConnection
 from plomtask.days import Day
 from plomtask.misc import HandledException
 from plomtask.db import DatabaseFile, DatabaseConnection


@@ -56,6 +60,10 @@ class TestsWithDB(TestCase):
         Day('2024-01-01').save(self.db_conn)
         self.assertEqual(Day('2024-01-01'),
                          Day.by_date(self.db_conn, '2024-01-01'))
         Day('2024-01-01').save(self.db_conn)
         self.assertEqual(Day('2024-01-01'),
                          Day.by_date(self.db_conn, '2024-01-01'))

+        self.assertEqual(None,
+                         Day.by_date(self.db_conn, '2024-01-02'))
+        self.assertEqual(Day('2024-01-02'),
+                         Day.by_date(self.db_conn, '2024-01-02', create=True))

 
     def test_Day_all(self):
         """Test Day.all(), especially in regards to date range filtering."""
 
     def test_Day_all(self):
         """Test Day.all(), especially in regards to date range filtering."""


@@ -93,3 +101,50 @@ class TestsWithDB(TestCase):
         d2.save(self.db_conn)
         day_retrieved = Day.by_date(self.db_conn, d2.date)
         self.assertEqual(day_retrieved.comment, 'foo')
         d2.save(self.db_conn)
         day_retrieved = Day.by_date(self.db_conn, d2.date)
         self.assertEqual(day_retrieved.comment, 'foo')

+
+
+class TestsWithServer(TestsWithDB):
+    """Tests against our HTTP server/handler (and database)."""
+
+    def setUp(self):
+        super().setUp()
+        self.httpd = TaskServer(self.db_file, ('localhost', 0), TaskHandler)
+        self.server_thread = Thread(target=self.httpd.serve_forever)
+        self.server_thread.daemon = True
+        self.server_thread.start()
+
+    def test_do_POST_day(self):
+        """Test POST /day and its effect on the database."""
+        conn = HTTPConnection(*self.httpd.server_address)
+        form_data = {'comment': 'foo'}
+        encoded_form_data = urlencode(form_data).encode('utf-8')
+        headers = {'Content-Type': 'application/x-www-form-urlencoded',
+                   'Content-Length': str(len(encoded_form_data))}
+        conn.request('POST', '/day?date=FOO',
+                     body=encoded_form_data, headers=headers)
+        self.assertEqual(conn.getresponse().status, 400)
+        self.assertEqual(Day.all(self.db_conn), [])
+        conn.request('POST', '/day?date=2024-01-01',
+                     body=encoded_form_data, headers=headers)
+        self.assertEqual(conn.getresponse().status, 302)
+        retrieved = Day.by_date(self.db_conn, '2024-01-01')
+        self.assertEqual(retrieved.comment, 'foo')
+        self.assertEqual(Day.all(self.db_conn), [retrieved])
+
+    def test_do_GET(self):
+        """Test /day and /calendar response codes."""
+        conn = HTTPConnection(*self.httpd.server_address)
+        conn.request('GET', '/day')
+        self.assertEqual(conn.getresponse().status, 200)
+        conn.request('GET', '/day?date=3000-01-01')
+        self.assertEqual(conn.getresponse().status, 200)
+        conn.request('GET', '/day?date=FOO')
+        self.assertEqual(conn.getresponse().status, 400)
+        conn.request('GET', '/calendar')
+        self.assertEqual(conn.getresponse().status, 200)
+
+    def tearDown(self):
+        self.httpd.shutdown()
+        self.httpd.server_close()
+        self.server_thread.join()
+        super().tearDown()