stefiosif/loft
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix(backend): skip auth rate limiter in test env

Browse Source
This commit is contained in:
stefiosif
2026-06-14 23:38:48 +03:00
parent 91459db5ec
commit be3f7da565
1 changed files with 25 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
backend/src/main.rs
View File

@@ -11,7 +11,6 @@ use axum::{
extract::DefaultBodyLimit, extract::DefaultBodyLimit,
http::{HeaderValue, Method, header}, http::{HeaderValue, Method, header},
middleware, middleware,
response::Response,
}; };
use sqlx::PgPool; use sqlx::PgPool;
use tower_cookies::CookieManagerLayer; use tower_cookies::CookieManagerLayer;
@@ -49,6 +48,19 @@ async fn main() -> Result<()> {
dotenvy::dotenv().ok(); dotenvy::dotenv().ok();
let database_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set");
const BODY_LIMIT: usize = 1000 * 1000 * 1000 * 5;
let pool = PgPool::connect(&database_url).await?;
sqlx::migrate!().run(&pool).await?;
let file_repository = FileRepository::new(pool.clone())?;
let routes_file = routes_file(file_repository.clone())
.route_layer(middleware::from_fn(mw_require_auth))
.layer(DefaultBodyLimit::max(BODY_LIMIT));
let user_repository = UserRepository::new(pool);
let mut routes_auth = routes_auth(user_repository.clone());
if std::env::var("ENVIRONMENT").unwrap_or_default() != "test" {
let governor_conf_auth = GovernorConfigBuilder::default() let governor_conf_auth = GovernorConfigBuilder::default()
.per_second(4) .per_second(4)
.burst_size(2) .burst_size(2)
@@ -69,18 +81,8 @@ async fn main() -> Result<()> {
} }
}); });
let database_url = std::env::var("DATABASE_URL").expect("DATABASE_URL must be set"); routes_auth = routes_auth.layer(GovernorLayer::new(governor_conf_auth));
const BODY_LIMIT: usize = 1000 * 1000 * 1000 * 5; }
let pool = PgPool::connect(&database_url).await?;
sqlx::migrate!().run(&pool).await?;
let file_repository = FileRepository::new(pool.clone())?;
let routes_file = routes_file(file_repository.clone())
.route_layer(middleware::from_fn(mw_require_auth))
.layer(DefaultBodyLimit::max(BODY_LIMIT));
let user_repository = UserRepository::new(pool);
let routes_auth =
routes_auth(user_repository.clone()).layer(GovernorLayer::new(governor_conf_auth));
let app = Router::new() let app = Router::new()
.nest("/api", routes_file) .nest("/api", routes_file)
@@ -114,4 +116,3 @@ async fn main() -> Result<()> {
Ok(()) Ok(())
} }