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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69

// crates/postgresflow/tests/error_classification.rs
mod common;

use common::setup_db;

use azums::jobs::timeline::build_timeline;
use azums::jobs::{AttemptsRepo, JobsRepo, PolicyDecisionsRepo};

use uuid::Uuid;

async fn insert_job(pool: &sqlx::PgPool) -> Uuid {
    sqlx::query_scalar(
        r#"
        INSERT INTO jobs (queue, job_type, payload_json, run_at, status, priority, max_attempts)
        VALUES ('default', 'fail_me', '{}'::jsonb, now(), 'queued', 0, 5)
        RETURNING id
        "#,
    )
    .fetch_one(pool)
    .await
    .unwrap()
}

#[tokio::test]
async fn timeline_includes_suggested_action_for_known_error_codes() {
    let Some(pool) = setup_db().await else {
        return;
    };

    let jobs = JobsRepo::new(pool.clone());
    let attempts = AttemptsRepo::new(pool.clone());
    let policy = PolicyDecisionsRepo::new(pool.clone()); // ✅ new

    let job_id = insert_job(&pool).await;

    // lease job so it becomes running
    let job = jobs
        .lease_one_job("default", "worker-1", 30)
        .await
        .unwrap()
        .expect("should lease job");
    assert_eq!(job.id, job_id);

    // start attempt
    let attempt = attempts.start_attempt(job_id, "worker-1").await.unwrap();

    // finish attempt as failed with a known code
    attempts
        .finish_failed(attempt.id, 12, "RATE_LIMIT", "429 from upstream")
        .await
        .unwrap();

    // timeline should include suggested action
    let tl = build_timeline(&jobs, &attempts, &policy, job_id) // ✅ new arg
        .await
        .unwrap()
        .expect("timeline exists");

    let a1 = tl.attempts.first().expect("has attempts");
    assert_eq!(a1.error_code.as_deref(), Some("RATE_LIMIT"));
    assert!(
        a1.suggested_action
            .as_deref()
            .unwrap_or("")
            .to_lowercase()
            .contains("back off"),
        "expected a runbook-style suggestion for RATE_LIMIT"
    );
}