PiCloud/crates/picloud/tests/dispatcher_e2e.rs

//! End-to-end dispatcher tests — one per trigger kind (v1.1.5 follow-up,
//! landed in v1.1.6). Each test wires the full all-in-one app via
//! `build_app` (which spawns the real dispatcher + cron scheduler +
//! executor), creates an app + a logging handler script + a trigger,
//! causes the originating event, and polls for the handler's side effect.
//!
//! ## Gating
//!
//! These need a Postgres reachable via `DATABASE_URL`. They follow the
//! `schema_snapshot` pattern (NOT `#[ignore]`): when `DATABASE_URL` is
//! unset the test prints a notice and returns early, so plain
//! `cargo test` stays green locally while CI (which sets `DATABASE_URL`)
//! runs them.
//!
//! ## How "the handler fired" is observed
//!
//! The dispatcher does not write `execution_log` rows for trigger
//! handlers, so each handler instead records its `ctx.event` into a KV
//! marker (`collection = "e2e_markers"`, which no trigger watches — no
//! recursion). The test polls `kv_entries` for that marker and asserts
//! the event shape. See HANDBACK §deviations for why this lives in
//! `picloud/tests/` rather than `manager-core/tests/` (build_app lives in
//! the `picloud` crate) and for the `dead_letter` reinterpretation.

#![allow(clippy::needless_pass_by_value)]

use std::time::Duration;

use axum_test::TestServer;
use serde_json::{json, Value};
use sqlx::postgres::PgPoolOptions;
use sqlx::PgPool;
use uuid::Uuid;

/// Connect + migrate, or return `None` (printing a skip notice) when
/// `DATABASE_URL` is unset — mirrors `schema_snapshot.rs`.
async fn pool_or_skip() -> Option<PgPool> {
    let Ok(url) = std::env::var("DATABASE_URL") else {
        eprintln!("dispatcher_e2e: DATABASE_URL unset — skipping");
        return None;
    };
    let pool = PgPoolOptions::new()
        .max_connections(5)
        .connect(&url)
        .await
        .expect("connect to DATABASE_URL");
    sqlx::migrate!("../manager-core/migrations")
        .run(&pool)
        .await
        .expect("apply migrations");
    Some(pool)
}

/// Build the app over the shared pool with a uniquely-named owner admin,
/// log in, and create a fresh app. `suffix` must be unique per test (the
/// pool is shared, so names must not collide).
async fn server_for(pool: PgPool, suffix: &str) -> (TestServer, String) {
    use picloud_manager_core::auth::hash_password;
    use picloud_shared::InstanceRole;

    let unique = format!("{suffix}-{}", Uuid::new_v4().simple());
    let auth = picloud::AuthDeps::from_pool(pool.clone());
    let username = format!("e2e-{unique}");
    let hash = hash_password("pw").expect("hash");
    auth.users
        .create(&username, &hash, InstanceRole::Owner, None)
        .await
        .expect("seed admin");

    let app = picloud::build_app(pool, auth).await.expect("build_app");
    let mut server = TestServer::new(app).expect("TestServer");
    let resp = server
        .post("/api/v1/admin/auth/login")
        .json(&json!({ "username": username, "password": "pw" }))
        .await;
    resp.assert_status_ok();
    let token = resp.json::<Value>()["token"]
        .as_str()
        .expect("login token")
        .to_string();
    server.add_header("authorization", format!("Bearer {token}"));

    // A fresh app keeps each test's KV / events isolated from siblings.
    let slug = format!("e2e-{unique}");
    let created: Value = server
        .post("/api/v1/admin/apps")
        .json(&json!({ "slug": slug, "name": slug }))
        .await
        .json();
    let app_id = created["id"].as_str().expect("app id").to_string();
    (server, app_id)
}

async fn create_script(server: &TestServer, app_id: &str, name: &str, source: &str) -> String {
    let created: Value = server
        .post("/api/v1/admin/scripts")
        .json(&json!({ "app_id": app_id, "name": name, "source": source }))
        .await
        .json();
    created["id"].as_str().expect("script id").to_string()
}

/// A handler that records its `ctx.event` into a KV marker the test can
/// observe. The marker collection is watched by no trigger.
const MARKER_HANDLER: &str = r#"
    let e = ctx.event;
    kv::collection("e2e_markers").set("marker", e);
    #{ ok: true }
"#;

