caterpillar 0.2.0

Detect and apply RAUC update bundles during boot
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

// SPDX-FileCopyrightText: 2023 David Runge <dave@sleepmap.de>
// SPDX-License-Identifier: Apache-2.0 OR MIT
use super::TestError;
use rstest::fixture;
use std::fs::remove_file;
use std::path::PathBuf;

/// Remove files in a list of files
pub fn remove_files(files: &[&str]) -> Result<(), std::io::Error> {
    for file in files {
        if let Err(error) = remove_file(file) {
            eprintln!("Unable to remove {}: {}", file, error);
        };
    }
    Ok(())
}

#[fixture]
/// A fixture to provide the first matching location of OVMF code files
pub fn input_path_ovmf_code() -> Result<PathBuf, TestError> {
    let candidates = [PathBuf::from("/usr/share/edk2/x64/OVMF_CODE.4m.fd")];

    match candidates.iter().find(|&candidate| candidate.exists()) {
        Some(candidate) => Ok(candidate.clone()),
        None => return Err(TestError::Missing("OVMF code".to_string())),
    }
}

#[fixture]
/// A fixture to provide the first matching location of OVMF variable files
pub fn input_path_ovmf_vars() -> Result<PathBuf, TestError> {
    let candidates = [PathBuf::from("/usr/share/edk2/x64/OVMF_VARS.4m.fd")];

    match candidates.iter().find(|&candidate| candidate.exists()) {
        Some(candidate) => Ok(candidate.clone()),
        None => return Err(TestError::Missing("OVMF code".to_string())),
    }
}

#[fixture]
/// A fixture to provide the output dir for integration tests
pub fn output_dir() -> PathBuf {
    PathBuf::from(env!("CARGO_TARGET_TMPDIR"))
}

#[fixture]
/// A fixture to provide the output path for the base image
pub fn output_path_base_image(output_dir: PathBuf) -> PathBuf {
    PathBuf::from(&output_dir).join("base")
}

#[fixture]
/// A fixture to provide the output path for the single image
pub fn output_path_single_image(output_dir: PathBuf) -> PathBuf {
    PathBuf::from(&output_dir).join("single")
}

#[fixture]
/// A fixture to provide the output path for the A/B image
pub fn output_path_ab_image(output_dir: PathBuf) -> PathBuf {
    PathBuf::from(&output_dir).join("ab")
}

#[fixture]
/// A fixture to provide the output path for the override directory of the A/B image
pub fn output_dir_ab_override(output_dir: PathBuf) -> PathBuf {
    PathBuf::from(&output_dir).join("override_for_ab")
}

#[fixture]
/// A fixture to provide the output path for modified OVMF vars file
///
/// This location is used for the OVMF vars that contain EFI bootloader entries for the A/B image
pub fn output_path_ovmf_vars(output_dir: PathBuf) -> PathBuf {
    PathBuf::from(&output_dir).join("ovmf_vars.fd")
}

#[fixture]
/// A fixture to provide the integration tests directory of the project
fn tests_dir() -> PathBuf {
    PathBuf::from(env!("CARGO_MANIFEST_DIR")).join("tests")
}

#[fixture]
/// A fixture to provide the directory of the mkosi base image setup
pub fn mkosi_dir_base_image(tests_dir: PathBuf) -> PathBuf {
    PathBuf::from(tests_dir).join("mkosi").join("base_image")
}

#[fixture]
/// A fixture to provide the directory of the mkosi single image setup
pub fn mkosi_dir_single_image(tests_dir: PathBuf) -> PathBuf {
    PathBuf::from(tests_dir).join("mkosi").join("single_image")
}

#[fixture]
/// A fixture to provide the directory of the mkosi A/B image setup
pub fn mkosi_dir_ab_image(tests_dir: PathBuf) -> PathBuf {
    PathBuf::from(tests_dir).join("mkosi").join("ab_image")
}

#[fixture]
/// A fixture to provide the directory of the mkosi update image setup
pub fn mkosi_dir_bundle_image(tests_dir: PathBuf) -> PathBuf {
    PathBuf::from(tests_dir).join("mkosi").join("bundle_image")
}