Function large_store

tests/test_visualization.py:326–429

Store with enough nodes/communities to test aggregation.

(tmp_path)

@pytest.fixture
def large_store(tmp_path):
    """Store with enough nodes/communities to test aggregation."""
    db_path = tmp_path / "large.db"
    store = GraphStore(db_path)

    # Create nodes across multiple files (simulates a larger codebase)
    files = [f"src/mod{i}.py" for i in range(5)]
    for fp in files:
        file_node = NodeInfo(
            kind="File", name=fp.split("/")[-1], file_path=fp,
            line_start=1, line_end=100, language="python",
            parent_name=None, params=None, return_type=None,
            modifiers=None, is_test=False, extra={},
        )
        store.upsert_node(file_node)
        # Add some functions per file
        for j in range(3):
            func_node = NodeInfo(
                kind="Function", name=f"func_{j}",
                file_path=fp, line_start=10 + j * 10, line_end=20 + j * 10,
                language="python", parent_name=None,
                params="x", return_type="int",
                modifiers=None, is_test=False, extra={},
            )
            store.upsert_node(func_node)
            # CONTAINS edge from file to function
            store.upsert_edge(EdgeInfo(
                kind="CONTAINS", source=fp,
                target=f"{fp}::func_{j}",
                file_path=fp, line=10 + j * 10, extra={},
            ))

    # Add some cross-file CALLS edges
    store.upsert_edge(EdgeInfo(
        kind="CALLS",
        source="src/mod0.py::func_0",
        target="src/mod1.py::func_1",
        file_path="src/mod0.py", line=15, extra={},
    ))
    store.upsert_edge(EdgeInfo(
        kind="CALLS",
        source="src/mod2.py::func_0",
        target="src/mod3.py::func_2",
        file_path="src/mod2.py", line=12, extra={},
    ))
    store.upsert_edge(EdgeInfo(
        kind="CALLS",
        source="src/mod1.py::func_2",
        target="src/mod4.py::func_0",
        file_path="src/mod1.py", line=35, extra={},
    ))

    # Set community_id on nodes (simulate community detection)
    store._conn.execute(
        "UPDATE nodes SET community_id = 0 WHERE file_path IN ('src/mod0.py', 'src/mod1.py')"
    )
    store._conn.execute(
        "UPDATE nodes SET community_id = 1 WHERE file_path IN ('src/mod2.py', 'src/mod3.py')"