43 lines
1.3 KiB
Python
43 lines
1.3 KiB
Python
from fastapi.testclient import TestClient
|
|
|
|
|
|
def test_profile_update_accepts_new_fields():
|
|
from app.main import app
|
|
body = {
|
|
"name": "테스트",
|
|
"preferred_districts": {
|
|
"S": ["강남구", "서초구"],
|
|
"A": ["송파구"],
|
|
"B": [],
|
|
"C": [],
|
|
"D": [],
|
|
},
|
|
"min_match_score": 75,
|
|
"notify_enabled": True,
|
|
}
|
|
with TestClient(app) as client:
|
|
resp = client.put("/api/realestate/profile", json=body)
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["preferred_districts"]["S"] == ["강남구", "서초구"]
|
|
assert data["min_match_score"] == 75
|
|
assert data["notify_enabled"] is True
|
|
|
|
|
|
def test_profile_get_returns_defaults_for_new_fields():
|
|
from app.main import app
|
|
from app.db import upsert_profile, _conn
|
|
|
|
# 이전 테스트 데이터 제거 후 새 프로필 삽입
|
|
with _conn() as conn:
|
|
conn.execute("DELETE FROM user_profile")
|
|
upsert_profile({"name": "기본"})
|
|
|
|
with TestClient(app) as client:
|
|
resp = client.get("/api/realestate/profile")
|
|
assert resp.status_code == 200
|
|
data = resp.json()
|
|
assert data["preferred_districts"] == {}
|
|
assert data["min_match_score"] == 70
|
|
assert data["notify_enabled"] is True
|