task_scope 0.1.1

use futures::pin_mut;
use std::time::Duration;
use tokio::io::*;
use tokio::time::timeout;

use task_scope::{cancelable, scope, spawn};

#[test]
fn test_infinite() {
    // this thread never ends as there is a task loops infinitely
    std::thread::spawn(|| {
        let mut rt = tokio::runtime::Runtime::new().unwrap();
        rt.block_on(async {
            timeout(
                Duration::from_millis(100),
                scope(async {
                    println!("parent started");
                    spawn(async {
                        println!("infinite loop");
                        let mut fast_read = repeat(0);
                        let mut fast_write = sink();

                        copy(&mut fast_read, &mut fast_write).await.unwrap();

                        panic!("never ends");
                    })
                    .await;

                    println!("parent end");
                }),
            )
            .await
            .unwrap_err();
        });
    });
}

#[tokio::test(threaded_scheduler)]
async fn test_cancel_infinite() {
    timeout(
        Duration::from_millis(100),
        scope(async {
            println!("parent started");
            spawn(async {
                let fast_read = cancelable(repeat(0));
                pin_mut!(fast_read); // for Unpin bound of copy
                let mut fast_write = sink();

                copy(&mut fast_read, &mut fast_write).await.unwrap_err();

                println!("copy is canceled (after parent is dropped)");
            })
            .await;

            println!("parent end");
        }),
    )
    .await
    .unwrap_err();

    println!("canceled");
}