forge_core/testing/

mock_dispatch.rs

//! Mock dispatchers for testing job and workflow dispatch.
//!
//! Provides mock implementations that record dispatched jobs and workflows
//! for verification in tests.

use std::sync::RwLock;

use chrono::{DateTime, Utc};
use uuid::Uuid;

use crate::error::{ForgeError, Result};
use crate::job::JobStatus;
use crate::workflow::WorkflowStatus;

/// Record of a dispatched job.
#[derive(Debug, Clone)]
pub struct DispatchedJob {
    /// Job ID.
    pub id: Uuid,
    /// Job type name.
    pub job_type: String,
    /// Job arguments (serialized).
    pub args: serde_json::Value,
    /// When the job was dispatched.
    pub dispatched_at: DateTime<Utc>,
    /// Current status (for test simulation).
    pub status: JobStatus,
    /// Cancellation reason (if any).
    pub cancel_reason: Option<String>,
}

/// Record of a started workflow.
#[derive(Debug, Clone)]
pub struct StartedWorkflow {
    /// Run ID.
    pub run_id: Uuid,
    /// Workflow name.
    pub workflow_name: String,
    /// Input (serialized).
    pub input: serde_json::Value,
    /// When the workflow was started.
    pub started_at: DateTime<Utc>,
    /// Current status.
    pub status: WorkflowStatus,
}

/// Mock job dispatcher for testing.
///
/// Records dispatched jobs for later verification.
///
/// # Example
///
/// ```ignore
/// let dispatch = MockJobDispatch::new();
/// dispatch.dispatch("send_email", json!({"to": "test@example.com"})).await?;
///
/// dispatch.assert_dispatched("send_email");
/// dispatch.assert_dispatched_with("send_email", |args| {
///     args["to"] == "test@example.com"
/// });
/// ```
pub struct MockJobDispatch {
    jobs: RwLock<Vec<DispatchedJob>>,
}

impl MockJobDispatch {
    /// Create a new mock job dispatcher.
    pub fn new() -> Self {
        Self {
            jobs: RwLock::new(Vec::new()),
        }
    }

    /// Dispatch a job (records for later verification).
    pub async fn dispatch<T: serde::Serialize>(&self, job_type: &str, args: T) -> Result<Uuid> {
        let id = Uuid::new_v4();
        let args_json =
            serde_json::to_value(args).map_err(|e| ForgeError::Serialization(e.to_string()))?;

        let job = DispatchedJob {
            id,
            job_type: job_type.to_string(),
            args: args_json,
            dispatched_at: Utc::now(),
            status: JobStatus::Pending,
            cancel_reason: None,
        };

        self.jobs.write().unwrap().push(job);
        Ok(id)
    }

    /// Get all dispatched jobs.
    pub fn dispatched_jobs(&self) -> Vec<DispatchedJob> {
        self.jobs.read().unwrap().clone()
    }

    /// Get jobs of a specific type.
    pub fn jobs_of_type(&self, job_type: &str) -> Vec<DispatchedJob> {
        self.jobs
            .read()
            .unwrap()
            .iter()
            .filter(|j| j.job_type == job_type)
            .cloned()
            .collect()
    }

    /// Assert that a job type was dispatched.
    pub fn assert_dispatched(&self, job_type: &str) {
        let jobs = self.jobs.read().unwrap();
        let found = jobs.iter().any(|j| j.job_type == job_type);
        assert!(
            found,
            "Expected job '{}' to be dispatched, but it wasn't. Dispatched jobs: {:?}",
            job_type,
            jobs.iter().map(|j| &j.job_type).collect::<Vec<_>>()
        );
    }

    /// Assert that a job was dispatched with matching arguments.
    pub fn assert_dispatched_with<F>(&self, job_type: &str, predicate: F)
    where
        F: Fn(&serde_json::Value) -> bool,
    {
        let jobs = self.jobs.read().unwrap();
        let found = jobs
            .iter()
            .any(|j| j.job_type == job_type && predicate(&j.args));
        assert!(
            found,
            "Expected job '{}' with matching args to be dispatched",
            job_type
        );
    }

    /// Assert that a job type was not dispatched.
    pub fn assert_not_dispatched(&self, job_type: &str) {
        let jobs = self.jobs.read().unwrap();
        let found = jobs.iter().any(|j| j.job_type == job_type);
        assert!(
            !found,
            "Expected job '{}' NOT to be dispatched, but it was",
            job_type
        );
    }

    /// Assert that a specific number of jobs were dispatched.
    pub fn assert_dispatch_count(&self, job_type: &str, expected: usize) {
        let jobs = self.jobs.read().unwrap();
        let count = jobs.iter().filter(|j| j.job_type == job_type).count();
        assert_eq!(
            count, expected,
            "Expected {} dispatches of '{}', but found {}",
            expected, job_type, count
        );
    }

