use anyhow::Result; use std::fs::OpenOptions; use std::io::Write; use std::str::from_utf8; mod common; use common::common_args::*; use common::fixture::*; use common::input_arg::*; use common::process::*; use common::program::*; use common::target::*; use common::test_dir::*; use common::thin::*; //------------------------------------------ const USAGE: &str = "Usage: thin_dump [options] {device|file}\n\ Options:\n \ {-h|--help}\n \ {-f|--format} {xml|human_readable|custom}\n \ {-r|--repair}\n \ {-m|--metadata-snap} [block#]\n \ {-o }\n \ {--dev-id} \n \ {--skip-mappings}\n \ {-V|--version}"; //----------------------------------------- struct ThinDump; impl<'a> Program<'a> for ThinDump { fn name() -> &'a str { "thin_dump" } fn path() -> &'a std::ffi::OsStr { THIN_DUMP.as_ref() } fn usage() -> &'a str { USAGE } fn arg_type() -> ArgType { ArgType::InputArg } fn bad_option_hint(option: &str) -> String { msg::bad_option_hint(option) } } impl<'a> InputProgram<'a> for ThinDump { fn mk_valid_input(td: &mut TestDir) -> Result { mk_valid_md(td) } fn file_not_found() -> &'a str { msg::FILE_NOT_FOUND } fn missing_input_arg() -> &'a str { msg::MISSING_INPUT_ARG } fn corrupted_input() -> &'a str { msg::BAD_SUPERBLOCK } } //------------------------------------------ test_accepts_help!(ThinDump); test_accepts_version!(ThinDump); test_rejects_bad_option!(ThinDump); test_missing_input_arg!(ThinDump); test_input_file_not_found!(ThinDump); test_input_cannot_be_a_directory!(ThinDump); test_unreadable_input_file!(ThinDump); //------------------------------------------ // test dump & restore cycle #[test] fn dump_restore_cycle() -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; let output = run_ok_raw(THIN_DUMP, args![&md])?; let xml = td.mk_path("meta.xml"); let mut file = OpenOptions::new() .read(false) .write(true) .create(true) .open(&xml)?; file.write_all(&output.stdout[0..])?; drop(file); let md2 = mk_zeroed_md(&mut td)?; run_ok(THIN_RESTORE, args!["-i", &xml, "-o", &md2])?; let output2 = run_ok_raw(THIN_DUMP, args![&md2])?; assert_eq!(output.stdout, output2.stdout); Ok(()) } //------------------------------------------ // test no stderr with a normal dump #[test] fn no_stderr() -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; let output = run_ok_raw(THIN_DUMP, args![&md])?; assert_eq!(output.stderr.len(), 0); Ok(()) } //------------------------------------------ // test superblock overriding & repair // TODO: share with thin_repair fn override_something(flag: &str, value: &str, pattern: &str) -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; let output = run_ok_raw(THIN_DUMP, args![&md, flag, value])?; assert_eq!(output.stderr.len(), 0); assert!(from_utf8(&output.stdout[0..])?.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\"") } // FIXME: duplicate with superblock_succeeds in thin_repair.rs #[test] fn repair_superblock() -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; let before = run_ok_raw( THIN_DUMP, args![ "--transaction-id=5", "--data-block-size=128", "--nr-data-blocks=4096000", &md ], )?; damage_superblock(&md)?; let after = run_ok_raw( THIN_DUMP, args![ "--repair", "--transaction-id=5", "--data-block-size=128", "--nr-data-blocks=4096000", &md ], )?; assert_eq!(after.stderr.len(), 0); assert_eq!(before.stdout, after.stdout); Ok(()) } //------------------------------------------ // test compatibility between options // TODO: share with thin_repair #[test] fn missing_transaction_id() -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; damage_superblock(&md)?; let stderr = run_fail( THIN_DUMP, args![ "--repair", "--data-block-size=128", "--nr-data-blocks=4096000", &md ], )?; assert!(stderr.contains("transaction id")); Ok(()) } #[test] fn missing_data_block_size() -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; damage_superblock(&md)?; let stderr = run_fail( THIN_DUMP, args![ "--repair", "--transaction-id=5", "--nr-data-blocks=4096000", &md ], )?; assert!(stderr.contains("data block size")); Ok(()) } #[test] fn missing_nr_data_blocks() -> Result<()> { let mut td = TestDir::new()?; let md = mk_valid_md(&mut td)?; damage_superblock(&md)?; let stderr = run_fail( THIN_DUMP, args![ "--repair", "--transaction-id=5", "--data-block-size=128", &md ], )?; assert!(stderr.contains("nr data blocks")); Ok(()) } //------------------------------------------