diff --git a/lib/crates/fabro-server/src/server.rs b/lib/crates/fabro-server/src/server.rs index 4b9dbcefa..64695ba5a 100644 --- a/lib/crates/fabro-server/src/server.rs +++ b/lib/crates/fabro-server/src/server.rs @@ -2057,7 +2057,7 @@ mod tests { #[tokio::test(flavor = "multi_thread", worker_threads = 2)] async fn run_completes_and_status_is_completed() { - let state = create_app_state(test_db().await); + let state = create_app_state_with_options(test_db().await, dry_run_settings(), 5); let app = test_app_with_scheduler(state); // Start a run @@ -2235,7 +2235,7 @@ mod tests { #[tokio::test(flavor = "multi_thread", worker_threads = 2)] async fn aggregate_usage_increments_after_run_completes() { - let state = create_app_state(test_db().await); + let state = create_app_state_with_options(test_db().await, dry_run_settings(), 5); let app = test_app_with_scheduler(state); // Start a run diff --git a/lib/crates/fabro-server/tests/it/api.rs b/lib/crates/fabro-server/tests/it/api.rs index 3614d9718..dc1b9308c 100644 --- a/lib/crates/fabro-server/tests/it/api.rs +++ b/lib/crates/fabro-server/tests/it/api.rs @@ -773,7 +773,7 @@ mod serve_dry_run { use axum::body::Body; use axum::http::{Request, StatusCode}; - use fabro_server::server::{build_router, create_app_state}; + use fabro_server::server::{build_router, create_app_state_with_options}; use tower::ServiceExt; const MINIMAL_DOT: &str = r#"digraph Test { @@ -785,7 +785,14 @@ mod serve_dry_run { /// Build the router exactly as `serve_command` does in dry-run mode. async fn dry_run_app() -> axum::Router { - let state = create_app_state(test_db().await); + let state = create_app_state_with_options( + test_db().await, + fabro_config::FabroSettings { + dry_run: Some(true), + ..Default::default() + }, + 5, + ); fabro_server::server::spawn_scheduler(Arc::clone(&state)); build_router(state, fabro_server::jwt_auth::AuthMode::Disabled) }