mod mock;

use tophat::{
    glitch, glitch_code,
    http::StatusCode,
    server::{
        accept,
        glitch::{Glitch, GlitchExt},
    },
};

use mock::Client;

const RESP_400: &str = "HTTP/1.1 400 Bad Request\r\ncontent-length: 0\r\n\r\n";
const RESP_500: &str = "HTTP/1.1 500 Internal Server Error\r\ncontent-length: 0\r\n\r\n";
const S_400: StatusCode = StatusCode::BAD_REQUEST;
const S_500: StatusCode = StatusCode::INTERNAL_SERVER_ERROR;

#[test]
fn test_request_manually_create_glitch() {
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_400,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one".parse::<usize>().map_err(|_| Glitch::bad_request())?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
}

#[test]
fn test_request_glitch_with_context() {
    // one test to see that just `?` works, and another to see that manual Glitch creation still
    // works even with anyhow enabled.

    // automatic
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_500,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one".parse::<usize>()?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });

    // context no message
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_500,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one".parse::<usize>().glitch(S_500)?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });

    // context
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            "HTTP/1.1 500 Internal Server Error\r\ncontent-length: 12\r\ncontent-type: text/plain\r\n\r\ncustom error",
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one".parse::<usize>().glitch_ctx(S_500, "custom error")?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });

    // context on Option
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            "HTTP/1.1 400 Bad Request\r\ncontent-length: 12\r\ncontent-type: text/plain\r\n\r\ncustom error",
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            let usr = None;
            usr.glitch_ctx(S_400, "custom error")?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });

    // manual
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_400,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one".parse::<usize>().map_err(|_| Glitch::bad_request())?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
}

#[test]
fn test_request_glitch_macro() {
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_500,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one".parse::<usize>().map_err(|_| glitch!())?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_400,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one"
                .parse::<usize>()
                .map_err(|_| glitch!(StatusCode::BAD_REQUEST))?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            "HTTP/1.1 400 Bad Request\r\ncontent-length: 12\r\ncontent-type: text/plain\r\n\r\ncustom error",
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one"
                .parse::<usize>()
                .map_err(|_| glitch!(StatusCode::BAD_REQUEST, "custom error"))?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
}

#[test]
fn test_request_glitch_code_macro() {
    // this one can panic if code incorrect
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            "HTTP/1.1 400 Bad Request\r\ncontent-length: 12\r\ncontent-type: text/plain\r\n\r\ncustom error",
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one"
                .parse::<usize>()
                .map_err(|_| glitch_code!(400, "custom error"))?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
}

#[test]
#[should_panic]
fn test_request_glitch_code_macro_panic() {
    // this one can panic if code incorrect
    smol::block_on(async {
        let testclient = Client::new(
            "GET /foo/bar HTTP/1.1\r\nHost: example.org\r\n\r\n",
            RESP_400,
        );

        accept(testclient.clone(), |_req, resp_wtr| async move {
            "one"
                .parse::<usize>()
                .map_err(|_| glitch_code!(1, "custom error"))?;
            let done = resp_wtr.send().await.unwrap();

            Ok(done)
        })
        .await
        .unwrap();

        testclient.assert();
    });
}