[router] refactor router and worker management 3/n (#10727)

sgl-router/tests/common/test_app.rs

@@ -51,3 +51,39 @@ pub fn create_test_app(
         router_config.cors_allowed_origins.clone(),
     )
 }
+
+/// Create a test Axum application with an existing AppContext
+#[allow(dead_code)]
+pub fn create_test_app_with_context(
+    router: Arc<dyn RouterTrait>,
+    app_context: Arc<AppContext>,
+) -> Router {
+    // Create AppState with the test router and context
+    let app_state = Arc::new(AppState {
+        router,
+        context: app_context.clone(),
+        concurrency_queue_tx: None,
+        router_manager: None,
+    });
+
+    // Get config from the context
+    let router_config = &app_context.router_config;
+
+    // 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(),
+    )
+}