mod fixtures;

use fixtures::{init_log, MemRouter, MemVoteFactor, NodeId};
use std::sync::Arc;
use std::thread::sleep;
use std::time::Duration;
use tender::{InitialMode, Options, State};

/// Single-node initialization test.
#[test]
fn test_single_node() {
    init_log();

    let mem_router = Arc::new(MemRouter::new(1));
    let node = NodeId::new(1, 1);
    mem_router.new_node(node, MemVoteFactor::new(0));

    sleep(Duration::from_secs(1));
    mem_router.assert_node_state(node, State::Startup, 0, None);

    let options = Options::builder()
        .election_timeout_min(1000)
        .election_timeout_max(1100)
        .heartbeat_interval(300)
        .build()
        .unwrap();
    mem_router.update_node_options(node, options);

    sleep(Duration::from_secs(1));
    mem_router.assert_node_state(node, State::Startup, 0, None);

    mem_router.init_node(node, Vec::new(), InitialMode::Normal);
    sleep(Duration::from_secs(1));
    mem_router.assert_node_state(node, State::Leader, 1, Some(node));
}

/// Single-node initialization test: observer.
#[test]
fn test_single_node_as_observer() {
    init_log();

    let mem_router = Arc::new(MemRouter::new(1));
    let node = NodeId::new(1, 1);
    mem_router.new_node(node, MemVoteFactor::new(0));

    sleep(Duration::from_secs(1));
    mem_router.assert_node_state(node, State::Startup, 0, None);

    let options = Options::builder()
        .election_timeout_min(1000)
        .election_timeout_max(1100)
        .heartbeat_interval(300)
        .build()
        .unwrap();
    mem_router.update_node_options(node, options);

    sleep(Duration::from_secs(1));
    mem_router.assert_node_state(node, State::Startup, 0, None);

    mem_router.init_node(node, Vec::new(), InitialMode::AsObserver);
    sleep(Duration::from_secs(1));
    mem_router.assert_node_state(node, State::Observer, 0, None);
}