21 lines
612 B
Python
21 lines
612 B
Python
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))
|