kreuzberg 4.7.4

//! Integration tests for the API module.

#![cfg(feature = "api")]

use axum::{
    body::Body,
    http::{Request, StatusCode},
};
use serde_json::json;
use tower::ServiceExt;

use kreuzberg::{
    ExtractionConfig,
    api::{HealthResponse, InfoResponse, create_router},
};

/// Test the health check endpoint.
#[tokio::test]
async fn test_health_endpoint() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .uri("/health")
                .body(Body::empty())
                .expect("Failed to create HTTP request body"),
        )
        .await
        .expect("Failed to send HTTP request");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let health: HealthResponse = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(health.status, "healthy");
    assert!(!health.version.is_empty());
}

/// Test the info endpoint.
#[tokio::test]
async fn test_info_endpoint() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .uri("/info")
                .body(Body::empty())
                .expect("Failed to create HTTP request body"),
        )
        .await
        .expect("Failed to send HTTP request");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let info: InfoResponse = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert!(!info.version.is_empty());
    assert!(info.rust_backend);
}

/// Test extract endpoint with no files returns 400.
#[tokio::test]
async fn test_extract_no_files() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let body_content = format!("--{}--\r\n", boundary);

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::BAD_REQUEST);
}

/// Test extract endpoint with a simple text file.
#[tokio::test]
async fn test_extract_text_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Hello, world!";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("Hello, world!")
    );

    assert!(
        results[0]["chunks"].is_null(),
        "Chunks should be null without chunking config"
    );
    assert!(
        results[0]["detected_languages"].is_null(),
        "Language detection not enabled"
    );
    assert!(results[0]["tables"].is_array(), "Tables field should be present");
    assert!(results[0]["metadata"].is_object(), "Metadata field should be present");
}

/// Test extract endpoint with JSON config.
#[tokio::test]
async fn test_extract_with_config() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Hello, world!";
    let config = json!({
        "force_ocr": false
    });

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"config\"\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary, config, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("Hello, world!")
    );

    assert!(
        results[0]["chunks"].is_null(),
        "Chunks should be null without chunking config"
    );
    assert!(
        results[0]["detected_languages"].is_null(),
        "Language detection not enabled"
    );
    assert!(results[0]["tables"].is_array(), "Tables field should be present");
    assert!(results[0]["metadata"].is_object(), "Metadata field should be present");
}

/// Test extract endpoint with invalid config returns 400.
#[tokio::test]
async fn test_extract_invalid_config() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Hello, world!";
    let invalid_config = "not valid json";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"config\"\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary, invalid_config, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::BAD_REQUEST);
}

/// Test extract endpoint with multiple files.
#[tokio::test]
async fn test_extract_multiple_files() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file1_content = "First file content";
    let file2_content = "Second file content";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test1.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test2.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file1_content, boundary, file2_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 2);
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("First file")
    );
    assert!(
        results[1]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("Second file")
    );

    for result in &results {
        assert!(
            result["chunks"].is_null(),
            "Chunks should be null without chunking config"
        );
        assert!(result["detected_languages"].is_null(), "Language detection not enabled");
        assert!(result["tables"].is_array(), "Tables field should be present");
        assert!(result["metadata"].is_object(), "Metadata field should be present");
        assert_eq!(result["mime_type"], "text/plain");
    }
}

/// Test extract endpoint with markdown content.
#[tokio::test]
async fn test_extract_markdown_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "# Heading\n\nSome **bold** text.";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.md\"\r\n\
         Content-Type: text/markdown\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/markdown");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("Heading")
    );
}

/// Test extract endpoint with JSON content.
#[tokio::test]
async fn test_extract_json_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = r#"{"key": "value", "number": 42}"#;

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.json\"\r\n\
         Content-Type: application/json\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "application/json");
}

/// Test extract endpoint with XML content.
#[tokio::test]
#[cfg(feature = "xml")]
async fn test_extract_xml_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = r#"<?xml version="1.0"?><root><item>test</item></root>"#;

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.xml\"\r\n\
         Content-Type: application/xml\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "application/xml");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("test")
    );
}

