mm1-sup 0.7.23

use std::fmt;
use std::time::Duration;

use test_case::test_case;

use super::*;

enum Action<K = usize> {
    Add(K, ChildType),
    Rm(K),
    Decide,
    Started(K, Address),
    Exited(Address, bool),
    Delay(Duration),
}

use Action::*;

#[test_case(
    Default::default(),
    None::<Action<usize>>
    ; "empty"
)]
#[test_case(
    Default::default(),
    [
        Add("one", ChildType::Permanent),
        Rm("one"),
        Decide,
    ]
    ; "add, remove, decide later"
)]
#[test_case(
    Default::default(),
    [
        Add("one", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Rm("one"),
        Decide,
        Exited(Address::from_u64(1), false),
        Decide,
    ]
    ; "add, started, remove"
)]
#[test_case(
    Default::default(),
    [
        Add("one", ChildType::Permanent),
        Add("two", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Decide,
        Started("two", Address::from_u64(2)),
        Decide,
    ]
    ; "add two, started, remove first"
)]
#[test_case(
    Default::default(),
    [
        Add("one", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Decide,
        Exited(Address::from_u64(1), false),
        Decide,
    ]
    ; "add, started, exited, max restart intensity reached"
)]
#[test_case(
    Default::default(),
    [
        Add("one", ChildType::Permanent),
        Add("two", ChildType::Permanent),
        Add("three", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Started("two", Address::from_u64(2)),
        Started("three", Address::from_u64(3)),
        Decide,
        Exited(Address::from_u64(1), false),
        Decide,
        Exited(Address::from_u64(2), false),
        Exited(Address::from_u64(3), false),
        Decide,
    ]
    ; "add several, started, exited, max restart intensity reached"
)]
#[test_case(
    RestartIntensity { max_restarts: 1, within: Duration::from_secs(30) },
    [
        Add("one", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Decide,
        Delay(Duration::from_secs(40)),
        Exited(Address::from_u64(1), false),
        Decide,
        Started("one", Address::from_u64(2)),
    ]
    ; "add, started, exited, max restart intensity not reached"
)]
#[test_case(
    RestartIntensity { max_restarts: 1, within: Duration::from_secs(30) },
    [
        Add("one", ChildType::Permanent),
        Add("two", ChildType::Permanent),
        Add("three", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Started("two", Address::from_u64(2)),
        Started("three", Address::from_u64(3)),
        Decide,
        Exited(Address::from_u64(1), false),
        Decide,
        Exited(Address::from_u64(3), false),
        Exited(Address::from_u64(2), false),
        Decide,
        Started("one", Address::from_u64(4)),
        Started("two", Address::from_u64(5)),
        Started("three", Address::from_u64(6)),
        Decide,
        Exited(Address::from_u64(5), false),
        Decide,
        Exited(Address::from_u64(6), false),
        Exited(Address::from_u64(4), false),
        Decide,
    ]
    ; "add 3, started 3, exited 2, max restart intensity reached"
)]
#[test_case(
    RestartIntensity { max_restarts: 1, within: Duration::from_secs(30) },
    [
        Add("one", ChildType::Permanent),
        Add("two", ChildType::Permanent),
        Add("three", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Started("two", Address::from_u64(2)),
        Started("three", Address::from_u64(3)),
        Decide,
        Exited(Address::from_u64(1), false),
        Decide,
        Exited(Address::from_u64(3), false),
        Exited(Address::from_u64(2), false),
        Decide,
        Started("one", Address::from_u64(4)),
        Started("two", Address::from_u64(5)),
        Started("three", Address::from_u64(6)),
        Decide,
        Delay(Duration::from_secs(40)),
        Exited(Address::from_u64(5), false),
        Decide,
        Exited(Address::from_u64(6), false),
        Exited(Address::from_u64(4), false),
        Decide,
        Started("one", Address::from_u64(7)),
        Started("two", Address::from_u64(8)),
        Started("three", Address::from_u64(9)),
        Decide,
    ]
    ; "add 3, started 3, exited 2, max restart intensity not reached"
)]
#[test_case(
    RestartIntensity { max_restarts: 1, within: Duration::from_secs(30) },
    [
        Add("one", ChildType::Permanent),
        Decide,
        Started("one", Address::from_u64(1)),
        Decide,
        Started("one", Address::from_u64(2)),
        Decide,
    ]
    ; "an orphan"
)]
#[tokio::test]
async fn run<K>(restart_intensity: RestartIntensity, script: impl IntoIterator<Item = Action<K>>)
where
    AllForOne<K>: RestartStrategy<K>,
    K: fmt::Display + Clone + Eq,
{
    let _ = mm1_logger::init(&logger_config());
    tokio::time::pause();

    let snapshot_name = std::thread::current()
        .name()
        .expect("empty thread name")
        .to_owned();
    let mut report = vec![];

    report.push(restart_intensity.to_string());
    let mut decider = AllForOne::<K>::new(restart_intensity).decider();

    for action in script {
        report.push(action.to_string());
        let result = match action {
            Delay(d) => {
                tokio::time::sleep(d).await;
                Ok(())
            },
            Started(key, addr) => {
                decider.started(&key, addr, tokio::time::Instant::now());
                Ok(())
            },
            Exited(addr, normal_exit) => {
                decider.exited(addr, normal_exit, tokio::time::Instant::now());
                Ok(())
            },
            Add(key, ty) => decider.add(key, ty),
            Rm(key) => decider.rm(&key),
            Decide => {
                let mut noop_counter = 0;
                loop {
                    match decider.next_action(tokio::time::Instant::now()) {
                        Err(reason) => break Err(reason),
                        Ok(None) => break Ok(()),
                        Ok(Some(super::Action::Noop)) => {
                            noop_counter += 1;
                            if noop_counter > 10 {
                                report.push("too many Noops: giving up...".into());
                                break Ok(())
                            }
                        },
                        Ok(Some(action)) => report.push(format!(">>> {action}")),
                    }
                }
            },
        };
        if let Err(reason) = result {
            report.push(format!("!!! ERROR: {reason}"));
            break
        }
    }

    insta::assert_yaml_snapshot!(snapshot_name, report);
}

