HoovesSocial/hippofish
2
0
Fork 0
Code Issues 3 Pull requests 1 Projects Releases Packages Wiki Activity Actions

chore (backend-rs): more connection checks

Browse source
This commit is contained in:
naskya 2024-06-10 18:09:05 +09:00
parent dd54924a0a
commit 581668c306
No known key found for this signature in database
GPG key ID: 712D413B3A9FED5C
2 changed files with 60 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

33
packages/backend-rs/src/database/postgresql.rs
View file

@ -41,8 +41,35 @@ mod unit_test {
use super::get_conn; use super::get_conn;
#[tokio::test] #[tokio::test]
async fn connect() { async fn connect_sequential() {
assert!(get_conn().await.is_ok()); get_conn().await.unwrap();
assert!(get_conn().await.is_ok()); get_conn().await.unwrap();
get_conn().await.unwrap();
get_conn().await.unwrap();
get_conn().await.unwrap();
}
#[tokio::test]
async fn connect_concurrent() {
let [c1, c2, c3, c4, c5] = [
get_conn(),
get_conn(),
get_conn(),
get_conn(),
get_conn(),
];
let _ = tokio::try_join!(c1, c2, c3, c4, c5).unwrap();
}
#[tokio::test]
async fn connect_spawn() {
let mut tasks = Vec::new();
for _ in 0..5 {
tasks.push(tokio::spawn(get_conn()));
}
for task in tasks {
task.await.unwrap().unwrap();
}
} }
} }

33
packages/backend-rs/src/database/redis.rs
View file

@ -119,9 +119,36 @@ mod unit_test {
use redis::AsyncCommands; use redis::AsyncCommands;
#[tokio::test] #[tokio::test]
async fn connect() { async fn connect_sequential() {
assert!(get_conn().await.is_ok()); get_conn().await.unwrap();
assert!(get_conn().await.is_ok()); get_conn().await.unwrap();
get_conn().await.unwrap();
get_conn().await.unwrap();
get_conn().await.unwrap();
}
#[tokio::test]
async fn connect_concurrent() {
let [c1, c2, c3, c4, c5] = [
get_conn(),
get_conn(),
get_conn(),
get_conn(),
get_conn(),
];
let _ = tokio::try_join!(c1, c2, c3, c4, c5).unwrap();
}
#[tokio::test]
async fn connect_spawn() {
let mut tasks = Vec::new();
for _ in 0..5 {
tasks.push(tokio::spawn(get_conn()));
}
for task in tasks {
task.await.unwrap().unwrap();
}
} }
#[tokio::test] #[tokio::test]