d-engine-server 0.2.3

Production-ready Raft consensus engine server and runtime
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77

//! Test helpers for Node unit tests
//!
//! Provides reusable fixtures and utilities to avoid code duplication
//! across Node test suites.

#[cfg(test)]
use std::sync::Arc;

use d_engine_core::RaftNodeConfig;
use d_engine_proto::common::NodeRole::Follower;
use d_engine_proto::common::NodeStatus;
use d_engine_proto::server::cluster::NodeMeta;
use tokio::sync::watch;

use crate::node::Node;
use crate::test_utils::MockBuilder;

/// Creates a test Node with minimal setup for unit testing
///
/// Returns (Node, shutdown_tx) where shutdown_tx can be used to trigger graceful shutdown
#[cfg(test)]
pub fn create_test_node() -> (Node<d_engine_core::MockTypeConfig>, watch::Sender<()>) {
    let (shutdown_tx, shutdown_rx) = watch::channel(());

    let node = MockBuilder::new(shutdown_rx.clone())
        .with_node_config({
            let mut config = RaftNodeConfig::new()
                .expect("Default config")
                .validate()
                .expect("Validate RaftNodeConfig successfully");
            config.cluster.initial_cluster = vec![NodeMeta {
                id: 1,
                address: "127.0.0.1:9001".to_string(),
                role: Follower as i32,
                status: NodeStatus::Active as i32,
            }];
            config
        })
        .build_node();

    (node, shutdown_tx)
}

/// Creates a test Node with custom node_id
#[cfg(test)]
pub fn create_test_node_with_id(
    node_id: u32
) -> (Node<d_engine_core::MockTypeConfig>, watch::Sender<()>) {
    let (shutdown_tx, shutdown_rx) = watch::channel(());

    let node = MockBuilder::new(shutdown_rx.clone())
        .id(node_id)
        .with_node_config({
            let mut config = RaftNodeConfig::new()
                .expect("Default config")
                .validate()
                .expect("Validate RaftNodeConfig successfully");
            config.cluster.node_id = node_id;
            config.cluster.initial_cluster = vec![NodeMeta {
                id: node_id,
                address: format!("127.0.0.1:{}", 9000 + node_id),
                role: Follower as i32,
                status: NodeStatus::Active as i32,
            }];
            config
        })
        .build_node();

    (node, shutdown_tx)
}

/// Creates a test Node wrapped in Arc for concurrent testing
#[cfg(test)]
pub fn create_test_node_arc() -> (Arc<Node<d_engine_core::MockTypeConfig>>, watch::Sender<()>) {
    let (node, shutdown_tx) = create_test_node();
    (Arc::new(node), shutdown_tx)
}