Use dotenvy for tests, so you can set DATABASE_URL in .env.

Author: duckinator

.gitignore

@@ -1,2 +1,3 @@
 target
 .DS_Store
+.env

Cargo.lock

@@ -24,6 +24,7 @@ tokio-postgres = "0.7"
 
 [dev-dependencies]
 chrono = "0.4"
+dotenvy = "0.15"
 macrotest = "1.1"
 serial_test = "3.2"
 tokio = { version = "1.43", features = ["full"] }

Cargo.toml

@@ -17,6 +17,9 @@ mod bucket_size {
 use bucket_size::*;
 
 static DB_POOL: std::sync::LazyLock<std::sync::Arc<deadpool_postgres::Pool>> = std::sync::LazyLock::new(|| {
+    if std::path::Path::new(".env").exists() {
+        dotenvy::dotenv().unwrap();
+    }
     let url = std::env::var("DATABASE_URL").unwrap_or("postgresql://localhost:5432/postgres".to_string());
     let pg_config = tokio_postgres::Config::from_str(&url).unwrap();
     let mgr_config = deadpool_postgres::ManagerConfig { recycling_method: deadpool_postgres::RecyclingMethod::Fast };

benches/bucket_size.rs

@@ -18,6 +18,9 @@ mod comparison {
 use comparison::*;
 
 static DB_POOL: std::sync::LazyLock<std::sync::Arc<deadpool_postgres::Pool>> = std::sync::LazyLock::new(|| {
+    if std::path::Path::new(".env").exists() {
+        dotenvy::dotenv().unwrap();
+    }
     let url = std::env::var("DATABASE_URL").unwrap_or("postgresql://localhost:5432/postgres".to_string());
     let pg_config = tokio_postgres::Config::from_str(&url).unwrap();
     let mgr_config = deadpool_postgres::ManagerConfig { recycling_method: deadpool_postgres::RecyclingMethod::Fast };

benches/comparison.rs

@@ -17,6 +17,9 @@ mod float {
 use float::*;
 
 static DB_POOL: std::sync::LazyLock<std::sync::Arc<deadpool_postgres::Pool>> = std::sync::LazyLock::new(|| {
+    if std::path::Path::new(".env").exists() {
+        dotenvy::dotenv().unwrap();
+    }
     let url = std::env::var("DATABASE_URL").unwrap_or("postgresql://localhost:5432/postgres".to_string());
     let pg_config = tokio_postgres::Config::from_str(&url).unwrap();
     let mgr_config = deadpool_postgres::ManagerConfig { recycling_method: deadpool_postgres::RecyclingMethod::Fast };

benches/float.rs

@@ -11,6 +11,9 @@ fn macrotest() {
 }
 
 pub static DB_POOL: std::sync::LazyLock<std::sync::Arc<deadpool_postgres::Pool>> = std::sync::LazyLock::new(|| {
+    if std::path::Path::new(".env").exists() {
+        dotenvy::dotenv().unwrap();
+    }
     let url = std::env::var("DATABASE_URL").unwrap_or("postgresql://localhost:5432/postgres".to_string());
     let pg_config = tokio_postgres::Config::from_str(&url).unwrap();
     let mgr_config = deadpool_postgres::ManagerConfig { recycling_method: deadpool_postgres::RecyclingMethod::Fast };