Create test_sql_adapter.py

tests_06/test_sql_adapter.py

@@ -0,0 +1,44 @@
+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]