我有一个异步函数需要测试,这个函数使用mongodb::Database
对象来运行,所以我在setup()
函数中初始化连接,并使用tokio_test::block_on()
将await
表达式 Package 在里面。
# [cfg(test)]
mod tests {
use mongodb::{options::ClientOptions, Client};
use tokio_test;
async fn setup() -> mongodb::Database {
tokio_test::block_on(async {
let client_uri = "mongodb://127.0.0.1:27017";
let options = ClientOptions::parse(&client_uri).await;
let client_result = Client::with_options(options.unwrap());
let client = client_result.unwrap();
client.database("my_database")
})
}
#[test]
fn test_something_async() {
// for some reason, test cannot be async
let DB = setup(); // <- the DB is impl std::future::Future type
// the DB variable will be used to run another
// async function named "some_async_func"
// but it doesn't work since DB is a Future type
// Future type need await keyword
// but if I use await-async keywords here, it complains
// error: async functions cannot be used for tests
// so what to do here ?
some_async_func(DB);
}
}
1条答案
按热度按时间xzv2uavs1#
只需在任何测试函数之前将
#[test]
替换为#[tokio::test]
即可。如果使用actix-web,则可以在测试函数之前将actix_rt添加到Cargo.toml
和#[actix_rt::test]
中