o2o-negosium-original/agent/tests/test_p7_apis.py
hbyang 18020a9e07 [feat] 카드 카탈로그 DB 정본화 — config 결합 제거 + 학습 보존 자동반영
카드 카탈로그(negodata)가 Q-table action space 를 정의하는 정본이 되고, 카드 변경이
config 수정·학습 손실 없이 agent 에 자동 반영되는 고리를 완성.

- action space 정리: 카탈로그 전체(NGC-001~011, 11장) 고정, 견적별 선택은 축소가 아니라
  available_mask(_selection_mask) 로 처리 — action_id↔카드 대응을 견적마다 일정하게 유지해
  Q-table 학습 일관성 보장. 구 인덱스 방식(selected[action_id]) 폐기.
- ① 카탈로그 DB 정본화: action_mapping.type=db 면 registry 가 card.nego_cards(user_id NULL,
  number 순) 조회로 action_to_card 동적 구성(파일은 폴백). port/adapter(card_catalog_*).
  _base=type:db. → negodata 카드 추가/삭제 시 config 수정 불필요.
- ② 차원 변경 학습 보존 마이그레이션: migrate_active_version_dim — 겹치는 셀 복사
  (append/truncate 안전) + 새 카드 fresh. model_store.load 가 차원 불일치 시 호출.
- ③ reload 엔드포인트: /v1/catalog-refresh(테넌트) · /v1/catalog-refresh-all(전역, 화이트리스트).
- ④ 브랜드: company_profile_repo — 자동 온보딩 고객사(company_id UUID)는
  company.companies.name 으로 {company_name} 채움. 데모 테넌트는 파일 유지.
- 크로스서비스: negodata card_service 가 공용 nego 카드 변경 시 agent_notify 로 전역 리로드 알림
  (best-effort, is_test skip). config 에 agent_base_url.
- 하니스 episodes 400→600(action 11 수렴). 테스트 갱신·추가로 agent 98/98.

알려진 갭(후속): per-company 카탈로그 스코프(회사 카드도 action space 포함), 카탈로그 중간
삭제 시 카드번호 기반 마이그레이션.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-08 14:56:28 +09:00

115 lines
4.8 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_catalog_refresh(client, db_engine):
# 카탈로그 발행 후 엔진 재조립 트리거 — 성공 + action_space 반환. 헤더 없으면 400.
r = await client.post("/v1/catalog-refresh", headers=H)
assert r.status_code == 200
body = r.json()
assert body["success"] and body["action_space_size"] >= 1
r2 = await client.post("/v1/catalog-refresh")
assert r2.status_code == 400
@pytest.mark.asyncio
async def test_catalog_refresh_all_no_tenant_header(client, db_engine):
# 공용 카탈로그 전역 반영 — 테넌트 헤더 없이도 200(화이트리스트) + 캐시 클리어.
await client.post("/v1/catalog-refresh", headers=H) # 엔진 하나 캐시
r = await client.post("/v1/catalog-refresh-all") # 헤더 없음
assert r.status_code == 200
assert r.json()["success"] is True
@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