80 lines
2.4 KiB
Rust
80 lines
2.4 KiB
Rust
extern crate gio;
|
|
extern crate glib;
|
|
extern crate ostree;
|
|
extern crate tempfile;
|
|
|
|
use gio::NONE_CANCELLABLE;
|
|
use glib::prelude::*;
|
|
use ostree::prelude::*;
|
|
use ostree::RepoFile;
|
|
use std::fs;
|
|
use std::io;
|
|
use std::io::Write;
|
|
|
|
fn create_repo(repodir: &tempfile::TempDir) -> Result<ostree::Repo, glib::Error> {
|
|
let repo = ostree::Repo::new_for_path(repodir.path());
|
|
repo.create(ostree::RepoMode::Archive, NONE_CANCELLABLE)?;
|
|
Ok(repo)
|
|
}
|
|
|
|
fn create_test_file(treedir: &tempfile::TempDir) -> Result<(), io::Error> {
|
|
let mut testfile = fs::File::create(treedir.path().join("test.txt"))?;
|
|
write!(testfile, "test")?;
|
|
Ok(())
|
|
}
|
|
|
|
fn create_mtree(
|
|
treedir: &tempfile::TempDir,
|
|
repo: &ostree::Repo,
|
|
) -> Result<ostree::MutableTree, glib::Error> {
|
|
let gfile = gio::File::new_for_path(treedir.path());
|
|
let mtree = ostree::MutableTree::new();
|
|
repo.write_directory_to_mtree(&gfile, &mtree, None, NONE_CANCELLABLE)?;
|
|
Ok(mtree)
|
|
}
|
|
|
|
fn commit_mtree(repo: &ostree::Repo, mtree: &ostree::MutableTree) -> Result<String, glib::Error> {
|
|
repo.prepare_transaction(NONE_CANCELLABLE)?;
|
|
let repo_file = repo
|
|
.write_mtree(mtree, NONE_CANCELLABLE)?
|
|
.downcast::<RepoFile>()
|
|
.unwrap();
|
|
let checksum = repo
|
|
.write_commit(
|
|
None,
|
|
"Test Commit".into(),
|
|
None,
|
|
None,
|
|
&repo_file,
|
|
NONE_CANCELLABLE,
|
|
)?
|
|
.to_string();
|
|
repo.transaction_set_ref(None, "test", checksum.as_str().into());
|
|
repo.commit_transaction(NONE_CANCELLABLE)?;
|
|
Ok(checksum)
|
|
}
|
|
|
|
fn open_repo(repodir: &tempfile::TempDir) -> Result<ostree::Repo, glib::Error> {
|
|
let repo = ostree::Repo::new_for_path(repodir.path());
|
|
repo.open(NONE_CANCELLABLE)?;
|
|
Ok(repo)
|
|
}
|
|
|
|
#[test]
|
|
fn should_commit_content_to_repo_and_list_refs_again() {
|
|
let repodir = tempfile::tempdir().unwrap();
|
|
let treedir = tempfile::tempdir().unwrap();
|
|
|
|
let repo = create_repo(&repodir).expect("failed to create repo");
|
|
create_test_file(&treedir).expect("failed to create test file");
|
|
let mtree = create_mtree(&treedir, &repo).expect("failed to build mtree");
|
|
let checksum = commit_mtree(&repo, &mtree).expect("failed to commit mtree");
|
|
|
|
let repo = open_repo(&repodir).expect("failed to open repo");
|
|
let refs = repo
|
|
.list_refs(None, NONE_CANCELLABLE)
|
|
.expect("failed to list refs");
|
|
assert_eq!(refs.len(), 1);
|
|
assert_eq!(refs["test"], checksum);
|
|
}
|