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).replace(tzinfo=None)
|
|
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()
|