firkin-runtime 0.0.3

Production Apple/VZ runtime composition for the firkin Rust containerization library
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

//! Runtime preflight tests.

use std::path::{Path, PathBuf};

use firkin_runtime::types::Size;
use firkin_runtime::{DiskPressureProbe, RuntimePreflight, RuntimePreflightError};

struct RecordingDiskProbe {
    available: Size,
    probed_paths: Vec<PathBuf>,
}

impl DiskPressureProbe for RecordingDiskProbe {
    type Error = &'static str;

    fn available_disk(&mut self, path: &Path) -> Result<Size, Self::Error> {
        self.probed_paths.push(path.to_path_buf());
        Ok(self.available)
    }
}

#[test]
fn runtime_preflight_checks_required_roots_and_disk_floor() {
    let root = tempfile::tempdir().expect("tempdir");
    let snapshots = root.path().join("snapshots");
    let logs = root.path().join("logs");
    std::fs::create_dir_all(&snapshots).expect("snapshots");
    std::fs::create_dir_all(&logs).expect("logs");
    let mut probe = RecordingDiskProbe {
        available: Size::gib(32),
        probed_paths: Vec::new(),
    };

    let report = RuntimePreflight::new(&snapshots, &logs, Size::gib(10))
        .check_with_disk_probe(&mut probe)
        .expect("preflight passes");

    assert_eq!(report.snapshot_root(), snapshots.as_path());
    assert_eq!(report.log_root(), logs.as_path());
    assert_eq!(probe.probed_paths, vec![snapshots, logs]);
}

#[test]
fn runtime_preflight_rejects_missing_required_roots() {
    let root = tempfile::tempdir().expect("tempdir");
    let snapshots = root.path().join("snapshots");
    let logs = root.path().join("logs");
    std::fs::create_dir_all(&logs).expect("logs");
    let mut probe = RecordingDiskProbe {
        available: Size::gib(32),
        probed_paths: Vec::new(),
    };

    let error = RuntimePreflight::new(&snapshots, &logs, Size::gib(10))
        .check_with_disk_probe(&mut probe)
        .expect_err("missing root rejects");

    assert!(matches!(
        error,
        RuntimePreflightError::MissingRoot { path } if path == snapshots
    ));
    assert!(probe.probed_paths.is_empty());
}

#[test]
fn runtime_preflight_rejects_low_disk_before_runtime_work() {
    let root = tempfile::tempdir().expect("tempdir");
    let snapshots = root.path().join("snapshots");
    let logs = root.path().join("logs");
    std::fs::create_dir_all(&snapshots).expect("snapshots");
    std::fs::create_dir_all(&logs).expect("logs");
    let mut probe = RecordingDiskProbe {
        available: Size::gib(9),
        probed_paths: Vec::new(),
    };

    let error = RuntimePreflight::new(&snapshots, &logs, Size::gib(10))
        .check_with_disk_probe(&mut probe)
        .expect_err("disk floor rejects");

    assert!(matches!(error, RuntimePreflightError::Disk { .. }));
    assert_eq!(probe.probed_paths, vec![snapshots]);
}