Function test_remove_node

tests/rust_integration_tests/graph_tests.rs:93–112

()

#[tokio::test]
async fn test_remove_node() {
    let mut graph = WorkflowGraph::new();

    let node = WorkflowNode::new(
        "Existing Node",
        "A node that exists",
        NodeType::Agent {
            config: AgentNodeConfig::new(AgentId::new(), "Test prompt"),
        },
    );

    let node_id = node.id.clone();
    graph.add_node(node).unwrap();

    assert_eq!(graph.node_count(), 1);

    graph.remove_node(&node_id).unwrap();
    assert_eq!(graph.node_count(), 0);
    assert!(graph.get_node(&node_id).is_none());
}

#[tokio::test]
async fn test_add_edge() {