/// Test extract endpoint with HTML content.
#[tokio::test]
#[cfg(feature = "html")]
async fn test_extract_html_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = r#"<html><body><h1>Title</h1><p>Content</p></body></html>"#;

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.html\"\r\n\
         Content-Type: text/html\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/html");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("Title")
    );
}

/// Test extract endpoint with missing Content-Type header.
#[tokio::test]
async fn test_extract_missing_content_type() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .body(Body::from("some data"))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert!(response.status() == StatusCode::BAD_REQUEST || response.status() == StatusCode::UNSUPPORTED_MEDIA_TYPE);
}

/// Test extract endpoint with empty file.
#[tokio::test]
async fn test_extract_empty_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"empty.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
}

/// Test extract endpoint with unsupported MIME type returns 400.
#[tokio::test]
async fn test_extract_unsupported_mime_type() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Binary data";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.bin\"\r\n\
         Content-Type: application/x-unknown-binary\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::BAD_REQUEST);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let error: serde_json::Value = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(error["error_type"], "UnsupportedFormatError");
    assert_eq!(error["status_code"], 400);
}

/// Test extract endpoint with octet-stream content type and a known filename extension
/// auto-detects the MIME type from the filename.
#[tokio::test]
async fn test_extract_octet_stream_with_known_extension_detects_mime() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Hello, world!";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"readme.txt\"\r\n\
         Content-Type: application/octet-stream\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    // Should succeed because .txt is auto-detected from the filename
    assert_eq!(response.status(), StatusCode::OK);
}

/// Test extract endpoint with an explicit unsupported MIME type and a filename
/// returns 400 with UnsupportedFormatError (not 500).
#[tokio::test]
async fn test_extract_unsupported_mime_type_returns_400_not_500() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "some data";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"report.xyz\"\r\n\
         Content-Type: application/vnd.fake-unsupported-type\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::BAD_REQUEST);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let error: serde_json::Value = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(error["error_type"], "UnsupportedFormatError");
    assert_eq!(error["status_code"], 400);
    assert!(error["message"].as_str().unwrap_or("").contains("Unsupported format"));
}

/// Test extract endpoint without filename in multipart field.
#[tokio::test]
async fn test_extract_without_filename() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Test content";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);
}

/// Test extract endpoint with malformed multipart data.
#[tokio::test]
async fn test_extract_malformed_multipart() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let body_content = format!("--{}\r\nmalformed data\r\n", boundary);

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert!(response.status().is_client_error() || response.status().is_server_error());
}

/// Test CORS headers are present.
#[tokio::test]
async fn test_cors_headers() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .uri("/health")
                .body(Body::empty())
                .expect("Failed to create HTTP request body"),
        )
        .await
        .expect("Failed to send HTTP request");

    assert_eq!(response.status(), StatusCode::OK);

    let headers = response.headers();
    assert!(headers.contains_key("access-control-allow-origin") || headers.contains_key("Access-Control-Allow-Origin"));
}

/// Test OPTIONS preflight request for CORS.
#[tokio::test]
async fn test_cors_preflight() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .method("OPTIONS")
                .uri("/extract")
                .header("origin", "http://example.com")
                .header("access-control-request-method", "POST")
                .body(Body::empty())
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert!(response.status().is_success() || response.status() == StatusCode::NO_CONTENT);
}

/// Test error response format for validation errors.
#[tokio::test]
async fn test_error_response_format_validation() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let body_content = format!("--{}--\r\n", boundary);

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::BAD_REQUEST);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let error: serde_json::Value = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert!(error["error_type"].is_string());
    assert!(error["message"].is_string());
    assert_eq!(error["status_code"], 400);
    assert_eq!(error["error_type"], "ValidationError");
}

