From: Philipp Spitzer <philipp@spitzer.priv.at>
Date: Tue, 26 Feb 2019 22:25:44 +0000 (+0100)
Subject: Add unittest for function first_of_month.
X-Git-Url: https://git.toastfreeware.priv.at/chrisu/seepark.git/commitdiff_plain/cd14da74ff0e9262da8f61669850406f532570bc

Add unittest for function first_of_month.
---

diff --git a/web/README b/web/README
index 6a6ee1f..0dd0ca6 100644
--- a/web/README
+++ b/web/README
@@ -40,3 +40,8 @@ https://gr33ndata.blogspot.com/2012/01/waitress-flask.html
 Generate documentation:
 cd doc
 SEEPARKINI=~/seewasser.ini make web
+
+---
+
+Execute unittests
+SEEPARKINI=~/seewasser.ini python3 -m unittest 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))