Add basic GET /calendar test.

[plomtask] / tests / test_days.py

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

+from threading import Thread
+from http.client import HTTPConnection

 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


@@ -97,3 +100,32 @@ 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_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()