You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
985 B
Python
35 lines
985 B
Python
"""N+1 쿼리 회귀 — 트리/대시보드의 실제 발행 쿼리 수를 센다."""
|
|
from contextlib import contextmanager
|
|
|
|
from sqlalchemy import event
|
|
|
|
|
|
@contextmanager
|
|
def count_queries(engine):
|
|
counter = {"n": 0}
|
|
|
|
def _before(conn, cursor, statement, *a, **k):
|
|
counter["n"] += 1
|
|
|
|
event.listen(engine, "before_cursor_execute", _before)
|
|
try:
|
|
yield counter
|
|
finally:
|
|
event.remove(engine, "before_cursor_execute", _before)
|
|
|
|
|
|
def test_tree_no_nplus1(client, session):
|
|
_s, engine = session
|
|
with count_queries(engine) as c:
|
|
r = client.get("/api/tree")
|
|
assert r.status_code == 200
|
|
assert c["n"] <= 6, f"트리 쿼리 {c['n']}개 — N+1 의심"
|
|
|
|
|
|
def test_dashboard_query_budget(client, session):
|
|
_s, engine = session
|
|
with count_queries(engine) as c:
|
|
r = client.get("/api/dashboard")
|
|
assert r.status_code == 200
|
|
assert c["n"] <= 12, f"대시보드 쿼리 {c['n']}개 — 집계 합치기 필요"
|