use tokio;
pub struct AsyncCaller {
runtime: tokio::runtime::Runtime,
}
impl AsyncCaller {
pub fn new() -> Self {
Self {
runtime: tokio::runtime::Builder::new_multi_thread()
.enable_all()
.build()
.unwrap(),
}
}
pub fn call<F: Future>(&self, future: F) -> F::Output {
self.runtime.block_on(future)
}
}
#[cfg(test)]
mod tests {
use super::*;
async fn sum(a: u32, b: u32) -> u32 {
a + b
}
#[test]
fn test_single_async_call() {
assert_eq!(5, AsyncCaller::new().call(sum(2, 3)));
}
#[test]
fn test_two_async_calls() {
let caller = AsyncCaller::new();
assert_eq!(5, caller.call(sum(2, 3)));
assert_eq!(3, caller.call(sum(1, 2)));
}
#[test]
fn test_multiple_runtime_async_calls() {
let caller1 = AsyncCaller::new();
let caller2 = AsyncCaller::new();
assert_eq!(5, caller1.call(sum(2, 3)));
assert_eq!(3, caller2.call(sum(1, 2)));
}
}