Drops .replace(tzinfo=None) from 8 fixture sites that mirrored the production-side strip convention. Two of these (test_live_state_poller.py test_new_player_opens_session_with_backfilled_join, test_models.py test_user_has_password_changed_at_default) now fail with TypeError when comparing aware in-memory values against naive DB reads -- that failure is intentional and describes the contract commit 2 must satisfy: DB-sourced datetimes return aware UTC. The remaining 6 sites were already cosmetic (fixture-seed only, no aware-vs-DB comparison) but are flipped here so future authors write aware fixtures. The three deliberately-naive sites in test_timeago.py (lines 67, 73, 113) are LEFT untouched -- they exercise _ensure_utc's normalize-up path and are feature tests, not workarounds. See docs/superpowers/specs/2026-05-16-tz-aware-datetime-design.md. Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
110 lines
4 KiB
Python
110 lines
4 KiB
Python
from sqlalchemy import select
|
|
|
|
from l4d2web.db import init_db, session_scope
|
|
from l4d2web.models import (
|
|
Blueprint,
|
|
Server,
|
|
ServerLiveState,
|
|
ServerPlayerSession,
|
|
SteamUserProfile,
|
|
User,
|
|
now_utc as now_utc_aware,
|
|
)
|
|
|
|
|
|
def test_create_user_and_blueprint(tmp_path, monkeypatch) -> None:
|
|
monkeypatch.setenv("DATABASE_URL", f"sqlite:///{tmp_path/'app.db'}")
|
|
|
|
init_db()
|
|
|
|
with session_scope() as session:
|
|
user = User(username="alice", password_digest="digest", admin=False)
|
|
session.add(user)
|
|
session.flush()
|
|
|
|
blueprint = Blueprint(user_id=user.id, name="default", arguments="[]", config="[]")
|
|
session.add(blueprint)
|
|
session.flush()
|
|
|
|
assert user.id is not None
|
|
assert blueprint.id is not None
|
|
|
|
|
|
def test_user_has_password_changed_at_default(tmp_path, monkeypatch):
|
|
from datetime import UTC, datetime
|
|
from l4d2web.app import create_app
|
|
from l4d2web.auth import hash_password
|
|
|
|
db_url = f"sqlite:///{tmp_path/'pw.db'}"
|
|
monkeypatch.setenv("DATABASE_URL", db_url)
|
|
create_app({"TESTING": True, "DATABASE_URL": db_url, "SECRET_KEY": "test"})
|
|
init_db()
|
|
|
|
before = datetime.now(UTC)
|
|
with session_scope() as db:
|
|
db.add(User(username="alice", password_digest=hash_password("secret")))
|
|
with session_scope() as db:
|
|
user = db.query(User).filter_by(username="alice").one()
|
|
|
|
assert user.password_changed_at is not None
|
|
assert user.password_changed_at >= before
|
|
|
|
|
|
def test_server_has_rcon_password_column(tmp_path, monkeypatch) -> None:
|
|
monkeypatch.setenv("DATABASE_URL", f"sqlite:///{tmp_path/'t.db'}")
|
|
init_db()
|
|
with session_scope() as db:
|
|
u = User(username="u", password_digest="x")
|
|
db.add(u); db.flush()
|
|
bp = Blueprint(user_id=u.id, name="bp", arguments="[]", config="[]")
|
|
db.add(bp); db.flush()
|
|
s = Server(
|
|
user_id=u.id, blueprint_id=bp.id, name="s", port=27500,
|
|
rcon_password="abc",
|
|
)
|
|
db.add(s); db.flush()
|
|
assert db.scalar(select(Server.rcon_password).where(Server.id == s.id)) == "abc"
|
|
|
|
|
|
def test_server_live_state_table_columns(tmp_path, monkeypatch) -> None:
|
|
monkeypatch.setenv("DATABASE_URL", f"sqlite:///{tmp_path/'t.db'}")
|
|
init_db()
|
|
with session_scope() as db:
|
|
u = User(username="u", password_digest="x"); db.add(u); db.flush()
|
|
bp = Blueprint(user_id=u.id, name="bp", arguments="[]", config="[]"); db.add(bp); db.flush()
|
|
s = Server(user_id=u.id, blueprint_id=bp.id, name="s", port=27501, rcon_password="x")
|
|
db.add(s); db.flush()
|
|
row = ServerLiveState(
|
|
server_id=s.id, started_at=now_utc_aware(), last_seen_at=now_utc_aware(),
|
|
players=2, max_players=4, bots=0, map="c1m1_hotel", hibernating=False,
|
|
)
|
|
db.add(row); db.flush()
|
|
|
|
|
|
def test_server_player_session_table_columns(tmp_path, monkeypatch) -> None:
|
|
monkeypatch.setenv("DATABASE_URL", f"sqlite:///{tmp_path/'t.db'}")
|
|
init_db()
|
|
with session_scope() as db:
|
|
u = User(username="u", password_digest="x"); db.add(u); db.flush()
|
|
bp = Blueprint(user_id=u.id, name="bp", arguments="[]", config="[]"); db.add(bp); db.flush()
|
|
s = Server(user_id=u.id, blueprint_id=bp.id, name="s", port=27502, rcon_password="x")
|
|
db.add(s); db.flush()
|
|
row = ServerPlayerSession(
|
|
server_id=s.id, steam_id_64="76561197960828710",
|
|
joined_at=now_utc_aware(), left_at=None,
|
|
name_at_join="Crone", min_ping=42, max_ping=185,
|
|
)
|
|
db.add(row); db.flush()
|
|
|
|
|
|
def test_steam_user_profile_table_columns(tmp_path, monkeypatch) -> None:
|
|
monkeypatch.setenv("DATABASE_URL", f"sqlite:///{tmp_path/'t.db'}")
|
|
init_db()
|
|
with session_scope() as db:
|
|
row = SteamUserProfile(
|
|
steam_id_64="76561197960828710",
|
|
persona_name="MrCool42",
|
|
avatar_url="https://avatars.cloudflare.steamstatic.com/abc_medium.jpg",
|
|
fetched_at=now_utc_aware(),
|
|
)
|
|
db.add(row); db.flush()
|