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

Add proper database pool initialization

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

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

@ -9,8 +9,23 @@ use diesel_async::RunQueryDsl;
use handlers::metrics::update_metric_handler::{update_metric_handler, UpdateMetricRequest}; use handlers::metrics::update_metric_handler::{update_metric_handler, UpdateMetricRequest};
use middleware::{AuthenticatedUser, OrganizationMembership}; use middleware::{AuthenticatedUser, OrganizationMembership};
use serde_json::json; use serde_json::json;
use std::sync::Once;
use uuid::Uuid; use uuid::Uuid;
// Used to initialize the database pool once for all tests
static INIT: Once = Once::new();
// Initialize database pool
fn initialize() {
INIT.call_once(|| {
let rt = tokio::runtime::Runtime::new().unwrap();
if let Err(e) = rt.block_on(init_pools()) {
panic!("Failed to initialize test pools: {}", e);
}
println!("Database pool initialized");
});
}
// Create a simplified test setup for our test // Create a simplified test setup for our test
struct TestSetup { struct TestSetup {
pub user: AuthenticatedUser, pub user: AuthenticatedUser,
@ -21,8 +36,8 @@ struct TestSetup {
// Integration test that tests metric status update functionality // Integration test that tests metric status update functionality
#[tokio::test] #[tokio::test]
async fn test_update_metric_status() -> Result<()> { async fn test_update_metric_status() -> Result<()> {
// Get database pool - it's already initialized // Initialize the database pool
let _ = get_pg_pool(); initialize();
// Create a test ID for unique naming // Create a test ID for unique naming
let test_id = format!("test-{}", Uuid::new_v4()); let test_id = format!("test-{}", Uuid::new_v4());
@ -276,8 +291,8 @@ fn create_default_chart_config() -> database::types::metric_yml::ChartConfig {
// Test unauthorized access // Test unauthorized access
#[tokio::test] #[tokio::test]
async fn test_update_metric_status_unauthorized() -> Result<()> { async fn test_update_metric_status_unauthorized() -> Result<()> {
// Get database pool - it's already initialized // Initialize the database pool
let _ = get_pg_pool(); initialize();
// Create a test ID for unique naming // Create a test ID for unique naming
let test_id = format!("test-{}", Uuid::new_v4()); let test_id = format!("test-{}", Uuid::new_v4());
@ -496,8 +511,8 @@ async fn test_update_metric_status_unauthorized() -> Result<()> {
// Test edge cases for status updates // Test edge cases for status updates
#[tokio::test] #[tokio::test]
async fn test_update_metric_status_null_value() -> Result<()> { async fn test_update_metric_status_null_value() -> Result<()> {
// Get database pool - it's already initialized // Initialize the database pool
let _ = get_pg_pool(); initialize();
// Create a test ID for unique naming // Create a test ID for unique naming
let test_id = format!("test-{}", Uuid::new_v4()); let test_id = format!("test-{}", Uuid::new_v4());