rust-bindings/tests/util/mod.rs - main - githubmirror/ostree

rust-bindings/tests/util/mod.rs
100 строк3 KB
Colin Walters
Drop cap-std from our public APIs
31 авг 2023, 22:34
31 авг 2023, 22:34eec67ec
#[cfg(feature = "v2017_10")]
use cap_tempfile::cap_std;
use glib::prelude::*;
use glib::GString;
#[cfg(feature = "v2017_10")]
use std::os::fd::AsFd;
use std::path::Path;

#[derive(Debug)]
pub struct TestRepo {
    pub dir: tempfile::TempDir,
    pub repo: ostree::Repo,
}

impl TestRepo {
    pub fn new() -> TestRepo {
        TestRepo::new_with_mode(ostree::RepoMode::Archive)
    }

    pub fn new_with_mode(repo_mode: ostree::RepoMode) -> TestRepo {
        let dir = tempfile::tempdir().expect("temp repo dir");
        let repo = ostree::Repo::new_for_path(dir.path());
        repo.create(repo_mode, gio::Cancellable::NONE)
            .expect("OSTree repo");
        TestRepo { dir, repo }
    }

    pub fn test_commit(&self, ref_: &str) -> GString {
        let mtree = create_mtree(&self.repo);
        commit(&self.repo, &mtree, ref_)
    }
}

#[derive(Debug)]
#[cfg(feature = "v2017_10")]
pub struct CapTestRepo {
    pub dir: cap_tempfile::TempDir,
    pub repo: ostree::Repo,
}

#[cfg(feature = "v2017_10")]
impl CapTestRepo {
    pub fn new() -> Self {
        Self::new_with_mode(ostree::RepoMode::Archive)
    }

    pub fn new_with_mode(repo_mode: ostree::RepoMode) -> Self {
        let dir = cap_tempfile::tempdir(cap_std::ambient_authority()).unwrap();
        let repo =
            ostree::Repo::create_at_dir(dir.as_fd(), ".", repo_mode, None).expect("repo create");
        Self { dir, repo }
    }
}

pub fn create_mtree(repo: &ostree::Repo) -> ostree::MutableTree {
    let mtree = ostree::MutableTree::new();
    assert_eq!(mtree.copy_files().len(), 0);
    assert_eq!(mtree.copy_subdirs().len(), 0);
    let file = gio::File::for_path(
        Path::new(env!("CARGO_MANIFEST_DIR")).join("rust-bindings/tests/data/test.tar"),
    );
    repo.write_archive_to_mtree(&file, &mtree, None, true, gio::Cancellable::NONE)
        .expect("test mtree");
    mtree
}

pub fn commit(repo: &ostree::Repo, mtree: &ostree::MutableTree, ref_: &str) -> GString {
    let txn = repo
        .auto_transaction(gio::Cancellable::NONE)
        .expect("prepare transaction");
    let repo_file = txn
        .repo()
        .write_mtree(mtree, gio::Cancellable::NONE)
        .expect("write mtree")
        .downcast::<ostree::RepoFile>()
        .unwrap();
    let checksum = repo
        .write_commit(
            None,
            "Test Commit".into(),
            None,
            None,
            &repo_file,
            gio::Cancellable::NONE,
        )
        .expect("write commit");
    repo.transaction_set_ref(None, ref_, checksum.as_str().into());
    txn.commit(gio::Cancellable::NONE)
        .expect("commit transaction");
    checksum
}

pub fn assert_test_file(checkout: &Path) {
    let testfile_path = checkout
        .join("test-checkout")
        .join("testdir")
        .join("testfile");
    let testfile_contents = std::fs::read_to_string(testfile_path).expect("test file");
    assert_eq!("test\n", testfile_contents);
}