impl<K> fmt::Display for Action<K>
where
    K: fmt::Display,
{
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Add(k, ty) => write!(f, "ADD     [{k}|{ty:?}]"),
            Self::Rm(k) => write!(f, "RM      [{k}]"),
            Self::Decide => write!(f, "DECIDE"),
            Self::Started(k, a) => write!(f, "STARTED [{k}] / {a}"),
            Self::Exited(a, n) => write!(f, "EXITED  {a} normal_exit={n}"),
            Self::Delay(d) => write!(f, "DELAY {d:?}"),
        }
    }
}

fn logger_config() -> mm1_logger::LoggingConfig {
    use mm1_logger::*;

    LoggingConfig {
        min_log_level:     Level::TRACE,
        log_target_filter: vec![
            // "mm1_node::runtime::local_system::protocol_actor=trace".parse().unwrap(),
            "mm1_sup::*=DEBUG".parse().unwrap(),
            "*=INFO".parse().unwrap(),
        ],
    }
}

// --- Regression tests for #126 -------------------------------------------
//
// An "acceptable" exit must not restart the child: a Temporary child that exits
// for any reason, and a Transient child that exits normally, must stay stopped.

fn assert_not_restarted(decider: &mut impl Decider<Key = usize>, at: tokio::time::Instant) {
    for _ in 0..16 {
        match decider.next_action(at) {
            Ok(Some(super::Action::Start { .. })) => panic!("child was restarted"),
            Ok(None) => return,
            Ok(Some(_)) => continue,
            Err(_) => return,
        }
    }
    // Reached the cap with no Start action: no restart happened.
}

#[tokio::test]
async fn temporary_child_is_not_restarted_after_any_exit() {
    tokio::time::pause();
    let at = tokio::time::Instant::now();
    let mut decider = AllForOne::<usize>::new(RestartIntensity {
        max_restarts: 100,
        within:       Duration::from_secs(30),
    })
    .decider();

    decider.add(1, ChildType::Temporary).unwrap();
    assert!(matches!(
        decider.next_action(at),
        Ok(Some(super::Action::Start { .. }))
    ));
    let addr = Address::from_u64(1);
    decider.started(&1, addr, at);
    // Abnormal exit is still "acceptable" for a Temporary child.
    decider.exited(addr, false, at);
    assert_not_restarted(&mut decider, at);
}

#[tokio::test]
async fn transient_child_is_not_restarted_after_normal_exit() {
    tokio::time::pause();
    let at = tokio::time::Instant::now();
    let mut decider = AllForOne::<usize>::new(RestartIntensity {
        max_restarts: 100,
        within:       Duration::from_secs(30),
    })
    .decider();

    decider.add(1, ChildType::Transient).unwrap();
    assert!(matches!(
        decider.next_action(at),
        Ok(Some(super::Action::Start { .. }))
    ));
    let addr = Address::from_u64(1);
    decider.started(&1, addr, at);
    decider.exited(addr, true, at); // normal exit
    assert_not_restarted(&mut decider, at);
}