Function collect_pipeline

agent/src/opentrace_agent/pipeline/pipeline.py:81–98

Convenience for tests — consume all events, return aggregated results.

(
    inp: PipelineInput,
    ctx: PipelineContext | None = None,
    store: Store | None = None,
)

def collect_pipeline(
    inp: PipelineInput,
    ctx: PipelineContext | None = None,
    store: Store | None = None,
) -> tuple[list[PipelineEvent], list[GraphNode], list[GraphRelationship]]:
    """Convenience for tests — consume all events, return aggregated results."""
    events: list[PipelineEvent] = []
    all_nodes: list[GraphNode] = []
    all_rels: list[GraphRelationship] = []

    for event in run_pipeline(inp, ctx, store):
        events.append(event)
        if event.nodes:
            all_nodes.extend(event.nodes)
        if event.relationships:
            all_rels.extend(event.relationships)

    return events, all_nodes, all_rels