curl-http-client 2.5.2

use std::fs;

use async_curl::CurlActor;
use http::{Method, Request, StatusCode};
use test_case::test_case;
use tokio::sync::mpsc::channel;
use url::Url;

use crate::collector::{Collector, FileInfo};
use crate::http_client::{Bps, BytesOffset, HttpClient};
use crate::test::test_setup::{setup_test_environment, MockResponder, ResponderType};

#[tokio::test]
async fn test_download() {
    let responder = MockResponder::new(ResponderType::File);
    let (server, tempdir) = setup_test_environment(responder).await;
    let target_url = Url::parse(format!("{}/test", server.uri()).as_str()).unwrap();

    let save_to = tempdir.path().join("downloaded_file.jpg");
    let actor = CurlActor::new();
    let collector = Collector::File(FileInfo::path(save_to.clone()));
    let request = Request::builder()
        .uri(target_url.as_str())
        .method(Method::GET)
        .body(None)
        .unwrap();

    let response = HttpClient::new(collector)
        .request(request)
        .unwrap()
        .nonblocking(actor)
        .perform()
        .await
        .unwrap();

    println!("Response: {:?}", response);
    assert_eq!(response.status(), StatusCode::OK);
    assert_eq!(*response.body(), None);
    assert_eq!(fs::read(save_to).unwrap(), include_bytes!("sample.jpg"));
    assert!(!response.headers().is_empty());
}

#[tokio::test]
async fn test_download_with_speed_control() {
    let responder = MockResponder::new(ResponderType::File);
    let (server, tempdir) = setup_test_environment(responder).await;
    let target_url = Url::parse(format!("{}/test", server.uri()).as_str()).unwrap();

    let save_to = tempdir.path().join("downloaded_file.jpg");
    let actor = CurlActor::new();
    let collector = Collector::File(FileInfo::path(save_to.clone()));
    let request = Request::builder()
        .uri(target_url.as_str())
        .method(Method::GET)
        .body(None)
        .unwrap();

    let response = HttpClient::new(collector)
        .download_speed(Bps::from(40000000))
        .unwrap()
        .request(request)
        .unwrap()
        .nonblocking(actor)
        .perform()
        .await
        .unwrap();

    println!("Response: {:?}", response);
    assert_eq!(response.status(), StatusCode::OK);
    assert_eq!(*response.body(), None);
    assert_eq!(fs::read(save_to).unwrap(), include_bytes!("sample.jpg"));
    assert!(!response.headers().is_empty());
}

#[test_case(4500, StatusCode::PARTIAL_CONTENT; "Offset 4500 bytes")]
#[test_case(0, StatusCode::OK ; "Offset 0 bytes")]
#[test_case(include_bytes!("sample.jpg").len(), StatusCode::PARTIAL_CONTENT ; "Offset max bytes")]
#[tokio::test]
async fn test_resume_download(offset: usize, expected_status_code: StatusCode) {
    let responder = MockResponder::new(ResponderType::File);
    let (server, tempdir) = setup_test_environment(responder).await;
    let target_url = Url::parse(format!("{}/test", server.uri()).as_str()).unwrap();

    let save_to = tempdir.path().join("downloaded_file.jpg");

    let partial_saved_file = include_bytes!("sample.jpg");
    fs::write(save_to.as_path(), &partial_saved_file[0..offset]).unwrap();

    let partial_file_size = fs::metadata(save_to.as_path()).unwrap().len() as usize;

    let actor = CurlActor::new();
    let collector = Collector::File(FileInfo::path(save_to.clone()));
    let request = Request::builder()
        .uri(target_url.as_str())
        .method(Method::GET)
        .body(None)
        .unwrap();

    let response = HttpClient::new(collector)
        .resume_from(BytesOffset::from(partial_file_size))
        .unwrap()
        .request(request)
        .unwrap()
        .nonblocking(actor)
        .perform()
        .await
        .unwrap();

    println!("Response: {:?}", response);
    assert_eq!(response.status(), expected_status_code);
    assert_eq!(*response.body(), None);
    assert_eq!(fs::read(save_to).unwrap(), include_bytes!("sample.jpg"));
    assert!(!response.headers().is_empty());
}

#[tokio::test]
async fn test_download_with_transfer_speed_sender() {
    let responder = MockResponder::new(ResponderType::File);
    let (server, tempdir) = setup_test_environment(responder).await;
    let target_url = Url::parse(format!("{}/test", server.uri()).as_str()).unwrap();

    let save_to = tempdir.path().join("downloaded_file.jpg");

    let actor = CurlActor::new();

    let (tx, mut rx) = channel(1);

    let file_info = FileInfo::path(save_to.clone()).with_transfer_speed_sender(tx);
    let collector = Collector::File(file_info);
    let request = Request::builder()
        .uri(target_url.as_str())
        .method(Method::GET)
        .body(None)
        .unwrap();

    let handle = tokio::spawn(async move {
        while let Some(speed) = rx.recv().await {
            println!("Download Speed: {} kB/s", speed.as_bytes_per_sec());
        }
        println!("Recieved ended.");
    });

    let response = HttpClient::new(collector)
        .download_speed(Bps::from(40000000))
        .unwrap()
        .request(request)
        .unwrap()
        .nonblocking(actor)
        .perform()
        .await
        .unwrap();

    println!("Response: {:?}", response);
    assert_eq!(response.status(), StatusCode::OK);
    assert_eq!(*response.body(), None);
    assert_eq!(fs::read(save_to).unwrap(), include_bytes!("sample.jpg"));
    assert!(!response.headers().is_empty());

    handle.abort();
}

#[tokio::test]
async fn test_download_with_headers() {
    let responder = MockResponder::new(ResponderType::File);
    let (server, tempdir) = setup_test_environment(responder).await;
    let target_url = Url::parse(format!("{}/test", server.uri()).as_str()).unwrap();

    let save_to = tempdir.path().join("downloaded_file.jpg");
    let actor = CurlActor::new();
    let collector = Collector::FileAndHeaders(FileInfo::path(save_to.clone()), Vec::new());
    let request = Request::builder()
        .uri(target_url.as_str())
        .method(Method::GET)
        .body(None)
        .unwrap();

    let response = HttpClient::new(collector)
        .request(request)
        .unwrap()
        .nonblocking(actor)
        .perform()
        .await
        .unwrap();

    println!("Response: {:?}", response);
    assert_eq!(response.status(), StatusCode::OK);
    assert_eq!(*response.body(), None);
    assert_eq!(fs::read(save_to).unwrap(), include_bytes!("sample.jpg"));
    assert!(!response.headers().is_empty());
}