    /// Clear all recorded jobs.
    pub fn clear(&self) {
        self.jobs.write().unwrap().clear();
    }

    /// Mark a job as completed (for test simulation).
    pub fn complete_job(&self, job_id: Uuid) {
        let mut jobs = self.jobs.write().unwrap();
        if let Some(job) = jobs.iter_mut().find(|j| j.id == job_id) {
            job.status = JobStatus::Completed;
        }
    }

    /// Mark a job as failed (for test simulation).
    pub fn fail_job(&self, job_id: Uuid) {
        let mut jobs = self.jobs.write().unwrap();
        if let Some(job) = jobs.iter_mut().find(|j| j.id == job_id) {
            job.status = JobStatus::Failed;
        }
    }

    /// Mark a job as cancelled (for test simulation).
    pub fn cancel_job(&self, job_id: Uuid, reason: Option<String>) {
        let mut jobs = self.jobs.write().unwrap();
        if let Some(job) = jobs.iter_mut().find(|j| j.id == job_id) {
            job.status = JobStatus::Cancelled;
            job.cancel_reason = reason;
        }
    }
}

impl Default for MockJobDispatch {
    fn default() -> Self {
        Self::new()
    }
}

impl crate::function::JobDispatch for MockJobDispatch {
    fn get_info(&self, _job_type: &str) -> Option<crate::job::JobInfo> {
        None
    }

    fn dispatch_by_name(
        &self,
        job_type: &str,
        args: serde_json::Value,
    ) -> std::pin::Pin<Box<dyn std::future::Future<Output = Result<Uuid>> + Send + '_>> {
        let job_type = job_type.to_string();
        Box::pin(async move { self.dispatch(&job_type, args).await })
    }

    fn cancel(
        &self,
        job_id: Uuid,
        reason: Option<String>,
    ) -> std::pin::Pin<Box<dyn std::future::Future<Output = Result<bool>> + Send + '_>> {
        Box::pin(async move {
            self.cancel_job(job_id, reason);
            Ok(true)
        })
    }
}

/// Mock workflow dispatcher for testing.
///
/// Records started workflows for later verification.
///
/// # Example
///
/// ```ignore
/// let dispatch = MockWorkflowDispatch::new();
/// dispatch.start("onboarding", json!({"user_id": "123"})).await?;
///
/// dispatch.assert_started("onboarding");
/// ```
pub struct MockWorkflowDispatch {
    workflows: RwLock<Vec<StartedWorkflow>>,
}

impl MockWorkflowDispatch {
    /// Create a new mock workflow dispatcher.
    pub fn new() -> Self {
        Self {
            workflows: RwLock::new(Vec::new()),
        }
    }

    /// Start a workflow (records for later verification).
    pub async fn start<T: serde::Serialize>(&self, workflow_name: &str, input: T) -> Result<Uuid> {
        let run_id = Uuid::new_v4();
        let input_json =
            serde_json::to_value(input).map_err(|e| ForgeError::Serialization(e.to_string()))?;

        let workflow = StartedWorkflow {
            run_id,
            workflow_name: workflow_name.to_string(),
            input: input_json,
            started_at: Utc::now(),
            status: WorkflowStatus::Created,
        };

        self.workflows.write().unwrap().push(workflow);
        Ok(run_id)
    }

    /// Get all started workflows.
    pub fn started_workflows(&self) -> Vec<StartedWorkflow> {
        self.workflows.read().unwrap().clone()
    }

    /// Get workflows of a specific name.
    pub fn workflows_named(&self, name: &str) -> Vec<StartedWorkflow> {
        self.workflows
            .read()
            .unwrap()
            .iter()
            .filter(|w| w.workflow_name == name)
            .cloned()
            .collect()
    }

    /// Assert that a workflow was started.
    pub fn assert_started(&self, workflow_name: &str) {
        let workflows = self.workflows.read().unwrap();
        let found = workflows.iter().any(|w| w.workflow_name == workflow_name);
        assert!(
            found,
            "Expected workflow '{}' to be started, but it wasn't. Started workflows: {:?}",
            workflow_name,
            workflows
                .iter()
                .map(|w| &w.workflow_name)
                .collect::<Vec<_>>()
        );
    }

    /// Assert that a workflow was started with matching input.
    pub fn assert_started_with<F>(&self, workflow_name: &str, predicate: F)
    where
        F: Fn(&serde_json::Value) -> bool,
    {
        let workflows = self.workflows.read().unwrap();
        let found = workflows
            .iter()
            .any(|w| w.workflow_name == workflow_name && predicate(&w.input));
        assert!(
            found,
            "Expected workflow '{}' with matching input to be started",
            workflow_name
        );
    }

    /// Assert that a workflow was not started.
    pub fn assert_not_started(&self, workflow_name: &str) {
        let workflows = self.workflows.read().unwrap();
        let found = workflows.iter().any(|w| w.workflow_name == workflow_name);
        assert!(
            !found,
            "Expected workflow '{}' NOT to be started, but it was",
            workflow_name
        );
    }

