75 lines
3.0 KiB
Python
75 lines
3.0 KiB
Python
|
"""会话时区回归测试：存储层必须统一 UTC。
|
||
|
|
|
||
|
|
为什么值得单独一条集成测试：这个缺陷的表现是"数据照写、接口照跑、时间悄悄错 8 小时"，
|
||
|
|
只有在比较 DB 默认值与默认值之外的第二个时间源时才会暴露。历史上它造成过一次真实
|
||
|
|
故障——给 `sys_user_role.assigned_at` 用 MySQL `NOW()` 写入的时间超前 UTC 8 小时，
|
||
|
|
被 RBAC 的 `assigned_at <= now` 判为"尚未生效"，接口直接 403。
|
||
|
|
|
||
|
|
同时锁住实现方式：`init_command` 必须仍在 DSN 上。曾试过用 SQLAlchemy 的 `connect`
|
||
|
|
事件写 `SET time_zone`，在 asyncmy 这套 asyncio 方言上**不报错也不生效**，
|
||
|
|
静默退回本地时区——所以这里断言的是"连接后的会话时区"这一可观测事实。
|
||
|
|
"""
|
||
|
|
|
||
|
|
import pytest
|
||
|
|
from sqlalchemy import text
|
||
|
|
|
||
|
|
from app.infrastructure.db import SessionFactory
|
||
|
|
|
||
|
|
UTC_OFFSET_SECONDS = 0
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.integration
|
||
|
|
async def test_mysql_session_timezone_is_utc():
|
||
|
|
async with SessionFactory() as session:
|
||
|
|
time_zone = (await session.execute(text("SELECT @@session.time_zone"))).scalar()
|
||
|
|
offset = (
|
||
|
|
await session.execute(
|
||
|
|
text("SELECT TIMESTAMPDIFF(SECOND, UTC_TIMESTAMP(6), NOW(6))")
|
||
|
|
)
|
||
|
|
).scalar()
|
||
|
|
|
||
|
|
assert time_zone == "+00:00", f"会话时区应被钉在 UTC，实际 {time_zone!r}"
|
||
|
|
assert offset == UTC_OFFSET_SECONDS, f"会话时间与 UTC 相差 {offset} 秒，应为 0"
|
||
|
|
|
||
|
|
|
||
|
|
@pytest.mark.integration
|
||
|
|
async def test_db_default_timestamp_matches_application_utc():
|
||
|
|
"""DB 的 `CURRENT_TIMESTAMP` 默认值必须与应用侧 `datetime.now(UTC)` 同源。
|
||
|
|
|
||
|
|
断言方式刻意不依赖具体时区：同一行里写入一次 DB 默认值和一次应用值，
|
||
|
|
两者相差必须为 0 秒——改动前这条断言会得到 28800 秒。
|
||
|
|
"""
|
||
|
|
from datetime import UTC, datetime
|
||
|
|
|
||
|
|
async with SessionFactory() as session:
|
||
|
|
await session.execute(
|
||
|
|
text(
|
||
|
|
"CREATE TEMPORARY TABLE tz_default_probe ("
|
||
|
|
" id INT PRIMARY KEY,"
|
||
|
|
" db_default DATETIME(6) NOT NULL DEFAULT CURRENT_TIMESTAMP(6),"
|
||
|
|
" app_value DATETIME(6) NOT NULL)"
|
||
|
|
)
|
||
|
|
)
|
||
|
|
app_now = datetime.now(UTC).replace(tzinfo=None)
|
||
|
|
await session.execute(
|
||
|
|
text(
|
||
|
|
"INSERT INTO tz_default_probe (id, app_value) VALUES (:id, :app_value)"
|
||
|
|
),
|
||
|
|
{"id": 1, "app_value": app_now},
|
||
|
|
)
|
||
|
|
row = (
|
||
|
|
await session.execute(
|
||
|
|
text(
|
||
|
|
"SELECT db_default, app_value,"
|
||
|
|
" TIMESTAMPDIFF(SECOND, app_value, db_default) AS gap_seconds"
|
||
|
|
" FROM tz_default_probe WHERE id = 1"
|
||
|
|
)
|
||
|
|
)
|
||
|
|
).mappings().one()
|
||
|
|
await session.execute(text("DROP TEMPORARY TABLE tz_default_probe"))
|
||
|
|
|
||
|
|
assert row["gap_seconds"] == UTC_OFFSET_SECONDS, (
|
||
|
|
f"DB 默认值 {row['db_default']} 与应用写入 {row['app_value']} 相差 "
|
||
|
|
f"{row['gap_seconds']} 秒，说明会话时区未统一到 UTC"
|
||
|
|
)
|