/// Test error response format for parsing errors.
#[tokio::test]
async fn test_error_response_format_parsing() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let invalid_config = "not valid json";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         content\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"config\"\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, boundary, invalid_config, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::BAD_REQUEST);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let error: serde_json::Value = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(error["error_type"], "ValidationError");
    assert!(
        error["message"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("configuration")
    );
}

/// Test 404 error for non-existent endpoint.
#[tokio::test]
async fn test_not_found_endpoint() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .uri("/nonexistent")
                .body(Body::empty())
                .expect("Failed to create HTTP request body"),
        )
        .await
        .expect("Failed to send HTTP request");

    assert_eq!(response.status(), StatusCode::NOT_FOUND);
}

/// Test extract endpoint with very large text content.
#[tokio::test]
async fn test_extract_large_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let large_content = "A".repeat(1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"large.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, large_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
}

/// Test concurrent requests to extract endpoint.
#[tokio::test]
async fn test_concurrent_requests() {
    use tower::ServiceExt;

    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Concurrent test";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let mut handles = vec![];

    for _ in 0..5 {
        let app_clone = app.clone();
        let body_clone = body_content.clone();

        let handle = tokio::spawn(async move {
            app_clone
                .oneshot(
                    Request::builder()
                        .method("POST")
                        .uri("/extract")
                        .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                        .body(Body::from(body_clone))
                        .expect("Operation failed"),
                )
                .await
        });

        handles.push(handle);
    }

    for handle in handles {
        let response = handle
            .await
            .expect("Async operation failed")
            .expect("Async operation failed");
        assert_eq!(response.status(), StatusCode::OK);
    }
}

/// Test cache stats endpoint.
#[tokio::test]
async fn test_cache_stats_endpoint() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .uri("/cache/stats")
                .body(Body::empty())
                .expect("Failed to create HTTP request body"),
        )
        .await
        .expect("Failed to send HTTP request");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let stats: serde_json::Value = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert!(stats["directory"].is_string());
    assert!(stats["total_files"].is_number());
    assert!(stats["total_size_mb"].is_number());
}

/// Test cache clear endpoint.
#[tokio::test]
async fn test_cache_clear_endpoint() {
    let app = create_router(ExtractionConfig::default());

    let response = app
        .oneshot(
            Request::builder()
                .method("DELETE")
                .uri("/cache/clear")
                .body(Body::empty())
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let clear_result: serde_json::Value = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert!(clear_result["directory"].is_string());
    assert!(clear_result["removed_files"].is_number());
    assert!(clear_result["freed_mb"].is_number());
}

/// Test extract endpoint with mixed content types.
#[tokio::test]
async fn test_extract_mixed_content_types() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let text_content = "Text file";
    let json_content = r#"{"test": "data"}"#;

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.json\"\r\n\
         Content-Type: application/json\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, text_content, boundary, json_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 2);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert_eq!(results[1]["mime_type"], "application/json");
}

/// Test extract endpoint with unknown field in multipart.
#[tokio::test]
async fn test_extract_unknown_multipart_field() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Test content";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"unknown_field\"\r\n\
         \r\n\
         some value\r\n\
         --{}--\r\n",
        boundary, file_content, boundary, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);
}

/// Test extract endpoint with default MIME type (application/octet-stream).
#[tokio::test]
async fn test_extract_default_mime_type() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "Test content";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert!(
        response.status() == StatusCode::OK
            || response.status() == StatusCode::UNPROCESSABLE_ENTITY
            || response.status() == StatusCode::INTERNAL_SERVER_ERROR
    );
}

/// Test size limits configuration with custom limits.
#[tokio::test]
async fn test_size_limits_custom_limits() {
    use kreuzberg::api::{ApiSizeLimits, create_router_with_limits};

    let config = ExtractionConfig::default();
    let limits = ApiSizeLimits::from_mb(50, 50);
    let app = create_router_with_limits(config, limits);

    assert_eq!(limits.max_request_body_bytes, 50 * 1024 * 1024);
    assert_eq!(limits.max_multipart_field_bytes, 50 * 1024 * 1024);

    let boundary = "----boundary";
    let file_content = "Test";

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"test.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK);
}

