Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix database pool initialization within async runtime

This commit is contained in:
dal 2025-04-07 16:28:37 -06:00
parent 1a6d712a3f
commit 9bb0410d03
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
api/libs/handlers/tests/metrics/update_metric_test.rs
View File

@ -16,14 +16,19 @@ use uuid::Uuid;
static INIT: Once = Once::new(); static INIT: Once = Once::new();
// Initialize database pool // Initialize database pool
fn initialize() { async fn initialize() {
INIT.call_once(|| { INIT.call_once(|| {
let rt = tokio::runtime::Runtime::new().unwrap(); println!("Database pool initialization called");
if let Err(e) = rt.block_on(init_pools()) { // Set environment variables for database connection
panic!("Failed to initialize test pools: {}", e); std::env::set_var("DATABASE_URL", "postgresql://postgres:postgres@127.0.0.1:54322/postgres");
} std::env::set_var("TEST_DATABASE_URL", "postgresql://postgres:postgres@127.0.0.1:54322/postgres");
println!("Database pool initialized");
}); });
// We only initialize the pools once but try to do it in each test to ensure they exist
match init_pools().await {
Ok(_) => println!("Database pool initialized successfully"),
Err(e) => println!("Database pool initialization error: {}", e),
}
} }
// Create a simplified test setup for our test // Create a simplified test setup for our test