Function test_reroute_node_deletion

tests/test_reroute_node_deletion.py:20–122

Test deletion of RerouteNode objects.

()

from core.reroute_node import RerouteNode

def test_reroute_node_deletion():
    """Test deletion of RerouteNode objects."""
    
    app = QApplication.instance()
    if app is None:
        app = QApplication(sys.argv)
    
    window = NodeEditorWindow()
    graph = window.graph
    
    # Clear any existing content
    graph.clear_graph()
    
    print("Creating test setup with RerouteNode...")
    
    # Create a regular node
    node1 = graph.create_node("Test Node", pos=(100, 100))
    node1.set_code('''
@node_entry
def test_function() -> str:
    return "test_output"
''')
    
    # Create a RerouteNode
    reroute = RerouteNode()
    reroute.setPos(300, 100)
    graph.addItem(reroute)
    graph.nodes.append(reroute)
    
    # Create another regular node
    node2 = graph.create_node("Test Node 2", pos=(500, 100))
    node2.set_code('''
@node_entry
def test_function_2(input_val: str):
    print(input_val)
''')
    
    print(f"Initial state:")
    print(f"  Nodes: {len(graph.nodes)}")
    print(f"  Scene items: {len(graph.items())}")
    
    # Create connections: node1 -> reroute -> node2
    print("Creating connections...")
    if node1.output_pins and node2.input_pins:
        # Connect node1 to reroute
        conn1 = graph.create_connection(node1.output_pins[0], reroute.input_pin)
        print(f"Created connection: node1 -> reroute")
        
        # Connect reroute to node2
        conn2 = graph.create_connection(reroute.output_pin, node2.input_pins[0])
        print(f"Created connection: reroute -> node2")
    
    print(f"After connections:")
    print(f"  Nodes: {len(graph.nodes)}")
    print(f"  Connections: {len(graph.connections)}")
    print(f"  Scene items: {len(graph.items())}")
    
    # Verify RerouteNode structure