mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
94 lines
3.9 KiB
Python
94 lines
3.9 KiB
Python
"""Tests for timezone utilities."""
|
|
|
|
from datetime import datetime, timezone
|
|
|
|
|
|
from basic_memory.config import DatabaseBackend
|
|
from basic_memory.utils import ensure_timezone_aware
|
|
|
|
|
|
class TestEnsureTimezoneAware:
|
|
"""Tests for ensure_timezone_aware function."""
|
|
|
|
def test_already_timezone_aware_returns_unchanged(self):
|
|
"""Timezone-aware datetime should be returned unchanged."""
|
|
dt = datetime(2024, 1, 15, 12, 30, 0, tzinfo=timezone.utc)
|
|
result = ensure_timezone_aware(dt)
|
|
assert result == dt
|
|
assert result.tzinfo == timezone.utc
|
|
|
|
def test_naive_datetime_cloud_mode_true_interprets_as_utc(self):
|
|
"""In cloud mode, naive datetimes should be interpreted as UTC."""
|
|
naive_dt = datetime(2024, 1, 15, 12, 30, 0)
|
|
result = ensure_timezone_aware(naive_dt, cloud_mode=True)
|
|
|
|
# Should have UTC timezone
|
|
assert result.tzinfo == timezone.utc
|
|
# Time values should be unchanged (just tagged as UTC)
|
|
assert result.year == 2024
|
|
assert result.month == 1
|
|
assert result.day == 15
|
|
assert result.hour == 12
|
|
assert result.minute == 30
|
|
|
|
def test_naive_datetime_cloud_mode_false_interprets_as_local(self):
|
|
"""In local mode, naive datetimes should be interpreted as local time."""
|
|
naive_dt = datetime(2024, 1, 15, 12, 30, 0)
|
|
result = ensure_timezone_aware(naive_dt, cloud_mode=False)
|
|
|
|
# Should have some timezone info (local)
|
|
assert result.tzinfo is not None
|
|
# The datetime should be converted to local timezone
|
|
# We can't assert exact timezone as it depends on system
|
|
|
|
def test_cloud_mode_true_does_not_shift_time(self):
|
|
"""Cloud mode should use replace() not astimezone() - time values unchanged."""
|
|
naive_dt = datetime(2024, 6, 15, 18, 0, 0) # Summer time
|
|
result = ensure_timezone_aware(naive_dt, cloud_mode=True)
|
|
|
|
# Hour should remain 18, not be shifted by timezone offset
|
|
assert result.hour == 18
|
|
assert result.tzinfo == timezone.utc
|
|
|
|
def test_explicit_cloud_mode_skips_config_loading(self):
|
|
"""When cloud_mode is explicitly passed, config should not be loaded."""
|
|
# This test verifies we can call ensure_timezone_aware without
|
|
# triggering ConfigManager import when cloud_mode is explicit
|
|
naive_dt = datetime(2024, 1, 15, 12, 30, 0)
|
|
|
|
# Should work without any config setup
|
|
result_cloud = ensure_timezone_aware(naive_dt, cloud_mode=True)
|
|
assert result_cloud.tzinfo == timezone.utc
|
|
|
|
result_local = ensure_timezone_aware(naive_dt, cloud_mode=False)
|
|
assert result_local.tzinfo is not None
|
|
|
|
def test_none_cloud_mode_falls_back_to_database_backend(self, config_manager):
|
|
"""When cloud_mode is None, should infer from database backend."""
|
|
naive_dt = datetime(2024, 1, 15, 12, 30, 0)
|
|
# Use the real config file (via test fixtures) rather than mocking.
|
|
cfg = config_manager.config
|
|
cfg.database_backend = DatabaseBackend.POSTGRES
|
|
config_manager.save_config(cfg)
|
|
|
|
result = ensure_timezone_aware(naive_dt, cloud_mode=None)
|
|
|
|
# Should have used cloud mode (UTC)
|
|
assert result.tzinfo == timezone.utc
|
|
|
|
def test_asyncpg_naive_utc_scenario(self):
|
|
"""Simulate asyncpg returning naive datetime that's actually UTC.
|
|
|
|
asyncpg binary protocol returns timestamps in UTC but as naive datetimes.
|
|
In cloud mode, we interpret these as UTC rather than local time.
|
|
"""
|
|
# Simulate what asyncpg returns: a naive datetime that's actually UTC
|
|
asyncpg_result = datetime(2024, 1, 15, 18, 30, 0) # 6:30 PM UTC
|
|
|
|
# In cloud mode, interpret as UTC
|
|
cloud_result = ensure_timezone_aware(asyncpg_result, cloud_mode=True)
|
|
assert cloud_result == datetime(2024, 1, 15, 18, 30, 0, tzinfo=timezone.utc)
|
|
|
|
# The hour should remain 18, not shifted
|
|
assert cloud_result.hour == 18
|