/// Poll the marker KV key until present (or ~10s timeout).
async fn poll_marker(pool: &PgPool, app_id: &str) -> Option<Value> {
    poll_marker_n(pool, app_id, 100).await
}

/// Poll the marker KV key for `iters` × 100ms.
async fn poll_marker_n(pool: &PgPool, app_id: &str, iters: u32) -> Option<Value> {
    let app_uuid = Uuid::parse_str(app_id).expect("app uuid");
    for _ in 0..iters {
        let row: Option<(Value,)> = sqlx::query_as(
            "SELECT value FROM kv_entries \
             WHERE app_id = $1 AND collection = 'e2e_markers' AND key = 'marker'",
        )
        .bind(app_uuid)
        .fetch_optional(pool)
        .await
        .expect("query marker");
        if let Some((value,)) = row {
            return Some(value);
        }
        tokio::time::sleep(Duration::from_millis(100)).await;
    }
    None
}

async fn execute(server: &TestServer, script_id: &str) {
    server
        .post(&format!("/api/v1/execute/{script_id}"))
        .json(&json!({}))
        .await
        .assert_status_ok();
}

#[tokio::test]
async fn dispatcher_delivers_kv_to_handler() {
    let Some(pool) = pool_or_skip().await else {
        return;
    };
    let (server, app_id) = server_for(pool.clone(), "kv").await;

    let handler = create_script(&server, &app_id, "kv-handler", MARKER_HANDLER).await;
    server
        .post(&format!("/api/v1/admin/apps/{app_id}/triggers/kv"))
        .json(&json!({ "script_id": handler, "collection_glob": "src" }))
        .await
        .assert_status(axum::http::StatusCode::CREATED);

    let source = create_script(
        &server,
        &app_id,
        "kv-source",
        r#"kv::collection("src").set("k", 42); #{ ok: true }"#,
    )
    .await;
    execute(&server, &source).await;

    let event = poll_marker(&pool, &app_id).await.expect("kv handler fired");
    assert_eq!(event["source"], "kv");
    assert_eq!(event["op"], "insert");
    assert_eq!(event["kv"]["collection"], "src");
    assert_eq!(event["kv"]["key"], "k");
}

#[tokio::test]
async fn dispatcher_delivers_docs_to_handler() {
    let Some(pool) = pool_or_skip().await else {
        return;
    };
    let (server, app_id) = server_for(pool.clone(), "docs").await;

    let handler = create_script(&server, &app_id, "docs-handler", MARKER_HANDLER).await;
    server
        .post(&format!("/api/v1/admin/apps/{app_id}/triggers/docs"))
        .json(&json!({ "script_id": handler, "collection_glob": "src" }))
        .await
        .assert_status(axum::http::StatusCode::CREATED);

    let source = create_script(
        &server,
        &app_id,
        "docs-source",
        r#"docs::collection("src").create(#{ x: 1 }); #{ ok: true }"#,
    )
    .await;
    execute(&server, &source).await;

    let event = poll_marker(&pool, &app_id)
        .await
        .expect("docs handler fired");
    assert_eq!(event["source"], "docs");
    assert_eq!(event["op"], "create");
    assert_eq!(event["docs"]["collection"], "src");
}

#[tokio::test]
async fn dispatcher_delivers_cron_to_handler() {
    let Some(pool) = pool_or_skip().await else {
        return;
    };
    let (server, app_id) = server_for(pool.clone(), "cron").await;

    let handler = create_script(&server, &app_id, "cron-handler", MARKER_HANDLER).await;
    // Fire every second (6-field cron, seconds-resolution).
    server
        .post(&format!("/api/v1/admin/apps/{app_id}/triggers/cron"))
        .json(&json!({ "script_id": handler, "schedule": "* * * * * *", "timezone": "UTC" }))
        .await
        .assert_status(axum::http::StatusCode::CREATED);

    // No source — the scheduler enqueues the due tick on its own. The
    // scheduler skips its first tick and then ticks every
    // PICLOUD_CRON_TICK_INTERVAL_MS (default 30s), so poll past that
    // (set the env var lower to speed CI up if desired).
    let event = poll_marker_n(&pool, &app_id, 450)
        .await
        .expect("cron handler fired");
    assert_eq!(event["source"], "cron");
    assert_eq!(event["op"], "tick");
    assert_eq!(event["cron"]["timezone"], "UTC");
}