    /// Assert that a specific number of workflows were started.
    pub fn assert_start_count(&self, workflow_name: &str, expected: usize) {
        let workflows = self.workflows.read().unwrap();
        let count = workflows
            .iter()
            .filter(|w| w.workflow_name == workflow_name)
            .count();
        assert_eq!(
            count, expected,
            "Expected {} starts of '{}', but found {}",
            expected, workflow_name, count
        );
    }

    /// Clear all recorded workflows.
    pub fn clear(&self) {
        self.workflows.write().unwrap().clear();
    }

    /// Mark a workflow as completed (for test simulation).
    pub fn complete_workflow(&self, run_id: Uuid) {
        let mut workflows = self.workflows.write().unwrap();
        if let Some(workflow) = workflows.iter_mut().find(|w| w.run_id == run_id) {
            workflow.status = WorkflowStatus::Completed;
        }
    }

    /// Mark a workflow as failed (for test simulation).
    pub fn fail_workflow(&self, run_id: Uuid) {
        let mut workflows = self.workflows.write().unwrap();
        if let Some(workflow) = workflows.iter_mut().find(|w| w.run_id == run_id) {
            workflow.status = WorkflowStatus::Failed;
        }
    }
}

impl Default for MockWorkflowDispatch {
    fn default() -> Self {
        Self::new()
    }
}

#[cfg(test)]
mod tests {
    use super::*;

    #[tokio::test]
    async fn test_mock_job_dispatch() {
        let dispatch = MockJobDispatch::new();

        let job_id = dispatch
            .dispatch("send_email", serde_json::json!({"to": "test@example.com"}))
            .await
            .unwrap();

        assert!(!job_id.is_nil());
        dispatch.assert_dispatched("send_email");
        dispatch.assert_not_dispatched("other_job");
    }

    #[tokio::test]
    async fn test_job_dispatch_with_args() {
        let dispatch = MockJobDispatch::new();

        dispatch
            .dispatch("send_email", serde_json::json!({"to": "test@example.com"}))
            .await
            .unwrap();

        dispatch.assert_dispatched_with("send_email", |args| args["to"] == "test@example.com");
    }

    #[tokio::test]
    async fn test_job_dispatch_count() {
        let dispatch = MockJobDispatch::new();

        dispatch
            .dispatch("job_a", serde_json::json!({}))
            .await
            .unwrap();
        dispatch
            .dispatch("job_b", serde_json::json!({}))
            .await
            .unwrap();
        dispatch
            .dispatch("job_a", serde_json::json!({}))
            .await
            .unwrap();

        dispatch.assert_dispatch_count("job_a", 2);
        dispatch.assert_dispatch_count("job_b", 1);
    }

    #[tokio::test]
    async fn test_mock_workflow_dispatch() {
        let dispatch = MockWorkflowDispatch::new();

        let run_id = dispatch
            .start("onboarding", serde_json::json!({"user_id": "123"}))
            .await
            .unwrap();

        assert!(!run_id.is_nil());
        dispatch.assert_started("onboarding");
        dispatch.assert_not_started("other_workflow");
    }

    #[tokio::test]
    async fn test_workflow_dispatch_with_input() {
        let dispatch = MockWorkflowDispatch::new();

        dispatch
            .start("onboarding", serde_json::json!({"user_id": "123"}))
            .await
            .unwrap();

        dispatch.assert_started_with("onboarding", |input| input["user_id"] == "123");
    }

    #[tokio::test]
    async fn test_clear() {
        let dispatch = MockJobDispatch::new();
        dispatch
            .dispatch("test", serde_json::json!({}))
            .await
            .unwrap();

        assert_eq!(dispatch.dispatched_jobs().len(), 1);
        dispatch.clear();
        assert_eq!(dispatch.dispatched_jobs().len(), 0);
    }

    #[tokio::test]
    async fn test_job_status_simulation() {
        let dispatch = MockJobDispatch::new();
        let job_id = dispatch
            .dispatch("test", serde_json::json!({}))
            .await
            .unwrap();

        let jobs = dispatch.dispatched_jobs();
        assert_eq!(jobs[0].status, JobStatus::Pending);

        dispatch.complete_job(job_id);

        let jobs = dispatch.dispatched_jobs();
        assert_eq!(jobs[0].status, JobStatus::Completed);
    }

    #[tokio::test]
    async fn test_job_cancel_simulation() {
        let dispatch = MockJobDispatch::new();
        let job_id = dispatch
            .dispatch("test", serde_json::json!({}))
            .await
            .unwrap();

        dispatch.cancel_job(job_id, Some("user request".to_string()));

        let jobs = dispatch.dispatched_jobs();
        assert_eq!(jobs[0].status, JobStatus::Cancelled);
        assert_eq!(jobs[0].cancel_reason.as_deref(), Some("user request"));
    }
}