zenwave 0.4.0

//! Tests for middleware components in Zenwave

use std::{
    convert::Infallible,
    sync::{
        Arc,
        atomic::{AtomicUsize, Ordering},
    },
    time::Duration,
};
use zenwave::cookie::CookieStore;

use zenwave::redirect::FollowRedirect;
use zenwave::{
    Body, Client, Endpoint, HttpError, Middleware, Request, Response, StatusCode, client,
};

mod common;
use common::httpbin_uri;

#[test_executors::async_test]
async fn test_cookie_store_middleware() {
    let mut client = client().enable_cookie();

    // First request to set a cookie
    let response = client
        .get(httpbin_uri("/cookies/set/test/value"))
        .unwrap()
        .await;
    assert!(response.is_ok());

    // Second request should include the cookie
    let response = client.get(httpbin_uri("/cookies")).unwrap().await;
    assert!(response.is_ok());

    let response = response.unwrap();
    let body = response.into_body().into_string().await.unwrap();
    assert!(body.contains("test"));
    assert!(body.contains("value"));
}

#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
#[cfg_attr(not(target_arch = "wasm32"), test)]
fn test_cookie_store_creation() {
    let cookie_store = CookieStore::default();
    assert!(!format!("{cookie_store:?}").is_empty());
}

#[test_executors::async_test]
async fn test_follow_redirect_middleware() {
    let mut client = client();

    // This should follow the redirect and return the final response
    let response = client.get(httpbin_uri("/redirect/1")).unwrap().await;
    assert!(response.is_ok());
    let response = response.unwrap();
    assert!(response.status().is_success());
}

#[cfg_attr(target_arch = "wasm32", wasm_bindgen_test::wasm_bindgen_test)]
#[cfg_attr(not(target_arch = "wasm32"), test)]
fn test_follow_redirect_creation() {
    let base_client = client();
    let _redirect_client = FollowRedirect::new(base_client);
    // Just ensure it can be created
}

#[test_executors::async_test]
async fn test_follow_redirect_multiple_redirects() {
    let mut client = client();

    // Test multiple redirects
    let response = client.get(httpbin_uri("/redirect/3")).unwrap().await;
    assert!(response.is_ok());
    let response = response.unwrap();
    assert!(response.status().is_success());
}

#[test_executors::async_test]
async fn test_client_with_multiple_middleware() {
    let mut client = client().enable_cookie();

    // Test that both middleware work together
    let response = client
        .get(httpbin_uri("/redirect-to?url=/cookies/set/test/redirect"))
        .unwrap()
        .await;
    assert!(response.is_ok());

    // Verify cookie was set after redirect
    let response2 = client.get(httpbin_uri("/cookies")).unwrap().await;
    assert!(response2.is_ok());
}

#[test_executors::async_test]
async fn test_without_redirect_middleware() {
    let mut client = client().disable_redirect();
    let response = client.get(httpbin_uri("/redirect/1")).unwrap().await;
    assert!(response.is_ok());
    let response = response.unwrap();
    // Should be a redirect status code, not success
    assert!(response.status().is_redirection());
}

#[test_executors::async_test]
async fn test_middleware_with_custom_middleware() {
    struct TestMiddleware;

    impl Middleware for TestMiddleware {
        type Error = Infallible;
        async fn handle<E: Endpoint>(
            &mut self,
            request: &mut Request,
            mut next: E,
        ) -> Result<Response, zenwave::middleware::MiddlewareError<E::Error, Self::Error>> {
            // Add a custom header
            let header_name: http_kit::header::HeaderName = "X-Test".parse().unwrap();
            let header_value: http_kit::header::HeaderValue = "middleware-test".parse().unwrap();
            request.headers_mut().insert(header_name, header_value);
            next.respond(request)
                .await
                .map_err(zenwave::middleware::MiddlewareError::Endpoint)
        }
    }

    let mut client = client().with(TestMiddleware);

    let response = client.get(httpbin_uri("/headers")).unwrap().await;
    assert!(response.is_ok());

    let response = response.unwrap();
    let body = response.into_body().into_string().await.unwrap();
    assert!(body.contains("X-Test"));
    assert!(body.contains("middleware-test"));
}

#[cfg(not(target_arch = "wasm32"))]
#[derive(Clone)]
struct SlowClient {
    delay: Duration,
    status: StatusCode,
}

#[cfg(not(target_arch = "wasm32"))]
impl Endpoint for SlowClient {
    type Error = Infallible;
    async fn respond(&mut self, _request: &mut Request) -> Result<Response, Self::Error> {
        async_io::Timer::after(self.delay).await;
        Ok(http::Response::builder()
            .status(self.status)
            .body(Body::empty())
            .unwrap())
    }
}

#[cfg(not(target_arch = "wasm32"))]
impl Client for SlowClient {}

#[derive(Clone)]
struct CountingBackend {
    hits: Arc<AtomicUsize>,
}

impl CountingBackend {
    const fn new(hits: Arc<AtomicUsize>) -> Self {
        Self { hits }
    }
}

impl Endpoint for CountingBackend {
    type Error = Infallible;
    fn respond(
        &mut self,
        _request: &mut Request,
    ) -> impl std::future::Future<Output = Result<Response, Self::Error>> {
        let hit = self.hits.fetch_add(1, Ordering::SeqCst) + 1;
        std::future::ready(Ok(http::Response::builder()
            .status(StatusCode::OK)
            .header(http::header::CACHE_CONTROL, "max-age=60")
            .body(Body::from(format!("hit-{hit}")))
            .unwrap()))
    }
}

impl Client for CountingBackend {}

#[cfg(not(target_arch = "wasm32"))]
#[test_executors::async_test]
async fn test_timeout_middleware_success() {
    let mut client = SlowClient {
        delay: Duration::from_millis(20),
        status: StatusCode::OK,
    }
    .timeout(Duration::from_secs(1));

    let response = client
        .get("https://example.com")
        .unwrap()
        .await
        .expect("request should complete before timeout");
    assert_eq!(response.status(), StatusCode::OK);
}

#[cfg(not(target_arch = "wasm32"))]
#[test_executors::async_test]
async fn test_timeout_middleware_triggers_gateway_timeout() {
    let mut client = SlowClient {
        delay: Duration::from_millis(200),
        status: StatusCode::OK,
    }
    .timeout(Duration::from_millis(10));

    let err = client
        .get("https://example.com")
        .unwrap()
        .await
        .expect_err("timeout should trigger before slow backend responds");

    assert_eq!(err.status(), StatusCode::GATEWAY_TIMEOUT);
    assert!(
        err.to_string().contains("timed out"),
        "error message should mention timeout"
    );
}

#[test_executors::async_test]
async fn test_enable_cache_serves_cached_response() {
    let hits = Arc::new(AtomicUsize::new(0));
    let backend = CountingBackend::new(hits.clone());
    let mut client = backend.enable_cache();

    let first = client
        .get("https://example.com/cache")
        .unwrap()
        .await
        .expect("initial request should succeed");
    let first_body = first.into_body().into_string().await.unwrap();

    let second = client
        .get("https://example.com/cache")
        .unwrap()
        .await
        .expect("cached request should succeed");
    let second_body = second.into_body().into_string().await.unwrap();

    assert_eq!(first_body, second_body);
    assert_eq!(first_body.as_str(), "hit-1");
    assert_eq!(hits.load(Ordering::SeqCst), 1, "backend should be hit once");
}