#[tokio::test]
async fn dispatcher_delivers_files_to_handler() {
    let Some(pool) = pool_or_skip().await else {
        return;
    };
    let (server, app_id) = server_for(pool.clone(), "files").await;

    let handler = create_script(&server, &app_id, "files-handler", MARKER_HANDLER).await;
    server
        .post(&format!("/api/v1/admin/apps/{app_id}/triggers/files"))
        .json(&json!({ "script_id": handler, "collection_glob": "src" }))
        .await
        .assert_status(axum::http::StatusCode::CREATED);

    let source = create_script(
        &server,
        &app_id,
        "files-source",
        r#"
            let data = base64::decode("aGk=");
            files::collection("src").create(#{ name: "f.txt", content_type: "text/plain", data: data });
            #{ ok: true }
        "#,
    )
    .await;
    execute(&server, &source).await;

    let event = poll_marker(&pool, &app_id)
        .await
        .expect("files handler fired");
    assert_eq!(event["source"], "files");
    assert_eq!(event["op"], "create");
    assert_eq!(event["files"]["collection"], "src");
    assert_eq!(event["files"]["name"], "f.txt");
}

#[tokio::test]
async fn dispatcher_delivers_pubsub_to_handler() {
    let Some(pool) = pool_or_skip().await else {
        return;
    };
    let (server, app_id) = server_for(pool.clone(), "pubsub").await;

    let handler = create_script(&server, &app_id, "pubsub-handler", MARKER_HANDLER).await;
    server
        .post(&format!("/api/v1/admin/apps/{app_id}/triggers/pubsub"))
        .json(&json!({ "script_id": handler, "topic_pattern": "e2e.topic" }))
        .await
        .assert_status(axum::http::StatusCode::CREATED);

    let source = create_script(
        &server,
        &app_id,
        "pubsub-source",
        r#"pubsub::publish_durable("e2e.topic", #{ hello: 1 }); #{ ok: true }"#,
    )
    .await;
    execute(&server, &source).await;

    let event = poll_marker(&pool, &app_id)
        .await
        .expect("pubsub handler fired");
    assert_eq!(event["source"], "pubsub");
    assert_eq!(event["op"], "publish");
    assert_eq!(event["pubsub"]["topic"], "e2e.topic");
    assert_eq!(event["pubsub"]["message"]["hello"], 1);
}

#[tokio::test]
async fn dispatcher_delivers_dead_letter_to_handler() {
    // NOTE: the dead-letter creation path (`dispatcher::handle_failure` →
    // `DeadLetterRepo::insert`) writes the `dead_letters` row but does not
    // appear to enqueue deliveries for `dead_letter`-kind triggers
    // (`TriggerRepo::list_matching_dead_letter` has no production caller —
    // see HANDBACK latent-findings). So this test asserts the wired
    // behavior: a failing handler that exhausts its (single) attempt
    // produces a dead-letter row. If/when DL→handler fan-out lands, this
    // can be upgraded to assert the handler marker like the others.
    let Some(pool) = pool_or_skip().await else {
        return;
    };
    let (server, app_id) = server_for(pool.clone(), "dl").await;

    // A handler that always throws, with a single attempt so it
    // dead-letters immediately (no retry backoff).
    let failing = create_script(&server, &app_id, "dl-failing", r#"throw "boom";"#).await;
    server
        .post(&format!("/api/v1/admin/apps/{app_id}/triggers/kv"))
        .json(&json!({
            "script_id": failing,
            "collection_glob": "dlsrc",
            "retry_max_attempts": 1,
            "retry_base_ms": 0
        }))
        .await
        .assert_status(axum::http::StatusCode::CREATED);

    let source = create_script(
        &server,
        &app_id,
        "dl-source",
        r#"kv::collection("dlsrc").set("k", 1); #{ ok: true }"#,
    )
    .await;
    execute(&server, &source).await;

    // Poll the dead_letters table for this app.
    let app_uuid = Uuid::parse_str(&app_id).unwrap();
    let mut count: i64 = 0;
    for _ in 0..100 {
        count = sqlx::query_scalar("SELECT COUNT(*) FROM dead_letters WHERE app_id = $1")
            .bind(app_uuid)
            .fetch_one(&pool)
            .await
            .expect("count dead_letters");
        if count > 0 {
            break;
        }
        tokio::time::sleep(Duration::from_millis(100)).await;
    }
    assert!(count > 0, "a dead-letter row should have been produced");
}