ostree/rust-bindings/tests/util/mod.rs
Colin Walters eec67ec109 Drop cap-std from our public APIs
Since it bumped semver (when I didn't expect it to; xref
963eebf3ab (r121651362)

It's not load-bearing enough here to matter versus just passing
an untyped file descriptor.

This mainly means that it will be the `glib` ecosystem which
forces transitive semver bumps for us, not both.
2023-08-31 15:34:26 -04:00

101 lines
2.9 KiB
Rust

#[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);
}