use crate::http::helper::test_router;
use reqwest::Client;
use tokio::net::TcpListener;

#[tokio::test]
async fn successfully_calls_the_predict_endpoint_and_return_200() {
    // Arrange
    let client = Client::new();
    let listener = TcpListener::bind("0.0.0.0:0").await.unwrap();
    let addr = listener.local_addr().unwrap();
    let router = test_router().await;
    let predict_url = format!("http://{}/api/predict", addr).to_string();

    tokio::spawn(async move {
        axum::serve(listener, router).await.unwrap();
    });

    // Act: Make Predictions
    let model_input = serde_json::json!(
            {
                "pclass": ["1", "3"],
                "sex": ["male", "female"],
                "age": [22.0, 23.79929292929293],
                "sibsp": ["0", "1", ],
                "parch": ["0", "0"],
                "fare": [151.55, 14.4542],
                "embarked": ["S", "C"],
                "class": ["First", "Third"],
                "who": ["man", "woman"],
                "adult_male": ["True", "False"],
                "deck": ["Unknown", "Unknown"],
                "embark_town": ["Southampton", "Cherbourg"],
                "alone": ["True", "False"]
            }
    )
    .to_string();

    let response = client
        .post(predict_url)
        .json(&serde_json::json!(
            {
                "model_name": "titanic_model",
                "input": model_input
            }

        ))
        .send()
        .await
        .expect("Failed to make request");

    // Assert
    println!("{:?}", response);
    assert!(response.status().is_success())
}

#[tokio::test]
async fn fails_to_calls_the_predict_endpoint_and_return_500_when_input_is_wrong() {
    // Arrange
    let client = Client::new();
    let listener = TcpListener::bind("0.0.0.0:0").await.unwrap();
    let addr = listener.local_addr().unwrap();
    let router = test_router().await;
    let predict_url = format!("http://{}/api/predict", addr).to_string();

    tokio::spawn(async move {
        axum::serve(listener, router).await.unwrap();
    });

    // Act: Make Predictions
    let incorrect_model_input = serde_json::json!(
            {
                "pclass": ["1", "3"],
                "sex": ["male", "female"],
                "age": [22.0, 23.79929292929293],
                "sibsp": ["0", "1", ],
                "alone": ["True", "False"]
            }
    )
    .to_string();

    let response = client
        .post(predict_url)
        .json(&serde_json::json!(
            {
                "model_name": "titanic_model",
                "input": incorrect_model_input
            }

        ))
        .send()
        .await
        .expect("Failed to make request");

    // Assert
    println!("{:?}", response);
    assert!(response.status().is_server_error())
}