refs #732: scope report titles to selected user

ai-web-agent-console/tests/test_scenarios.py

@@ -6,6 +6,7 @@ import json
 from pathlib import Path
 import re
 import tempfile
+from types import SimpleNamespace
 from typing import Any, Mapping
 import unittest
 from unittest.mock import patch
@@ -237,6 +238,132 @@ class DemoScenarioConfigTest(unittest.TestCase):
             source,
         )
 
+    def test_chat_context_is_scoped_to_current_selected_user(self) -> None:
+        source = (Path(__file__).parents[1] / "app.py").read_text(encoding="utf-8")
+        tree = ast.parse(source)
+        helper = next(
+            node
+            for node in tree.body
+            if isinstance(node, ast.FunctionDef)
+            and node.name == "load_chat_context"
+        )
+        captured: dict[str, Any] = {}
+
+        class FakeConnection:
+            def __enter__(self):
+                return self
+
+            def __exit__(self, *_args):
+                return False
+
+            def execute(self, sql, params):
+                captured["sql"] = sql
+                captured["params"] = params
+                return self
+
+            def fetchall(self):
+                return [{"question": "내 담당 선사", "answer": "2건"}]
+
+        namespace: dict[str, Any] = {
+            "CHAT_CONTEXT_TURNS": 8,
+            "MAX_CONVERSATION_MESSAGES": 16,
+            "_chat_db_connect": FakeConnection,
+            "_is_failed_synthesis_answer": lambda _value: False,
+        }
+        exec(compile(ast.Module(body=[helper], type_ignores=[]), "app.py", "exec"), namespace)
+
+        messages = namespace["load_chat_context"](
+            "conversation-1",
+            selected_user_id="E1002",
+        )
+
+        self.assertIn("selected_user_id = ?", captured["sql"])
+        self.assertEqual(
+            captured["params"],
+            ("conversation-1", "E1002", "E1002", 8),
+        )
+        self.assertEqual(messages[0]["content"], "내 담당 선사")
+
+    def test_standalone_question_uses_current_user_without_prior_context(self) -> None:
+        source = (Path(__file__).parents[1] / "app.py").read_text(encoding="utf-8")
+        tree = ast.parse(source)
+        helpers = [
+            node
+            for node in tree.body
+            if isinstance(node, ast.FunctionDef)
+            and node.name in {"_conversation_context", "resolve_standalone_question"}
+        ]
+        captured: dict[str, Any] = {}
+
+        class FakeClient:
+            def complete(self, **kwargs):
+                captured.update(kwargs)
+                return json.dumps(
+                    {
+                        "standalone_question": (
+                            "E1002 사용자의 담당 선사 최신 실적을 조회해줘"
+                        )
+                    },
+                    ensure_ascii=False,
+                )
+
+        namespace: dict[str, Any] = {
+            "Any": Any,
+            "Mapping": Mapping,
+            "MAX_CONVERSATION_MESSAGES": 16,
+            "json": json,
+            "resolve_model_profile": lambda _key: SimpleNamespace(
+                model_id="model",
+                answer_model_region="region",
+                answer_model_endpoint="endpoint",
+            ),
+            "build_oci_genai_completion_client": lambda *_args: FakeClient(),
+            "temperature_for_model_profile": lambda _profile: 0.0,
+        }
+        exec(compile(ast.Module(body=helpers, type_ignores=[]), "app.py", "exec"), namespace)
+
+        rewritten = namespace["resolve_standalone_question"](
+            question="내 담당 선사 최신 실적을 리포트로 보여줘",
+            messages=[],
+            model_profile_key="test",
+            selected_user_id="E1002",
+        )
+        prompt_payload = json.loads(captured["user_prompt"])
+
+        self.assertTrue(rewritten.startswith("E1002"))
+        self.assertEqual(prompt_payload["current_selected_user_id"], "E1002")
+        self.assertIn("authoritative", captured["system_prompt"])
+
+    def test_report_payload_requester_prefers_current_selected_user(self) -> None:
+        source = (Path(__file__).parents[1] / "app.py").read_text(encoding="utf-8")
+        tree = ast.parse(source)
+        helper = next(
+            node
+            for node in tree.body
+            if isinstance(node, ast.FunctionDef)
+            and node.name == "_presentation_payload"
+        )
+        namespace: dict[str, Any] = {
+            "Any": Any,
+            "Mapping": Mapping,
+            "datetime": __import__("datetime").datetime,
+            "timezone": __import__("datetime").timezone,
+            "re": re,
+            "_mcp_structured_rows": lambda _value: [],
+            "_normalize_presentation_value": lambda value: value,
+            "_clean_presentation_title": lambda value: str(value),
+        }
+        exec(compile(ast.Module(body=[helper], type_ignores=[]), "app.py", "exec"), namespace)
+
+        payload = namespace["_presentation_payload"](
+            "E1001 팀장 문맥이 남은 질문",
+            [],
+            title="담당 선사 실적",
+            selected_user_id="E1002",
+        )
+
+        self.assertEqual(payload["report"]["requestedBy"], "E1002")
+
     def test_hmm_report_data_query_drops_html_format_request(self) -> None:
         source = (Path(__file__).parents[1] / "app.py").read_text(encoding="utf-8")
         tree = ast.parse(source)