95 lines
4.0 KiB
Python
95 lines
4.0 KiB
Python
"""P7 — 14개 API 보존 스모크 (tenant 헤더 격리).
|
|
|
|
Chat_server 14개 API 대응: health, chat, invalidate-session, card-update, card-search,
|
|
reset-learning, reset-all, q-table/{versions,switch,current}, experience-logs, train, verification-report.
|
|
모두 X-Tenant-ID 헤더 필요(누락 400). 동작 + company_id 격리 확인.
|
|
"""
|
|
|
|
import pytest
|
|
|
|
H = {"X-Tenant-ID": "ktcommerce"}
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_health_no_tenant(client):
|
|
assert (await client.get("/v1/health")).status_code == 200
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_tenant_header_required_on_management(client):
|
|
for path in ["/v1/q-table/versions", "/v1/experience-logs", "/v1/card-search"]:
|
|
r = await client.get(path)
|
|
assert r.status_code == 400, path
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_qtable_lifecycle_and_logs(client, db_engine):
|
|
# 활성 버전 없음 → current 빈 상태
|
|
r = await client.get("/v1/q-table/current", headers=H)
|
|
assert r.status_code == 200
|
|
|
|
# /chat 한 번 돌려 학습 데이터 생성 (가격협상까지)
|
|
sid = None
|
|
for ui in [None, "확인", "예", "확인", "11000", "예", "10200", "예", "9800", "예",
|
|
"협상 내용을 확인했으며, 이의가 없음에 동의합니다."]:
|
|
cr = await client.post("/v1/chat", headers=H, json={"session_id": sid, "user_input": ui, "rq_type": "재협상"})
|
|
sid = cr.json()["session_id"]
|
|
if cr.json().get("chat_end"):
|
|
break
|
|
|
|
# 버전 생성됨 + 활성
|
|
versions = (await client.get("/v1/q-table/versions", headers=H)).json()
|
|
assert versions["versions"] and any(v["is_active"] for v in versions["versions"])
|
|
cur = (await client.get("/v1/q-table/current", headers=H)).json()
|
|
assert cur["active_version"] is not None and cur["q_value_rows"] >= 1
|
|
|
|
# 경험 로그
|
|
logs = (await client.get("/v1/experience-logs?limit=10", headers=H)).json()
|
|
assert logs["total"] >= 1 and len(logs["logs"]) >= 1
|
|
|
|
# 검증 리포트
|
|
rep = (await client.get("/v1/verification-report", headers=H)).json()
|
|
assert rep["experience_total"] >= 1
|
|
|
|
# 오프라인 학습
|
|
tr = (await client.post("/v1/train", headers=H, json={"epochs": 2})).json()
|
|
assert tr["success"] and tr["trained_transitions"] >= 1
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_card_update_search(client, db_engine):
|
|
up = (await client.post("/v1/card-update", headers=H, json={"action_id": 0, "card_id": "CUSTOM-X"})).json()
|
|
assert up["success"]
|
|
s = (await client.get("/v1/card-search?card_id=CUSTOM-X", headers=H)).json()
|
|
assert s["found"] and 0 in s["action_ids"]
|
|
allm = (await client.get("/v1/card-search", headers=H)).json()
|
|
assert any(m["card_id"] == "CUSTOM-X" for m in allm["mapping"])
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_invalidate_and_reset_scoped(client, db_engine):
|
|
# 가격협상 카드선택이 일어나는 긴 경로(850→900→990)로 양 테넌트 데이터 생성
|
|
convo = [None, "확인", "예", "확인", "11000", "예", "10200", "예", "9800", "예",
|
|
"협상 내용을 확인했으며, 이의가 없음에 동의합니다."]
|
|
sid = None
|
|
for ui in convo:
|
|
cr = await client.post("/v1/chat", headers=H, json={"session_id": sid, "user_input": ui})
|
|
sid = cr.json()["session_id"]
|
|
if cr.json().get("chat_end"):
|
|
break
|
|
H2 = {"X-Tenant-ID": "imarketkorea"}
|
|
sid2 = None
|
|
for ui in convo:
|
|
cr = await client.post("/v1/chat", headers=H2, json={"session_id": sid2, "user_input": ui})
|
|
sid2 = cr.json()["session_id"]
|
|
if cr.json().get("chat_end"):
|
|
break
|
|
|
|
before2 = (await client.get("/v1/experience-logs", headers=H2)).json()["total"]
|
|
assert before2 >= 1
|
|
|
|
# ktcommerce reset-all → imarketkorea 무영향
|
|
assert (await client.post("/v1/reset-all", headers=H)).json()["success"]
|
|
assert (await client.get("/v1/experience-logs", headers=H)).json()["total"] == 0
|
|
assert (await client.get("/v1/experience-logs", headers=H2)).json()["total"] == before2
|