X-Git-Url: https://git.toastfreeware.priv.at/chrisu/seepark.git/blobdiff_plain/a6944dd859cdb37e57ece09281a187097585f65b..cd14da74ff0e9262da8f61669850406f532570bc:/web/seepark_web_tests.py

diff --git a/web/seepark_web_tests.py b/web/seepark_web_tests.py
new file mode 100644
index 0000000..aedc7c7
--- /dev/null
+++ b/web/seepark_web_tests.py
@@ -0,0 +1,17 @@
+import datetime
+import unittest
+import seepark_web
+
+
+class SeeparkWebTestCase(unittest.TestCase):
+    def test_first_of_month(self):
+        for d in [1, 8, 25, 28]:
+            date_a = datetime.date(2019, 2, d)
+            date_b = seepark_web.first_of_month(date_a, 1)
+            self.assertEqual(date_b, datetime.date(2019, 3, 1))
+
+            date_b = seepark_web.first_of_month(date_a, -1)
+            self.assertEqual(date_b, datetime.date(2019, 1, 1))
+
+            date_b = seepark_web.first_of_month(date_a, 0)
+            self.assertEqual(date_b, datetime.date(2019, 2, 1))