/// Test size limits with asymmetric limits (different request vs field sizes).
#[tokio::test]
async fn test_size_limits_asymmetric() {
    use kreuzberg::api::{ApiSizeLimits, create_router_with_limits};

    let config = ExtractionConfig::default();
    let limits = ApiSizeLimits::new(100 * 1024 * 1024, 50 * 1024 * 1024);
    let _app = create_router_with_limits(config, limits);

    assert_eq!(limits.max_request_body_bytes, 100 * 1024 * 1024);
    assert_eq!(limits.max_multipart_field_bytes, 50 * 1024 * 1024);
}

/// Test default size limits are 100 MB.
#[test]
fn test_default_size_limits_100mb() {
    use kreuzberg::api::ApiSizeLimits;

    let limits = ApiSizeLimits::default();

    assert_eq!(limits.max_request_body_bytes, 100 * 1024 * 1024);
    assert_eq!(limits.max_multipart_field_bytes, 100 * 1024 * 1024);
}

/// Test ApiSizeLimits from_mb convenience method.
#[test]
fn test_api_size_limits_from_mb() {
    use kreuzberg::api::ApiSizeLimits;

    let limits = ApiSizeLimits::from_mb(100, 50);
    assert_eq!(limits.max_request_body_bytes, 100 * 1024 * 1024);
    assert_eq!(limits.max_multipart_field_bytes, 50 * 1024 * 1024);
}

/// Test ApiSizeLimits new method.
#[test]
fn test_api_size_limits_new() {
    use kreuzberg::api::ApiSizeLimits;

    let limits = ApiSizeLimits::new(1_000_000, 500_000);
    assert_eq!(limits.max_request_body_bytes, 1_000_000);
    assert_eq!(limits.max_multipart_field_bytes, 500_000);
}

/// Test extracting a file larger than 2MB (issue #248).
///
/// This test verifies that the API can handle files larger than Axum's old
/// default multipart field limit of 2MB. The issue reported files >2MB being
/// rejected with HTTP 400, which was due to Axum's default `DefaultBodyLimit`.
///
/// With the fix, we now explicitly set `DefaultBodyLimit::max()` to match our
/// configured size limits (default 10 GB), allowing large file uploads.
#[tokio::test]
async fn test_extract_file_larger_than_2mb() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "A".repeat(3 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"large.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(
        response.status(),
        StatusCode::OK,
        "3MB file should be accepted. If this fails with 400 or 413, the size limit fix is not working correctly."
    );

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("A")
    );
}

/// Test extracting a 2MB file (just above the old Axum limit).
///
/// This tests the boundary case at exactly 2MB, which was the old Axum default limit.
/// Files at this size should now be accepted with the DefaultBodyLimit::max() fix.
#[tokio::test]
async fn test_extract_2mb_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "X".repeat(2 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"boundary.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(
        response.status(),
        StatusCode::OK,
        "2MB file should be accepted (boundary case)"
    );

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("X")
    );
}

/// Test extracting a 5MB file.
///
/// Verifies that files significantly larger than the old 2MB limit work correctly.
#[tokio::test]
async fn test_extract_5mb_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "B".repeat(5 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"medium.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK, "5MB file should be accepted");

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("B")
    );
}

/// Test extracting a 10MB file.
///
/// Verifies that moderately large files (10MB) are handled correctly.
#[tokio::test]
async fn test_extract_10mb_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "C".repeat(10 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"large.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK, "10MB file should be accepted");

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("C")
    );
}

/// Test extracting a 50MB file (half the default limit).
///
/// Verifies that very large files (50MB) are handled correctly,
/// well within the 100MB default limit.
#[tokio::test]
async fn test_extract_50mb_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "D".repeat(50 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"very_large.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(response.status(), StatusCode::OK, "50MB file should be accepted");

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("D")
    );
}

