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

pub mod common;
use aws_sdk_s3::primitives::ByteStream;
use bytes::Bytes;
use common::*;
use mountpoint_s3_client::error::{CopyObjectError, ObjectClientError};
use mountpoint_s3_client::S3RequestError;
use mountpoint_s3_client::{ObjectClient, S3CrtClient};

#[tokio::test]
async fn test_copy_objects() {
    let sdk_client = get_test_sdk_client().await;
    let (bucket, prefix) = get_test_bucket_and_prefix("test_copy_objects");

    let key = format!("{prefix}/hello");
    let body = b"hello world!";
    sdk_client
        .put_object()
        .bucket(&bucket)
        .key(&key)
        .body(ByteStream::from(Bytes::from_static(body)))
        .send()
        .await
        .unwrap();

    let client: S3CrtClient = get_test_client();
    let copy_key = format!("{prefix}/hello2");

    let _result = client
        .copy_object(&bucket, &key, &bucket, &copy_key, &Default::default())
        .await
        .expect("copy_object operation should succeed");

    sdk_client
        .head_object()
        .bucket(&bucket)
        .key(&copy_key)
        .send()
        .await
        .expect("copied object should exist");
}

#[tokio::test]
async fn test_copy_object_no_permission() {
    let (_bucket, prefix) = get_test_bucket_and_prefix("test_copy_object_no_permission");
    let bucket = get_test_bucket_without_permissions();
    let key = format!("{prefix}/hello");
    let copy_key = format!("{prefix}/hello2");

    let client: S3CrtClient = get_test_client();
    let result = client
        .copy_object(&bucket, &key, &bucket, &copy_key, &Default::default())
        .await;

    assert!(matches!(
        result,
        Err(ObjectClientError::ClientError(S3RequestError::Forbidden(_, _)))
    ));
}

#[tokio::test]
async fn test_copy_object_non_existing_key() {
    let (bucket, prefix) = get_test_bucket_and_prefix("test_copy_objects");
    let key = format!("{prefix}/hello");
    let copy_key = format!("{prefix}/hello2");

    let client: S3CrtClient = get_test_client();
    let result = client
        .copy_object(&bucket, &key, &bucket, &copy_key, &Default::default())
        .await;

    assert!(matches!(
        result,
        Err(ObjectClientError::ServiceError(CopyObjectError::NotFound))
    ));
}

// TODO: Add integration test for cross bucket copy but before that need to set up a new environment variable for a new bucket.