unknownrori-simple-thread-pool 0.2.0

#[cfg(feature = "mpsc")]
#[cfg(test)]
mod mpsc {
    use std::sync::mpsc::channel;
    use std::{thread, time::Duration};

    use unknownrori_simple_thread_pool::{error::FailedToSendJob, ThreadPool};

    /// Test the crossbeam thread pooling implementation
    ///
    /// ## Panic
    ///
    /// It may panic if the OS cannot create a thread
    #[test]
    fn test_mpsc() -> Result<(), FailedToSendJob> {
        let pool = ThreadPool::new(2).unwrap();

        let (send, recv) = channel();

        for _ in 0..4 {
            let send = send.clone();

            pool.execute(move || {
                for _ in 0..40 {
                    // Simulate long process
                    thread::sleep(Duration::from_millis(10));
                }

                send.send(40).unwrap();
            })?;
        }

        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.try_recv().is_err(), true);

        Ok(())
    }

    #[test]
    #[should_panic]
    fn panic_inside_worker() {
        let pool = ThreadPool::new(2).unwrap();

        pool.execute(|| {
            panic!("Oh no!");
        })
        .unwrap();
    }
}

#[cfg(feature = "crossbeam")]
#[cfg(test)]
mod crossbeam {
    use std::{thread, time::Duration};

    use unknownrori_simple_thread_pool::{
        crossbeam_channel::unbounded, error::FailedToSendJob, ThreadPool,
    };

    /// Test the crossbeam thread pooling implementation
    ///
    /// ## Panic
    ///
    /// It may panic if the OS cannot create a thread
    #[test]
    fn test_crossbeam() -> Result<(), FailedToSendJob> {
        let pool = ThreadPool::new(2).unwrap();
        let (send, recv) = unbounded();

        for _ in 0..4 {
            let send = send.clone();

            pool.execute(move || {
                for _ in 0..40 {
                    // Simulate long process
                    thread::sleep(Duration::from_millis(10));
                }

                send.send(40).unwrap();
            })?;
        }

        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.recv().unwrap(), 40);
        assert_eq!(recv.try_recv().is_err(), true);

        Ok(())
    }

    #[test]
    #[should_panic]
    fn panic_inside_worker() {
        let pool = ThreadPool::new(2).unwrap();

        pool.execute(|| {
            panic!("Oh no!");
        })
        .unwrap();
    }
}