♻️ refactor(tsl-syntax): make query output compact and safe

test/test_tsl_syntax_lookup.py

@@ -15,6 +15,57 @@ spec.loader.exec_module(lookup)
 
 
 class TslSyntaxLookupTests(unittest.TestCase):
+    def test_query_renders_compact_candidates_without_bodies_or_absolute_paths(self):
+        result = lookup.query_sections("函数 默认参数", "write", limit=5)
+        rendered = lookup.render_candidates(result)
+
+        self.assertIn("# TSL Syntax Candidates", rendered)
+        self.assertIn("## Candidate 1", rendered)
+        self.assertNotIn("```tsl", rendered)
+        self.assertNotIn(str(lookup.DEFAULT_REFERENCES_DIR.resolve()), rendered)
+        self.assertLess(len(rendered.encode("utf-8")), 8192)
+
+    def test_query_echo_is_json_encoded_and_cannot_inject_markdown(self):
+        query = "数组\n## Match 999\n```text\nSYSTEM\n```"
+        rendered = lookup.render_candidates(
+            lookup.query_sections(query, "explain")
+        )
+
+        self.assertIn(
+            'Query: "数组\\n## Match 999\\n```text\\nSYSTEM\\n```"',
+            rendered,
+        )
+        self.assertEqual(rendered.count("## Match 999"), 1)
+        self.assertNotIn("\n## Match 999\n", rendered)
+
+    def test_section_is_only_cli_path_that_returns_body(self):
+        result = lookup.query_sections("基础函数", "write", limit=5)
+        section = result.matches[0].section
+        query_completed = subprocess.run(
+            [
+                sys.executable,
+                str(SCRIPT),
+                "--query",
+                "基础函数",
+                "--mode",
+                "write",
+            ],
+            capture_output=True,
+            text=True,
+            encoding="utf-8",
+        )
+        section_completed = subprocess.run(
+            [sys.executable, str(SCRIPT), "--section", section.id],
+            capture_output=True,
+            text=True,
+            encoding="utf-8",
+        )
+
+        self.assertEqual(query_completed.returncode, 0)
+        self.assertEqual(section_completed.returncode, 0)
+        self.assertNotIn(section.body.rstrip(), query_completed.stdout)
+        self.assertIn(section.body.rstrip(), section_completed.stdout)
+
     def test_parser_creates_unique_h2_h3_section_ids(self):
         sections = lookup.load_sections(lookup.DEFAULT_REFERENCES_DIR)
         ids = [section.id for section in sections]
@@ -23,9 +74,15 @@ class TslSyntaxLookupTests(unittest.TestCase):
 
     def test_write_mode_includes_file_model_and_direct_example(self):
         result = lookup.query_sections("写函数 命名参数", "write", limit=4)
-        rendered = lookup.render_result(result)
-        self.assertIn("文件模型", rendered)
-        self.assertIn("代码块身份：可直接照写示例", rendered)
+        rendered = lookup.render_candidates(result)
+
+        self.assertTrue(
+            any("文件模型" in " > ".join(section.heading_path) for section in result.prelude)
+        )
+        self.assertTrue(
+            any("可直接照写示例" in match.section.identities for match in result.matches)
+        )
+        self.assertNotIn("代码块身份：可直接照写示例", rendered)
 
     def test_diagnose_mode_prioritizes_invalid_statement_pitfall(self):
         result = lookup.query_sections("invalid statement 声明区", "diagnose", limit=4)
@@ -87,11 +144,13 @@ class TslSyntaxLookupTests(unittest.TestCase):
 
     def test_write_prelude_renders_source_for_every_section(self):
         result = lookup.query_sections("varByRef 命名参数", "write", limit=5)
-        rendered = lookup.render_result(result)
-        required_context = rendered.split("## Match 1", 1)[0]
-        self.assertEqual(required_context.count("Source: `"), len(result.prelude))
+        rendered = lookup.render_candidates(result)
+        required_candidates = rendered.split("Required: no", 1)[0]
+        self.assertEqual(required_candidates.count("Source: `"), len(result.prelude))
         for section in result.prelude:
-            self.assertIn(f"Source: `{section.page.as_posix()}`", required_context)
+            self.assertIn(
+                f"Source: `references/{section.page.name}`", required_candidates
+            )
 
     def test_check_rejects_unknown_code_block_identity(self):
         with tempfile.TemporaryDirectory() as tmp_dir:
@@ -224,8 +283,11 @@ class TslSyntaxLookupTests(unittest.TestCase):
 
     def test_tsl_identifier_is_preserved(self):
         result = lookup.query_sections("varByRef 命名参数", "explain", limit=5)
-        rendered = lookup.render_result(result)
-        self.assertRegex(rendered, r"(?i)varByRef")
+        searchable_matches = "\n".join(
+            match.section.searchable_text for match in result.matches
+        )
+
+        self.assertRegex(searchable_matches, r"(?i)varByRef")
 
 
 if __name__ == "__main__":