85 lines
2.9 KiB
Rust
85 lines
2.9 KiB
Rust
use application::processing::{CompleteJobCommand, CompleteJobHandler};
|
|
use application::testing::{InMemoryJobBatchRepository, InMemoryJobRepository, StubEventPublisher};
|
|
use domain::entities::{Job, JobBatch, JobStatus, JobType};
|
|
use domain::events::DomainEvent;
|
|
use domain::ports::{JobBatchRepository, JobRepository};
|
|
use domain::value_objects::StructuredData;
|
|
use std::sync::Arc;
|
|
|
|
#[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));
|
|
}
|