[functional-tests] Recode thin_restore tests in Rust.

Now part of 'cargo test'

tests/thin_restore.rs

@@ -0,0 +1,144 @@
+use anyhow::Result;
+use duct::{cmd, Expression};
+use std::path::{Path, PathBuf};
+use std::str::from_utf8;
+use tempfile::{tempdir, TempDir};
+use thinp::file_utils;
+use thinp::version::TOOLS_VERSION;
+
+mod common;
+
+use common::xml_generator::{write_xml, FragmentedS, SingleThinS};
+use common::*;
+
+//------------------------------------------
+
+#[test]
+fn accepts_v() -> Result<()> {
+    let stdout = thin_restore!("-V").read()?;
+    assert_eq!(stdout, TOOLS_VERSION);
+    Ok(())
+}
+
+#[test]
+fn accepts_version() -> Result<()> {
+    let stdout = thin_restore!("--version").read()?;
+    assert_eq!(stdout, TOOLS_VERSION);
+    Ok(())
+}
+
+const USAGE: &'static str = "Usage: thin_restore [options]\nOptions:\n  {-h|--help}\n  {-i|--input} <input xml file>\n  {-o|--output} <output device or file>\n  {--transaction-id} <natural>\n  {--data-block-size} <natural>\n  {--nr-data-blocks} <natural>\n  {-q|--quiet}\n  {-V|--version}";
+
+#[test]
+fn accepts_h() -> Result<()> {
+    let stdout = thin_restore!("-h").read()?;
+    assert_eq!(stdout, USAGE);
+    Ok(())
+}
+
+#[test]
+fn accepts_help() -> Result<()> {
+    let stdout = thin_restore!("--help").read()?;
+    assert_eq!(stdout, USAGE);
+    Ok(())
+}
+
+#[test]
+fn no_input_file() -> Result<()> {
+    let dir = tempdir()?;
+    let md = mk_zeroed_md(&dir)?;
+    let stderr = run_fail(thin_restore!("-o", &md))?;
+    assert!(stderr.contains("No input file provided."));
+    Ok(())
+}
+
+#[test]
+fn missing_input_file() -> Result<()> {
+    let dir = tempdir()?;
+    let md = mk_zeroed_md(&dir)?;
+    let stderr = run_fail(thin_restore!("-i", "no-such-file", "-o", &md))?;
+    assert!(superblock_all_zeroes(&md)?);
+    assert!(stderr.contains("Couldn't stat file"));
+    Ok(())
+}
+
+#[test]
+fn garbage_input_file() -> Result<()> {
+    let dir = tempdir()?;
+    let xml = mk_zeroed_md(&dir)?;
+    let md = mk_zeroed_md(&dir)?;
+    let _stderr = run_fail(thin_restore!("-i", &xml, "-o", &md))?;
+    assert!(superblock_all_zeroes(&md)?);
+    Ok(())
+}
+
+#[test]
+fn no_output_file() -> Result<()> {
+    let dir = tempdir()?;
+    let xml = mk_valid_xml(&dir)?;
+    let stderr = run_fail(thin_restore!("-i", &xml))?;
+    assert!(stderr.contains("No output file provided."));
+    Ok(())
+}
+
+#[test]
+fn tiny_output_file() -> Result<()> {
+    let dir = tempdir()?;
+    let xml = mk_valid_xml(&dir)?;
+    let md = mk_path(dir.path(), "meta.bin");
+    let _file = file_utils::create_sized_file(&md, 4096);
+    let stderr = run_fail(thin_restore!("-i", &xml, "-o", &md))?;
+    eprintln!("{}", stderr);
+    assert!(stderr.contains("Output file too small"));
+    Ok(())
+}
+
+fn quiet_flag(flag: &str) -> Result<()> {
+    let dir = tempdir()?;
+    let xml = mk_valid_xml(&dir)?;
+    let md = mk_zeroed_md(&dir)?;
+
+    let output = thin_restore!("-i", &xml, "-o", &md, flag).run()?;
+
+    assert!(output.status.success());
+    assert_eq!(output.stdout.len(), 0);
+    assert_eq!(output.stderr.len(), 0);
+    Ok(())
+}
+
+#[test]
+fn accepts_q() -> Result<()> {
+    quiet_flag("-q")
+}
+
+#[test]
+fn accepts_quiet() -> Result<()> {
+    quiet_flag("--quiet")
+}
+
+fn override_something(flag: &str, value: &str, pattern: &str) -> Result<()> {
+    let dir = tempdir()?;
+    let xml = mk_valid_xml(&dir)?;
+    let md = mk_zeroed_md(&dir)?;
+
+    thin_restore!("-i", &xml, "-o", &md, flag, value).run()?;
+
+    let output = thin_dump!(&md).run()?;
+    assert!(from_utf8(&output.stdout)?.contains(pattern));
+    Ok(())
+}
+
+#[test]
+fn override_transaction_id() -> Result<()> {
+    override_something("--transaction-id", "2345", "transaction=\"2345\"")
+}
+
+#[test]
+fn override_data_block_size() -> Result<()> {
+    override_something("--data-block-size", "8192", "data_block_size=\"8192\"")
+}
+
+#[test]
+fn override_nr_data_blocks() -> Result<()> {
+    override_something("--nr-data-blocks", "234500", "nr_data_blocks=\"234500\"")
+}