fix tests

DefGuard · Aug 17, 2023 · 3704850 · 3704850
1 parent ced94ca
commit 3704850
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src/config.rs b/src/config.rs
@@ -207,6 +207,7 @@ impl DefGuardConfig {
     pub fn new_test_config() -> Self {
         let mut config = Self::parse_from::<[_; 0], String>([]);
         config.validate_rp_id();
+        config.validate_cookie_domain();
         config
     }
 

diff --git a/tests/common/mod.rs b/tests/common/mod.rs
@@ -6,7 +6,7 @@ use defguard::{
     db::{init_db, AppEvent, DbPool, GatewayEvent, User},
     grpc::{GatewayMap, WorkerState},
 };
-use defguard::{db::UserDetails, mail::Mail};
+use defguard::{db::UserDetails, mail::Mail, SERVER_CONFIG};
 use rocket::http::Status;
 use rocket::local::asynchronous::Client;
 use sqlx::{postgres::PgConnectOptions, query, types::Uuid};
@@ -17,6 +17,7 @@ use tokio::sync::mpsc::unbounded_channel;
 
 pub async fn init_test_db() -> (DbPool, DefGuardConfig) {
     let config = DefGuardConfig::new_test_config();
+    let _ = SERVER_CONFIG.set(config.clone());
     let opts = PgConnectOptions::new()
         .host(&config.database_host)
         .port(config.database_port)