Replace core database utcnow defaults (#1457)

Co-authored-by: ghreprimand <203024559+ghreprimand@users.noreply.github.com>
This commit is contained in:
ghreprimand
2026-06-03 20:50:19 -05:00
committed by GitHub
parent 6e66e69451
commit 82fcec6bb6
2 changed files with 53 additions and 14 deletions
+33
View File
@@ -0,0 +1,33 @@
import types
import pytest
sqlalchemy = pytest.importorskip("sqlalchemy")
if not isinstance(sqlalchemy, types.ModuleType):
pytest.skip("sqlalchemy is stubbed in this environment", allow_module_level=True)
from core.database import ChatMessage, DocumentVersion, Session, TaskRun, UserToolData, utcnow_naive
def test_utcnow_naive_returns_naive_utc_datetime():
now = utcnow_naive()
assert now.tzinfo is None
assert abs((now - utcnow_naive()).total_seconds()) < 2
def test_database_timestamp_defaults_use_utcnow_naive():
defaults = (
Session.created_at.default.arg,
Session.updated_at.default.arg,
Session.updated_at.onupdate.arg,
ChatMessage.timestamp.default.arg,
DocumentVersion.created_at.default.arg,
UserToolData.created_at.default.arg,
UserToolData.updated_at.default.arg,
UserToolData.updated_at.onupdate.arg,
TaskRun.started_at.default.arg,
)
for fn in defaults:
assert fn.__name__ == "utcnow_naive"