/// Test extracting a 90MB file (near the 100MB default limit).
///
/// Verifies that very large files close to the default limit (90MB out of 100MB)
/// are handled correctly. This tests the upper boundary of the default configuration.
#[tokio::test]
async fn test_extract_90mb_file() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "E".repeat(90 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"huge.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(
        response.status(),
        StatusCode::OK,
        "90MB file should be accepted (within default 100MB limit)"
    );

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 1);
    assert_eq!(results[0]["mime_type"], "text/plain");
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("E")
    );
}

/// Test extracting a file over the 100MB default limit (HTTP 400/413).
///
/// Verifies that files exceeding the 100MB default limit are rejected
/// with a 4xx error (typically HTTP 400 from tower-http RequestBodyLimitLayer or 413)
/// rather than silently failing or accepting the request.
#[tokio::test]
async fn test_extract_file_over_default_limit() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file_content = "F".repeat(101 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"over_limit.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert!(
        response.status() == StatusCode::BAD_REQUEST || response.status() == StatusCode::PAYLOAD_TOO_LARGE,
        "101MB file should be rejected with HTTP 400 or 413 (over 100MB default limit), got {}",
        response.status()
    );
}

/// Test extracting multiple large files within limit.
///
/// Verifies that multiple large files can be uploaded together as long as
/// the total size is within the default 100MB limit.
#[tokio::test]
async fn test_extract_multiple_large_files_within_limit() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file1_content = "G".repeat(25 * 1024 * 1024);
    let file2_content = "H".repeat(25 * 1024 * 1024);
    let file3_content = "I".repeat(25 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"file1.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"file2.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"file3.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file1_content, boundary, file2_content, boundary, file3_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert_eq!(
        response.status(),
        StatusCode::OK,
        "Multiple files totaling 75MB should be accepted"
    );

    let body = axum::body::to_bytes(response.into_body(), usize::MAX)
        .await
        .expect("Failed to read HTTP response body");
    let results: Vec<serde_json::Value> = serde_json::from_slice(&body).expect("Failed to deserialize JSON response");

    assert_eq!(results.len(), 3, "Should have 3 results");
    for result in &results {
        assert_eq!(result["mime_type"], "text/plain");
    }
    assert!(
        results[0]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("G")
    );
    assert!(
        results[1]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("H")
    );
    assert!(
        results[2]["content"]
            .as_str()
            .expect("Failed to extract string from JSON value")
            .contains("I")
    );
}

/// Test extracting multiple large files exceeding limit (HTTP 400/413).
///
/// Verifies that when multiple files together exceed the 100MB default limit,
/// the request is rejected with a 4xx error (typically HTTP 400 or 413).
#[tokio::test]
async fn test_extract_multiple_large_files_exceeding_limit() {
    let app = create_router(ExtractionConfig::default());

    let boundary = "----boundary";
    let file1_content = "J".repeat(50 * 1024 * 1024);
    let file2_content = "K".repeat(55 * 1024 * 1024);

    let body_content = format!(
        "--{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"file1.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}\r\n\
         Content-Disposition: form-data; name=\"files\"; filename=\"file2.txt\"\r\n\
         Content-Type: text/plain\r\n\
         \r\n\
         {}\r\n\
         --{}--\r\n",
        boundary, file1_content, boundary, file2_content, boundary
    );

    let response = app
        .oneshot(
            Request::builder()
                .method("POST")
                .uri("/extract")
                .header("content-type", format!("multipart/form-data; boundary={}", boundary))
                .body(Body::from(body_content))
                .expect("Operation failed"),
        )
        .await
        .expect("Operation failed");

    assert!(
        response.status() == StatusCode::BAD_REQUEST || response.status() == StatusCode::PAYLOAD_TOO_LARGE,
        "Multiple files totaling 105MB should be rejected with HTTP 400 or 413, got {}",
        response.status()
    );
}