chore: init from code-review-graph-main snapshot (v2.3.7)

tests/test_ansible_parser.py

@@ -0,0 +1,96 @@
+"""Regression tests for safe, connected Ansible graph extraction."""
+
+from pathlib import Path
+
+from code_review_graph.graph import GraphStore
+from code_review_graph.parser import CodeParser, NodeInfo
+
+
+def _qualified(node: NodeInfo) -> str:
+    if node.kind == "File":
+        return node.file_path
+    if node.parent_name:
+        return f"{node.file_path}::{node.parent_name}.{node.name}"
+    return f"{node.file_path}::{node.name}"
+
+
+def test_ordinary_yaml_in_tasks_directory_is_not_treated_as_ansible() -> None:
+    parser = CodeParser()
+    source = b"""\
+- name: frontend
+  image: nginx:latest
+  ports:
+    - 8080
+"""
+
+    nodes, edges = parser.parse_bytes(
+        Path("roles/example/tasks/application.yaml"),
+        source,
+    )
+
+    assert nodes == []
+    assert edges == []
+
+
+def test_ansible_relationships_reference_real_unique_nodes(tmp_path: Path) -> None:
+    parser = CodeParser()
+    path = Path("playbooks/deploy.yml")
+    source = b"""\
+- name: Deploy application
+  hosts: all
+  tasks:
+    - name: Restart application
+      ansible.builtin.debug:
+        msg: first
+      notify: Reload application
+    - name: Restart application
+      ansible.builtin.debug:
+        msg: second
+  handlers:
+    - name: Restart service
+      listen: Reload application
+      ansible.builtin.service:
+        name: application
+        state: restarted
+"""
+
+    nodes, edges = parser.parse_bytes(path, source)
+
+    tasks = [
+        node
+        for node in nodes
+        if node.extra.get("ansible_kind") == "task"
+    ]
+    assert len(tasks) == 2
+    assert len({node.name for node in tasks}) == 2
+    assert all(node.name.startswith("Restart application") for node in tasks)
+
+    qualified = {_qualified(node) for node in nodes}
+    internal_edges = [
+        edge
+        for edge in edges
+        if edge.kind == "CONTAINS"
+        or edge.extra.get("ansible_kind") == "notify"
+    ]
+    assert internal_edges
+    assert all(edge.source in qualified for edge in internal_edges)
+    assert all(edge.target in qualified for edge in internal_edges)
+
+    store = GraphStore(tmp_path / "graph.db")
+    try:
+        for node in nodes:
+            store.upsert_node(node)
+        for edge in edges:
+            store.upsert_edge(edge)
+
+        notify = next(
+            edge
+            for edge in internal_edges
+            if edge.extra.get("ansible_kind") == "notify"
+        )
+        assert store.get_node(notify.source) is not None
+        handler = store.get_node(notify.target)
+        assert handler is not None
+        assert handler.extra["ansible_kind"] == "handler"
+    finally:
+        store.close()