atlas-cli 0.2.0

use crate::cc_attestation;
use crate::cc_attestation::mock::MockAttestationProvider;
use crate::error::{Error, Result};
use serde_json::Value;
use tdx_workload_attestation::get_platform_name;
use tdx_workload_attestation::provider::AttestationProvider;

// Test that the mock attestation provider generates valid reports
#[test]
fn test_mock_attestation() -> Result<()> {
    // Create a mock provider with a test platform
    let provider = MockAttestationProvider::new("test-platform");

    // Get attestation report
    let report = provider
        .get_attestation_report()
        .map_err(|e| Error::CCAttestationError(e.to_string()))?;

    // Verify the report is valid JSON
    let report_json: Value = serde_json::from_str(&report)?;

    // Assert the report contains expected fields
    assert_eq!(report_json["report_type"], "mock_attestation");
    assert_eq!(report_json["platform"], "test-platform");
    assert!(report_json["timestamp"].is_string());
    assert_eq!(report_json["version"], "1.0");
    assert_eq!(report_json["status"], "simulated");
    assert_eq!(
        report_json["message"],
        "This is a mock attestation report for non-Linux or unsupported platforms"
    );

    Ok(())
}

// Test that the get_report function works for unsupported platforms
#[test]
fn test_platform_selection() -> Result<()> {
    // Try to get a report for any platform
    let report = cc_attestation::get_report(false)?;

    // Verify the report is valid JSON
    let report_json: Value = serde_json::from_str(&report)?;

    // Get the platform name
    let platform_name =
        get_platform_name().map_err(|e| Error::CCAttestationError(e.to_string()))?;

    // Verify the selected platform
    if platform_name == "tdx-linux" {
        // Verify the report contains an MRTD
        assert_ne!(report_json["td_info"]["mrtd"].as_array(), None);
    } else {
        // Verify it's a mock report
        assert_eq!(report_json["report_type"], "mock_attestation");
    }

    Ok(())
}

// Test that the get_report function with show parameter works
#[test]
fn test_get_report_with_show() -> Result<()> {
    let report = cc_attestation::get_report(true)?;

    // Verify it returned a non-empty string
    assert!(!report.is_empty());

    Ok(())
}

// Integration test for using attestation in manifests
#[test]
fn test_attestation_in_manifest() -> Result<()> {
    use atlas_c2pa_lib::assertion::CustomAssertion;
    use tempfile::tempdir;

    // Create a simple temporary file for testing
    let temp_dir = tempdir()?;
    let file_path = temp_dir.path().join("test_file.txt");
    std::fs::write(&file_path, b"test data")?;

    // Get a report for a test platform
    let report = cc_attestation::get_report(false)?;

    // Create a custom assertion with the report
    let assertion = CustomAssertion {
        label: "test-platform".to_string(),
        data: serde_json::Value::String(report),
    };

    // Verify that the assertion is properly formed
    assert_eq!(assertion.label, "test-platform");
    assert!(assertion.data.is_string());

    // Clean up test directory
    temp_dir.close()?;

    Ok(())
}