use pdf_process::{pdf_info, Password, PdfInfoArgs, PdfInfoError};
use tokio::fs::read;

/// Tests from actual files
#[tokio::test]
async fn test_actual_files() {
    let data = read("./tests/samples/test-pdf-2-pages.pdf").await.unwrap();
    let info = pdf_info(&data, &PdfInfoArgs::default()).await.unwrap();
    assert_eq!(info.pages(), Some(Ok(2)));

    let data = read("./tests/samples/test-pdf.pdf").await.unwrap();
    let info = pdf_info(&data, &PdfInfoArgs::default()).await.unwrap();
    assert_eq!(info.pages(), Some(Ok(1)));
}

/// Tests getting pdfinfo from an encrypted file when the password is not set
#[tokio::test]
async fn test_encrypted() {
    let data = read("./tests/samples/test-pdf-2-pages-encrypted.pdf")
        .await
        .unwrap();

    let err = pdf_info(&data, &PdfInfoArgs::default()).await.unwrap_err();

    assert!(matches!(err, PdfInfoError::PdfEncrypted));
}

/// Tests getting pdfinfo from a encrypted file when the password is set
#[tokio::test]
async fn test_encrypted_with_password() {
    let data = read("./tests/samples/test-pdf-2-pages-encrypted.pdf")
        .await
        .unwrap();
    let args = PdfInfoArgs::default().set_password(Password::user("password"));
    let info = pdf_info(&data, &args).await.unwrap();

    assert_eq!(info.pages(), Some(Ok(2)));
    assert_eq!(info.encrypted(), Some(true));

    let encryption = info.encryption().unwrap().unwrap();
    assert!(encryption.is_encrypted());
    assert!(encryption.is_print_allowed());
    assert!(encryption.is_copy_allowed());
    assert!(encryption.is_change_allowed());
    assert!(encryption.is_add_notes_allowed());
    assert_eq!(encryption.algorithm(), Some("AES"));

    let args = PdfInfoArgs::default().set_password(Password::user("password"));
    let info = pdf_info(&data, &args).await.unwrap();

    assert_eq!(info.pages(), Some(Ok(2)));
    assert_eq!(info.encrypted(), Some(true));
}

/// Tests getting pdfinfo from a encrypted file when the password is set
#[tokio::test]
async fn test_encrypted_with_incorrect_password() {
    let data = read("./tests/samples/test-pdf-2-pages-encrypted.pdf")
        .await
        .unwrap();
    let args = PdfInfoArgs::default().set_password(Password::user("incorrect"));
    let err = pdf_info(&data, &args).await.unwrap_err();

    assert!(matches!(err, PdfInfoError::IncorrectPassword));
}