tests_06/test_sql_adapter.py

from engine_03.flow.sql_adapter import apply_command


def test_add_node_and_edge_and_style():
    nodes = []
    edges = []

    # Add node
    nodes, edges, msg1 = apply_command(
        'ADD NODE id=event_in label="Incoming Event" type=event',
        nodes,
        edges,
    )
    assert "event_in" in msg1
    assert any(r[0] == "event_in" for r in nodes)

    # Add edge
    nodes, edges, msg2 = apply_command(
        'ADD EDGE source=event_in target=handler label="to handler"',
        nodes,
        edges,
    )
    assert "Edge event_in -> handler added." in msg2
    assert any(r[0] == "event_in" and r[1] == "handler" for r in edges)

    # Style node
    nodes, edges, msg3 = apply_command(
        'STYLE NODE id=event_in style="fill:#0ea5e9,stroke:#0369a1"',
        nodes,
        edges,
    )
    assert "Style applied to node `event_in`." in msg3
    row = next(r for r in nodes if r[0] == "event_in")
    assert "fill:#0ea5e9" in row[3]

    # Style edge
    nodes, edges, msg4 = apply_command(
        'STYLE EDGE source=event_in target=handler style="stroke:#22c55e,stroke-width:2px"',
        nodes,
        edges,
    )
    assert "Style applied to edge event_in -> handler." in msg4
    erow = next(r for r in edges if r[0] == "event_in" and r[1] == "handler")
    assert "stroke:#22c55e" in erow[3]