Фикс состояния

tests/unit_tests/rag/test_retrieval_statement_builder.py

@@ -0,0 +1,65 @@
+from app.modules.rag.persistence.retrieval_statement_builder import RetrievalStatementBuilder
+from app.modules.rag.retrieval.test_filter import build_test_filters, is_test_path
+
+
+def test_retrieve_builder_adds_test_exclusion_filters() -> None:
+    builder = RetrievalStatementBuilder()
+    test_filters = build_test_filters()
+
+    sql, params = builder.build_retrieve(
+        "rag-1",
+        [0.1, 0.2],
+        query_text="Explain user service",
+        layers=["C0_SOURCE_CHUNKS"],
+        exclude_path_prefixes=test_filters.exclude_path_prefixes,
+        exclude_like_patterns=test_filters.exclude_like_patterns,
+    )
+
+    assert "NOT (" in sql
+    assert "vector_dims(embedding) = vector_dims(CAST(:emb AS vector))" in sql
+    assert "path LIKE :exclude_prefix_0" in sql
+    assert "lower(path) LIKE :exclude_like_0" in sql
+    assert "ESCAPE E'\\\\'" in sql
+    assert params["exclude_prefix_0"] == "tests/%"
+    assert "%.test.%" in params.values()
+    assert "%\\_test.%" in params.values()
+
+
+def test_retrieve_builder_adds_prefer_bonus_sorting() -> None:
+    builder = RetrievalStatementBuilder()
+
+    sql, params = builder.build_retrieve(
+        "rag-1",
+        [0.1, 0.2],
+        query_text="find context tests",
+        layers=["C1_SYMBOL_CATALOG"],
+        prefer_path_prefixes=["tests/"],
+        prefer_like_patterns=["%/test\\_%.py"],
+    )
+
+    assert "AS prefer_bonus" in sql
+    assert "AS structural_rank" in sql
+    assert "WHEN layer = 'C4_SEMANTIC_ROLES' THEN 2" in sql
+    assert "ORDER BY prefer_bonus ASC, test_penalty ASC, layer_rank ASC" in sql
+    assert params["prefer_prefix_0"] == "tests/%"
+    assert params["prefer_like_0"] == "%/test\\_%.py"
+
+
+def test_lexical_builder_omits_test_filters_when_not_requested() -> None:
+    builder = RetrievalStatementBuilder()
+
+    sql, params = builder.build_lexical_code(
+        "rag-1",
+        query_text="Explain user service",
+        prefer_non_tests=False,
+    )
+
+    assert sql is not None
+    assert "exclude_prefix" not in sql
+    assert "exclude_like" not in sql
+    assert not any(key.startswith("exclude_") for key in params)
+
+
+def test_test_filter_does_not_treat_contest_file_as_test() -> None:
+    assert is_test_path("app/contest.py") is False
+    assert is_test_path("tests/test_users.py") is True