azums 0.1.0

High-performance job queue & streaming engine for Rust — from embedded to cloud
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44

use azums::{Job, MockBackend, QuickstartFlow, StorageBackend};
use std::sync::Arc;

#[tokio::test]
async fn test_mock_backend_call_recording() -> anyhow::Result<()> {
    let mock = MockBackend::with_memory();
    let backend: Arc<dyn StorageBackend> = Arc::new(mock.clone());

    let flow = QuickstartFlow::new(backend);

    let _id = flow
        .enqueue(Job::new(
            "send_notification",
            serde_json::json!({"user": "alice"}),
        ))
        .await?;

    mock.assert_enqueued_job_type("send_notification");

    let executed = Arc::new(std::sync::atomic::AtomicBool::new(false));
    let executed_clone = executed.clone();

    flow.register_handler("send_notification", move |_job| {
        let ex = executed_clone.clone();
        async move {
            ex.store(true, std::sync::atomic::Ordering::SeqCst);
            Ok(())
        }
    })
    .await;

    let processed = flow.run_until_empty().await?;
    assert_eq!(processed, 1);
    assert!(executed.load(std::sync::atomic::Ordering::SeqCst));

    let calls = mock.calls();
    assert!(!calls.is_empty());
    println!("Recorded calls in MockBackend: {calls:?}");

    mock.clear_calls();
    assert!(mock.calls().is_empty());

    Ok(())
}