/*
    Stress test
*/
use ppl::{prelude::*, templates::misc::Parallel};

struct Source {
    streamlen: usize,
    counter: usize,
}
impl Out<usize> for Source {
    fn run(&mut self) -> Option<usize> {
        if self.counter < self.streamlen {
            self.counter += 1;
            Some(self.counter)
        } else {
            None
        }
    }
}

pub fn fibonacci_recursive(n: usize) -> usize {
    match n {
        0 => panic!("zero is not a right argument to fibonacci_reccursive()!"),
        1 | 2 => 1,
        3 => 2,
        _ => fibonacci_recursive(n - 1) + fibonacci_recursive(n - 2),
    }
}

struct Sink {
    counter: usize,
}
impl In<usize, usize> for Sink {
    fn run(&mut self, _input: usize) {
        self.counter += 1;
    }

    fn finalize(self) -> Option<usize> {
        Some(self.counter)
    }
}

#[test]
fn test_stress() {
    env_logger::init();

    for _i in 0..10000 {
        let mut p = pipeline![
            Source {
                streamlen: 20,
                counter: 0
            },
            Parallel::build(2, fibonacci_recursive),
            Sink { counter: 0 }
        ];

        p.start();
        let res = p.wait_end().unwrap();
        assert_eq!(res, 20);
    }

    unsafe { Orchestrator::delete_global_orchestrator() }
}