1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105
|
#![cfg(all(tokio_unstable, feature = "tracing"))]
use std::rc::Rc;
use tokio::task::{Builder, LocalSet};
#[tokio::test]
async fn spawn_with_name() {
let result = Builder::new()
.name("name")
.spawn(async { "task executed" })
.unwrap()
.await;
assert_eq!(result.unwrap(), "task executed");
}
#[tokio::test(flavor = "local")]
async fn spawn_local_on_local_runtime() {
let result = Builder::new()
.spawn_local(async { "task executed" })
.unwrap()
.await;
assert_eq!(result.unwrap(), "task executed");
}
#[tokio::test]
#[should_panic = "`spawn_local` called from outside of a `task::LocalSet` or `runtime::LocalRuntime`"]
async fn spawn_local_panics_outside_local_set_or_local_runtime() {
let _ = Builder::new()
.spawn_local(async { "task executed" })
.unwrap()
.await;
}
#[tokio::test(flavor = "multi_thread")]
#[should_panic = "`spawn_local` called from outside of a `task::LocalSet` or `runtime::LocalRuntime`"]
async fn spawn_local_panics_in_multi_thread_runtime() {
let _ = Builder::new()
.spawn_local(async { "task executed" })
.unwrap()
.await;
}
#[tokio::test]
async fn spawn_blocking_with_name() {
let result = Builder::new()
.name("name")
.spawn_blocking(|| "task executed")
.unwrap()
.await;
assert_eq!(result.unwrap(), "task executed");
}
#[tokio::test]
async fn spawn_local_with_name() {
let unsend_data = Rc::new("task executed");
let result = LocalSet::new()
.run_until(async move {
Builder::new()
.name("name")
.spawn_local(async move { unsend_data })
.unwrap()
.await
})
.await;
assert_eq!(*result.unwrap(), "task executed");
}
#[tokio::test]
async fn spawn_without_name() {
let result = Builder::new()
.spawn(async { "task executed" })
.unwrap()
.await;
assert_eq!(result.unwrap(), "task executed");
}
#[tokio::test]
async fn spawn_blocking_without_name() {
let result = Builder::new()
.spawn_blocking(|| "task executed")
.unwrap()
.await;
assert_eq!(result.unwrap(), "task executed");
}
#[tokio::test]
async fn spawn_local_without_name() {
let unsend_data = Rc::new("task executed");
let result = LocalSet::new()
.run_until(async move {
Builder::new()
.spawn_local(async move { unsend_data })
.unwrap()
.await
})
.await;
assert_eq!(*result.unwrap(), "task executed");
}
|