use std::sync::Arc;
use application::testing::{InMemoryJobBatchRepository, InMemoryJobRepository, StubEventPublisher};
use application::processing::{CompleteJobCommand, CompleteJobHandler};
use domain::entities::{Job, JobBatch, JobStatus, JobType};
use domain::events::DomainEvent;
use domain::ports::{JobBatchRepository, JobRepository};
use domain::value_objects::StructuredData;

#[tokio::test]
async fn completes_job() {
    let job_repo = Arc::new(InMemoryJobRepository::new());
    let batch_repo = Arc::new(InMemoryJobBatchRepository::new());
    let event_pub = Arc::new(StubEventPublisher::new());

    let mut job = Job::new(JobType::ExtractMetadata, 5, StructuredData::new());
    job.start().unwrap();
    let job_id = job.job_id;
    job_repo.save(&job).await.unwrap();

    let handler = CompleteJobHandler::new(job_repo.clone(), batch_repo.clone(), event_pub.clone());
    let result = handler.execute(CompleteJobCommand {
        job_id,
        result: StructuredData::new(),
    }).await.unwrap();

    assert_eq!(result.status, JobStatus::Completed);
    assert!(result.result_data.is_some());
}

#[tokio::test]
async fn completes_job_and_updates_batch() {
    let job_repo = Arc::new(InMemoryJobRepository::new());
    let batch_repo = Arc::new(InMemoryJobBatchRepository::new());
    let event_pub = Arc::new(StubEventPublisher::new());

    let batch = JobBatch::new("test-batch", 2);
    let batch_id = batch.batch_id;
    batch_repo.save(&batch).await.unwrap();

    let mut job = Job::new(JobType::ExtractMetadata, 5, StructuredData::new())
        .with_batch(batch_id);
    job.start().unwrap();
    let job_id = job.job_id;
    job_repo.save(&job).await.unwrap();

    let handler = CompleteJobHandler::new(job_repo.clone(), batch_repo.clone(), event_pub.clone());
    handler.execute(CompleteJobCommand {
        job_id,
        result: StructuredData::new(),
    }).await.unwrap();

    let updated_batch = batch_repo.find_by_id(&batch_id).await.unwrap().unwrap();
    assert_eq!(updated_batch.completed_count, 1);
}

#[tokio::test]
async fn publishes_event() {
    let job_repo = Arc::new(InMemoryJobRepository::new());
    let batch_repo = Arc::new(InMemoryJobBatchRepository::new());
    let event_pub = Arc::new(StubEventPublisher::new());

    let mut job = Job::new(JobType::ExtractMetadata, 5, StructuredData::new());
    job.start().unwrap();
    let job_id = job.job_id;
    job_repo.save(&job).await.unwrap();

    let handler = CompleteJobHandler::new(job_repo.clone(), batch_repo.clone(), event_pub.clone());
    handler.execute(CompleteJobCommand {
        job_id,
        result: StructuredData::new(),
    }).await.unwrap();

    let events = event_pub.published().await;
    assert_eq!(events.len(), 1);
    assert!(matches!(&events[0], DomainEvent::JobCompleted { job_id: id, .. } if *id == job_id));
}