[router] migrate router from actix to axum (#8479)

sgl-router/tests/common/mod.rs

@@ -1,62 +1,2 @@
 pub mod mock_worker;
-
-use actix_web::web;
-use reqwest::Client;
-use sglang_router_rs::config::{PolicyConfig, RouterConfig, RoutingMode};
-use sglang_router_rs::server::AppState;
-
-/// Helper function to create test router configuration
-pub fn create_test_config(worker_urls: Vec<String>) -> RouterConfig {
-    RouterConfig {
-        mode: RoutingMode::Regular { worker_urls },
-        policy: PolicyConfig::Random,
-        host: "127.0.0.1".to_string(),
-        port: 3001,
-        max_payload_size: 256 * 1024 * 1024, // 256MB
-        request_timeout_secs: 600,
-        worker_startup_timeout_secs: 300,
-        worker_startup_check_interval_secs: 10,
-        dp_aware: false,
-        api_key: None,
-        discovery: None,
-        metrics: None,
-        log_dir: None,
-        log_level: None,
-        request_id_headers: None,
-    }
-}
-
-/// Helper function to create test router configuration with no health check
-pub fn create_test_config_no_workers() -> RouterConfig {
-    RouterConfig {
-        mode: RoutingMode::Regular {
-            worker_urls: vec![],
-        }, // Empty to skip health check
-        policy: PolicyConfig::Random,
-        host: "127.0.0.1".to_string(),
-        port: 3001,
-        max_payload_size: 256 * 1024 * 1024, // 256MB
-        request_timeout_secs: 600,
-        worker_startup_timeout_secs: 0, // No wait
-        worker_startup_check_interval_secs: 10,
-        dp_aware: false,
-        api_key: None,
-        discovery: None,
-        metrics: None,
-        log_dir: None,
-        log_level: None,
-        request_id_headers: None,
-    }
-}
-
-/// Helper function to create test app state
-pub async fn create_test_app_state(config: RouterConfig) -> Result<web::Data<AppState>, String> {
-    // Create a non-blocking client
-    let client = Client::builder()
-        .timeout(std::time::Duration::from_secs(config.request_timeout_secs))
-        .build()
-        .map_err(|e| e.to_string())?;
-
-    let app_state = AppState::new(config, client)?;
-    Ok(web::Data::new(app_state))
-}
+pub mod test_app;

sgl-router/tests/common/test_app.rs

@@ -0,0 +1,42 @@
+use axum::Router;
+use reqwest::Client;
+use sglang_router_rs::{
+    config::RouterConfig,
+    routers::RouterTrait,
+    server::{build_app, AppState},
+};
+use std::sync::Arc;
+
+/// Create a test Axum application using the actual server's build_app function
+pub fn create_test_app(
+    router: Arc<dyn RouterTrait>,
+    client: Client,
+    router_config: &RouterConfig,
+) -> Router {
+    // Create AppState with the test router
+    let app_state = Arc::new(AppState {
+        router,
+        client,
+        _concurrency_limiter: Arc::new(tokio::sync::Semaphore::new(
+            router_config.max_concurrent_requests,
+        )),
+    });
+
+    // Configure request ID headers (use defaults if not specified)
+    let request_id_headers = router_config.request_id_headers.clone().unwrap_or_else(|| {
+        vec![
+            "x-request-id".to_string(),
+            "x-correlation-id".to_string(),
+            "x-trace-id".to_string(),
+            "request-id".to_string(),
+        ]
+    });
+
+    // Use the actual server's build_app function
+    build_app(
+        app_state,
+        router_config.max_payload_size,
+        request_id_headers,
+        router_config.cors_allowed_origins.clone(),
+    )
+}