Test refactor

This commit is contained in:
kdeng00
2025-04-26 16:26:10 -04:00
parent cdbff9c2c5
commit aad8f0d742
+5 -15
View File
@@ -117,9 +117,8 @@ mod tests {
use common_multipart_rfc7578::client::multipart::{ use common_multipart_rfc7578::client::multipart::{
Body as MultipartBody, Form as MultipartForm, Body as MultipartBody, Form as MultipartForm,
}; };
use std::{time::Duration, usize}; use std::{usize};
use tower::ServiceExt; use tower::ServiceExt;
use tower_http::timeout::TimeoutLayer;
mod db_mgr { mod db_mgr {
use std::str::FromStr; use std::str::FromStr;
@@ -199,12 +198,11 @@ mod tests {
mod init { mod init {
pub async fn app(pool: sqlx::PgPool) -> axum::Router { pub async fn app(pool: sqlx::PgPool) -> axum::Router {
let app = crate::init::routes() crate::init::routes()
.await .await
.layer(axum::Extension(pool)) .layer(axum::Extension(pool))
.layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024)) .layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024))
.layer(tower_http::timeout::TimeoutLayer::new(std::time::Duration::from_secs(300))); .layer(tower_http::timeout::TimeoutLayer::new(std::time::Duration::from_secs(300)))
app
} }
} }
@@ -278,11 +276,7 @@ mod tests {
let pool = db_mgr::connect_to_db(&db_name).await.unwrap(); let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
db::migrations(&pool).await; db::migrations(&pool).await;
let app = crate::init::routes() let app = init::app(pool).await;
.await
.layer(axum::Extension(pool))
.layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024))
.layer(TimeoutLayer::new(Duration::from_secs(300)));
// Create multipart form // Create multipart form
let mut form = MultipartForm::default(); let mut form = MultipartForm::default();
@@ -356,11 +350,7 @@ mod tests {
let pool = db_mgr::connect_to_db(&db_name).await.unwrap(); let pool = db_mgr::connect_to_db(&db_name).await.unwrap();
db::migrations(&pool).await; db::migrations(&pool).await;
let app = crate::init::routes() let app = init::app(pool).await;
.await
.layer(axum::Extension(pool))
.layer(axum::extract::DefaultBodyLimit::max(1024 * 1024 * 1024))
.layer(TimeoutLayer::new(Duration::from_secs(300)));
// Create multipart form // Create multipart form
let mut form = MultipartForm::default(); let mut form = MultipartForm::default();