"""채팅(chat) 도메인 e2e 테스트 — init / messages(오프닝 seed) / send(협상 진행~종료). agent 는 config.use_mock=true 로 내장 MockAgentClient 를 쓴다(결정론적 플로우). dev negosium_db 를 그대로 쓰므로 전용 테스트 행만 시드/정리한다. """ import uuid import bcrypt import pytest_asyncio from sqlalchemy import text TEST_LOGIN_ID = "pytest_chat_user" TEST_PW = "pytest1234" TEST_SUPPLIER_NAME = "파이테스트채팅공급사" MARK = "PYTESTCHAT-" @pytest_asyncio.fixture async def chat_seed(db_engine): """공급사 + 유저 + 세션 2건(본인: 협상중 P / 협상생성 C) + 1건(타 공급사 X) 시드.""" supplier_id = uuid.uuid4() other_supplier_id = uuid.uuid4() pw_hash = bcrypt.hashpw(TEST_PW.encode("utf-8"), bcrypt.gensalt()).decode("utf-8") # (code, session.status, qt_type, 마감까지 h, quotation.status, 소속 공급사) specs = [ ("P", 2, 1, 2, 2, supplier_id), # 협상중 / 재협상 / +2h / 견적진행중 ("C", 1, 1, 2, 1, supplier_id), # 협상생성 / 재협상 / +2h / 견적생성 ("X", 2, 1, 2, 2, other_supplier_id), # 타 공급사 → 차단 ] sids, qids = {}, {} async def _cleanup(conn): await conn.execute(text(f"DELETE FROM negotiation.chats WHERE session_id IN (SELECT session_id FROM negotiation.sessions WHERE qt_number LIKE '{MARK}%')")) await conn.execute(text(f"DELETE FROM negotiation.sessions WHERE qt_number LIKE '{MARK}%'")) await conn.execute(text(f"DELETE FROM quotation.quotations WHERE number LIKE '{MARK}%'")) await conn.execute(text(f"DELETE FROM partner.items WHERE code LIKE '{MARK}%'")) await conn.execute(text("DELETE FROM supplier.supplier_users WHERE id = :id"), {"id": TEST_LOGIN_ID}) await conn.execute(text("DELETE FROM partner.suppliers WHERE name = :n"), {"n": TEST_SUPPLIER_NAME}) async with db_engine.begin() as conn: await _cleanup(conn) await conn.execute( text("INSERT INTO partner.suppliers (supplier_id, company_id, user_id, name) VALUES (:sid, gen_random_uuid(), gen_random_uuid(), :name)"), {"sid": supplier_id, "name": TEST_SUPPLIER_NAME}, ) await conn.execute( text( "INSERT INTO supplier.supplier_users (supplier_id, id, password, name, last_accessed_at, status, role) " "VALUES (:sid, :id, :pw, '채팅담당자', now(), 1, 1)" ), {"sid": supplier_id, "id": TEST_LOGIN_ID, "pw": pw_hash}, ) for code, sess_st, qt_type, hrs, quote_st, sup in specs: item_id, qt_id, session_id = uuid.uuid4(), uuid.uuid4(), uuid.uuid4() sids[code], qids[code] = session_id, qt_id await conn.execute( text( "INSERT INTO partner.items (item_id, company_id, user_id, name, code, price, model_name, manufacturer, moq, spec) " "VALUES (:iid, gen_random_uuid(), gen_random_uuid(), :name, :code, 100000, :model, '테스트제조사', '10', '규격A')" ), {"iid": item_id, "name": f"상품 {code}", "code": f"{MARK}{code}", "model": f"MODEL-{code}"}, ) await conn.execute( text( "INSERT INTO quotation.quotations (qt_id, user_id, qt_setting_id, version_id, name, number, type, status, start_time, end_time, memo) " "VALUES (:qid, gen_random_uuid(), gen_random_uuid(), gen_random_uuid(), :name, :num, :tp, :st, now(), now() + make_interval(hours => :hrs), '메모')" ), {"qid": qt_id, "name": f"견적 {code}", "num": f"{MARK}{code}", "tp": qt_type, "st": quote_st, "hrs": hrs}, ) await conn.execute( text( "INSERT INTO negotiation.sessions " "(session_id, quotation_id, item_id, supplier_id, qt_number, qt_round, qt_type, target_price, status, end_time) " "VALUES (:sesid, :qid, :iid, :sup, :qtn, 1, :qtt, 100000, :st, now() + make_interval(hours => 2))" ), {"sesid": session_id, "qid": qt_id, "iid": item_id, "sup": sup, "qtn": f"{MARK}{code}", "qtt": qt_type, "st": sess_st}, ) yield {"supplier_id": supplier_id, "sids": sids, "qids": qids} async with db_engine.begin() as conn: await _cleanup(conn) async def _login_token(client): r = await client.post("/v1/auth/login", json={"id": TEST_LOGIN_ID, "pw": TEST_PW}) return r.json()["access_token"] def _h(token): return {"Authorization": f"Bearer {token}"} async def _init(client, token, sid): return await client.get(f"/v1/negotiation/sessions/{sid}/chat/init", headers=_h(token)) async def _messages(client, token, sid): return await client.get(f"/v1/negotiation/sessions/{sid}/chat/messages", headers=_h(token)) async def _send(client, token, sid, user_input, user_input_type=None): body = {"user_input": user_input, "user_input_type": user_input_type} return await client.post(f"/v1/negotiation/sessions/{sid}/chat/send", headers=_h(token), json=body) async def _session_status(db_engine, session_id): async with db_engine.begin() as conn: return (await conn.execute(text("SELECT status FROM negotiation.sessions WHERE session_id = :sid"), {"sid": session_id})).scalar() async def _session_bid(db_engine, session_id): async with db_engine.begin() as conn: return (await conn.execute(text("SELECT bid_price FROM negotiation.sessions WHERE session_id = :sid"), {"sid": session_id})).scalar() async def _session_reject(db_engine, session_id): async with db_engine.begin() as conn: r = (await conn.execute(text("SELECT status, reject_reason FROM negotiation.sessions WHERE session_id = :sid"), {"sid": session_id})).first() return r[0], r[1] # ---- init ------------------------------------------------------------------- async def test_chat_init_returns_meta(client, chat_seed): token = await _login_token(client) body = (await _init(client, token, chat_seed["sids"]["P"])).json() assert body["result"]["success"] is True assert body["session_status"] == 2 assert body["item_name"] == "상품 P" and body["item_price"] == 100000 assert body["item_maker_name"] == "테스트제조사" assert body["quotation_end_time"] # 타이머용 마감 시각 async def test_chat_init_forbidden_other_supplier(client, chat_seed): token = await _login_token(client) body = (await _init(client, token, chat_seed["sids"]["X"])).json() assert body["result"]["code"] == 1300 # NEGO_FORBIDDEN # ---- messages (오프닝 seed) ------------------------------------------------- async def test_messages_seeds_opening(client, chat_seed): token = await _login_token(client) body = (await _messages(client, token, chat_seed["sids"]["P"])).json() assert body["result"]["success"] is True assert len(body["items"]) == 1 msg = body["items"][0] assert msg["sender"] == 1 # ChatSender.BOT (봇) assert msg["next_input_mode"] == "confirm" assert msg["script"] # ---- send (협상 진행 → 종료) ------------------------------------------------ async def test_send_flow_to_completion(client, chat_seed, db_engine): token = await _login_token(client) sid = chat_seed["sids"]["P"] await _messages(client, token, sid) # 오프닝(턴0) seed r1 = (await _send(client, token, sid, "네, 시작할게요")).json() assert r1["result"]["success"] is True assert r1["message"]["next_input_mode"] == "confirm" # 품목안내 assert r1["session_status"] == 2 r2 = (await _send(client, token, sid, "가격 협상 진행")).json() assert r2["message"]["next_input_mode"] == "price" # 가격입력 요청 r3 = (await _send(client, token, sid, "90000", user_input_type="price")).json() assert r3["result"]["success"] is True assert r3["message"]["chat_end"] is True assert r3["session_status"] == 3 # 협상완료(DONE) assert await _session_status(db_engine, sid) == 3 assert await _session_bid(db_engine, sid) == 90000 # 입찰가 확정 async def test_send_price_out_of_range(client, chat_seed): token = await _login_token(client) sid = chat_seed["sids"]["P"] await _messages(client, token, sid) # 목표가 100000 → 허용 [30000, 170000]. 10 은 하한 미만. body = (await _send(client, token, sid, "10", user_input_type="price")).json() assert body["result"]["code"] == 1401 # CHAT_PRICE_OUT_OF_RANGE async def test_send_not_in_progress(client, chat_seed): token = await _login_token(client) sid = chat_seed["sids"]["C"] # 협상생성(미참여 전 단계) body = (await _send(client, token, sid, "네")).json() assert body["result"]["code"] == 1400 # CHAT_NOT_IN_PROGRESS async def test_send_requires_auth(client, chat_seed): sid = chat_seed["sids"]["P"] r = await client.post(f"/v1/negotiation/sessions/{sid}/chat/send", json={"user_input": "네"}) assert r.status_code in (401, 403) # ---- 보완: 거부 저장 / 동시전송 가드 / init 만료 정리 ------------------------ async def test_send_rejection_persists_reason(client, chat_seed, db_engine): token = await _login_token(client) sid = chat_seed["sids"]["P"] await _messages(client, token, sid) # 오프닝 body = (await _send(client, token, sid, "협상 포기합니다")).json() assert body["result"]["success"] is True assert body["message"]["chat_end"] is True assert body["session_status"] == 5 # 협상거부(REJECTED) status, reason = await _session_reject(db_engine, sid) assert status == 5 and reason == "협상 포기합니다" # 거부 사유 저장 async def test_send_blocked_when_prev_turn_pending(client, chat_seed, db_engine): """직전 메시지가 USER(이전 턴 처리 중)면 중복 전송을 거절한다 → CHAT_IN_PROGRESS.""" token = await _login_token(client) sid = chat_seed["sids"]["P"] await _messages(client, token, sid) # 오프닝(seq=1, BOT) # 봇 응답이 아직 안 온 상태를 모사: USER 메시지를 마지막(seq=2)으로 직접 삽입 async with db_engine.begin() as conn: await conn.execute( text("INSERT INTO negotiation.chats (session_id, seq, sender, target_price) VALUES (:sid, 2, 2, 0)"), {"sid": sid}, ) body = (await _send(client, token, sid, "네")).json() assert body["result"]["code"] == 1403 # CHAT_IN_PROGRESS async def test_init_marks_expired_created_as_not_participated(client, chat_seed, db_engine): token = await _login_token(client) sid, qid = chat_seed["sids"]["C"], chat_seed["qids"]["C"] # 협상생성(1) async with db_engine.begin() as conn: await conn.execute(text("UPDATE quotation.quotations SET end_time = now() - make_interval(hours => 1) WHERE qt_id = :qid"), {"qid": qid}) body = (await _init(client, token, sid)).json() assert body["result"]["success"] is True assert body["session_status"] == 4 # 미참여로 정리되어 내려옴 assert await _session_status(db_engine, sid) == 4 # DB 도 전이됨