arcbox-agent 0.6.3

Guest agent for ArcBox VMs
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 std::collections::HashSet;

use arcbox_connect::v1::SandboxCleanupTicket;

pub fn unseen_rescan_tickets(
    sent: &mut HashSet<(String, String)>,
    tickets: Vec<SandboxCleanupTicket>,
) -> Vec<SandboxCleanupTicket> {
    let pending: HashSet<_> = tickets
        .iter()
        .map(|ticket| (ticket.id.clone(), ticket.token.clone()))
        .collect();
    sent.retain(|generation| pending.contains(generation));
    tickets
        .into_iter()
        .filter(|ticket| !sent.contains(&(ticket.id.clone(), ticket.token.clone())))
        .collect()
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn periodic_rescan_discovers_marker_without_terminal_event() {
        let old = SandboxCleanupTicket {
            id: "old".into(),
            token: "old-generation".into(),
            startup: false,
            ..Default::default()
        };
        let new = SandboxCleanupTicket {
            id: "new".into(),
            token: "new-generation".into(),
            startup: false,
            ..Default::default()
        };
        let mut sent = HashSet::from([(old.id.clone(), old.token.clone())]);

        let unseen = unseen_rescan_tickets(&mut sent, vec![old, new.clone()]);

        assert_eq!(unseen, vec![new]);
    }
}