harn-vm 0.10.180

Async bytecode virtual machine for the Harn programming language
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

//! A mount the sandbox backend refuses must leave no trace: the anchor, the
//! redo stack and the transcript read as before the attempt.

use crate::workspace_anchor::{MountMode, WorkspaceAnchor};

fn root_mounted_events(session: &str) -> usize {
    crate::agent_sessions::transcript(session)
        .as_ref()
        .map(crate::llm::helpers::vm_value_to_json)
        .and_then(|transcript| transcript.get("events").cloned())
        .and_then(|events| events.as_array().cloned())
        .map_or(0, |events| {
            events
                .iter()
                .filter(|event| event["kind"] == "RootMounted")
                .count()
        })
}

fn anchored_session(name: &str) -> String {
    crate::reset_thread_local_state();
    let session = crate::agent_sessions::open_or_create_for_test(Some(name.to_string()));
    let anchor = WorkspaceAnchor {
        primary: std::env::temp_dir(),
        additional_roots: Vec::new(),
        anchored_at: "0".to_string(),
    };
    crate::agent_sessions::set_workspace_anchor(&session, Some(anchor)).unwrap();
    session
}

#[test]
fn a_refused_new_mount_records_no_event_and_leaves_no_root() {
    let session = anchored_session("refused-new-mount");
    let root = tempfile::tempdir().unwrap();
    let text = root.path().to_string_lossy().to_string();
    let refused = crate::agent_sessions::add_workspace_root_verified(
        &session,
        &text,
        Some(MountMode::ReadOnly),
        None,
        |_| Err("refused".to_string()),
    );
    assert_eq!(refused.unwrap_err(), "refused");
    assert_eq!(root_mounted_events(&session), 0);
    let (_, roots) = crate::agent_sessions::list_workspace_roots(&session).unwrap();
    assert!(roots.is_empty());
}

#[test]
fn a_refused_remount_restores_the_prior_mount_and_adds_no_event() {
    let session = anchored_session("refused-remount");
    let root = tempfile::tempdir().unwrap();
    let text = root.path().to_string_lossy().to_string();
    crate::agent_sessions::add_workspace_root(&session, &text, Some(MountMode::Extend), None)
        .unwrap();
    let (_, before) = crate::agent_sessions::list_workspace_roots(&session).unwrap();
    assert_eq!(root_mounted_events(&session), 1);
    let refused = crate::agent_sessions::add_workspace_root_verified(
        &session,
        &text,
        Some(MountMode::ReadOnly),
        None,
        |_| Err("refused".to_string()),
    );
    assert!(refused.is_err());
    let (_, after) = crate::agent_sessions::list_workspace_roots(&session).unwrap();
    assert_eq!(after, before, "the earlier mount, timestamp included");
    assert_eq!(root_mounted_events(&session), 1);
}