bat/tests/tester.rs

98 lines
2.9 KiB
Rust
Raw Normal View History

2018-05-12 16:48:45 +02:00
use std::env;
use std::fs::{self, File};
2018-08-18 20:44:25 +02:00
use std::io::Read;
use std::path::{Path, PathBuf};
2018-05-12 16:48:45 +02:00
use std::process::Command;
extern crate tempdir;
use self::tempdir::TempDir;
extern crate git2;
use self::git2::build::CheckoutBuilder;
use self::git2::Repository;
use self::git2::Signature;
2018-05-12 16:48:45 +02:00
pub struct BatTester {
/// Temporary working directory
temp_dir: TempDir,
/// Path to the *bat* executable
2018-05-12 16:48:45 +02:00
exe: PathBuf,
}
impl BatTester {
pub fn new() -> Self {
let temp_dir = create_sample_directory().expect("sample directory");
2018-05-12 16:48:45 +02:00
let root = env::current_exe()
.expect("tests executable")
.parent()
.expect("tests executable directory")
.parent()
.expect("bat executable directory")
.to_path_buf();
let exe_name = if cfg!(windows) { "bat.exe" } else { "bat" };
let exe = root.join(exe_name);
2018-05-12 16:48:45 +02:00
BatTester { temp_dir, exe }
2018-05-12 16:48:45 +02:00
}
pub fn test_snapshot(&self, style: &str) {
let output = Command::new(&self.exe)
.current_dir(self.temp_dir.path())
.args(&[
"sample.rs",
"--decorations=always",
&format!("--style={}", style),
]).output()
2018-05-12 16:48:45 +02:00
.expect("bat failed");
2018-05-12 16:48:45 +02:00
// have to do the replace because the filename in the header changes based on the current working directory
let actual = String::from_utf8_lossy(&output.stdout)
.as_ref()
.replace("tests/snapshots/", "");
let mut expected = String::new();
let mut file = File::open(format!("tests/snapshots/output/{}.snapshot.txt", style))
.expect("snapshot file missing");
file.read_to_string(&mut expected)
.expect("could not read snapshot file");
assert_eq!(expected, actual);
}
}
fn create_sample_directory() -> Result<TempDir, git2::Error> {
// Create temp directory and initialize repository
let temp_dir = TempDir::new("bat-tests").expect("Temp directory");
let repo = Repository::init(&temp_dir)?;
2018-05-12 16:48:45 +02:00
// Copy over `sample.rs`
let sample_path = temp_dir.path().join("sample.rs");
println!("{:?}", &sample_path);
fs::copy("tests/snapshots/sample.rs", &sample_path).expect("successful copy");
2018-05-12 16:48:45 +02:00
// Commit
let mut index = repo.index()?;
index.add_path(Path::new("sample.rs"))?;
2018-05-12 16:48:45 +02:00
let oid = index.write_tree()?;
let signature = Signature::now("bat test runner", "bat@test.runner")?;
let tree = repo.find_tree(oid)?;
2018-08-18 20:44:25 +02:00
let _ = repo.commit(
Some("HEAD"), // point HEAD to our new commit
&signature, // author
&signature, // committer
"initial commit",
&tree,
&[],
);
let mut opts = CheckoutBuilder::new();
repo.checkout_head(Some(opts.force()))?;
fs::copy("tests/snapshots/sample.modified.rs", &sample_path).expect("successful copy");
Ok(temp_dir)
2018-05-12 16:48:45 +02:00
}