feat: align tariff resets to calendar month

2026-04-28 15:25:46 +03:00
parent 338008bbd0
commit 258b2f4dec
9 changed files with 193 additions and 37 deletions
@@ -0,0 +1,20 @@
+import unittest
+from datetime import datetime, timezone
+
+from bot.utils.date_utils import month_start
+
+
+class DateUtilsTests(unittest.TestCase):
+    def test_month_start_normalizes_aware_datetime(self):
+        dt = datetime(2026, 4, 28, 15, 45, tzinfo=timezone.utc)
+
+        result = month_start(dt)
+
+        self.assertEqual(result, datetime(2026, 4, 1, 0, 0, tzinfo=timezone.utc))
+
+    def test_month_start_normalizes_naive_datetime_as_utc(self):
+        dt = datetime(2026, 12, 31, 23, 59)
+
+        result = month_start(dt)
+
+        self.assertEqual(result, datetime(2026, 12, 1, 0, 0, tzinfo=timezone.utc))