opencrabs 0.3.75

The autonomous, self-improving AI agent. Single Rust binary. Every channel. Install with: cargo install opencrabs
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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

//! Background tasks survive a restart as reported interruptions (#763).
//!
//! A detached command used to live only in an in-memory map, so killing the
//! process dropped both the child and the record of it and the session waited
//! forever on a resume that could never arrive.

use crate::db::{BackgroundTaskRepository, Database};
use uuid::Uuid;

async fn repo() -> BackgroundTaskRepository {
    let db = Database::connect_in_memory()
        .await
        .expect("in-memory db connects");
    db.run_migrations().await.expect("migrations run");
    BackgroundTaskRepository::new(db.pool().clone())
}

#[tokio::test]
async fn a_recorded_task_survives_to_be_read_back() {
    let repo = repo().await;
    let (id, session) = (Uuid::new_v4(), Uuid::new_v4());
    repo.record(
        id,
        session,
        "cargo test",
        "cargo test --all-features",
        "/tmp",
    )
    .await
    .expect("record");

    let rows = repo.all().await.expect("read back");
    assert_eq!(rows.len(), 1, "the row is what a restart finds");
    assert_eq!(rows[0].id, id);
    assert_eq!(rows[0].session_id, session, "resume targets this session");
    assert_eq!(rows[0].label, "cargo test");
    assert_eq!(
        rows[0].command, "cargo test --all-features",
        "the full command is kept so the agent can re-run it"
    );
}

#[tokio::test]
async fn a_task_that_finished_leaves_nothing_behind() {
    // Otherwise the next startup reports a phantom interruption for a command
    // that actually completed.
    let repo = repo().await;
    let (id, session) = (Uuid::new_v4(), Uuid::new_v4());
    repo.record(id, session, "cargo build", "cargo build", "/tmp")
        .await
        .expect("record");
    repo.clear(id).await.expect("clear");

    assert!(repo.all().await.expect("read back").is_empty());
}

#[tokio::test]
async fn clearing_one_task_leaves_its_siblings_running() {
    // Two identical commands can be in flight at once, which is why rows are
    // keyed by id rather than matched by label.
    let repo = repo().await;
    let session = Uuid::new_v4();
    let (first, second) = (Uuid::new_v4(), Uuid::new_v4());
    repo.record(first, session, "cargo test", "cargo test", "/tmp")
        .await
        .expect("record");
    repo.record(second, session, "cargo test", "cargo test", "/tmp")
        .await
        .expect("record");

    repo.clear(first).await.expect("clear");

    let rows = repo.all().await.expect("read back");
    assert_eq!(rows.len(), 1, "only the finished one is gone");
    assert_eq!(rows[0].id, second);
}

#[tokio::test]
async fn tasks_from_several_sessions_are_all_reported() {
    // Startup accounts for every session's orphans, not just the foreground one.
    let repo = repo().await;
    let (a, b) = (Uuid::new_v4(), Uuid::new_v4());
    repo.record(Uuid::new_v4(), a, "cargo test", "cargo test", "/tmp")
        .await
        .expect("record");
    repo.record(Uuid::new_v4(), b, "npm test", "npm test", "/tmp")
        .await
        .expect("record");

    let rows = repo.all().await.expect("read back");
    assert_eq!(rows.len(), 2);
    let sessions: Vec<Uuid> = rows.iter().map(|r| r.session_id).collect();
    assert!(sessions.contains(&a) && sessions.contains(&b));
}

#[tokio::test]
async fn clear_all_empties_the_table_after_reporting() {
    let repo = repo().await;
    let session = Uuid::new_v4();
    repo.record(Uuid::new_v4(), session, "cargo test", "cargo test", "/tmp")
        .await
        .expect("record");
    repo.clear_all().await.expect("clear all");
    assert!(
        repo.all().await.expect("read back").is_empty(),
        "leaving rows behind would re-report the same interruption forever"
    );
}