use anyhow::Result; mod common; use common::common_args::*; use common::fixture::*; use common::input_arg::*; use common::output_option::*; use common::process::*; use common::program::*; use common::target::*; use common::test_dir::*; use common::thin::*; //------------------------------------------ const USAGE: &str = "Usage: thin_repair [options] {device|file}\n\ Options:\n \ {-h|--help}\n \ {-i|--input} \n \ {-o|--output} \n \ {--transaction-id} \n \ {--data-block-size} \n \ {--nr-data-blocks} \n \ {-V|--version}"; //----------------------------------------- struct ThinRepair; impl<'a> Program<'a> for ThinRepair { fn name() -> &'a str { "thin_repair" } fn path() -> &'a std::ffi::OsStr { THIN_REPAIR.as_ref() } fn usage() -> &'a str { USAGE } fn arg_type() -> ArgType { ArgType::IoOptions } fn bad_option_hint(option: &str) -> String { cpp_msg::bad_option_hint(option) } } impl<'a> InputProgram<'a> for ThinRepair { fn mk_valid_input(td: &mut TestDir) -> Result { mk_valid_md(td) } fn file_not_found() -> &'a str { cpp_msg::FILE_NOT_FOUND } fn missing_input_arg() -> &'a str { cpp_msg::MISSING_INPUT_ARG } fn corrupted_input() -> &'a str { "The following field needs to be provided on the command line due to corruption in the superblock" } } impl<'a> OutputProgram<'a> for ThinRepair { fn file_not_found() -> &'a str { cpp_msg::FILE_NOT_FOUND } fn missing_output_arg() -> &'a str { cpp_msg::MISSING_OUTPUT_ARG } } //----------------------------------------- test_accepts_help!(ThinRepair); test_accepts_version!(ThinRepair); test_rejects_bad_option!(ThinRepair); test_input_file_not_found!(ThinRepair); test_input_cannot_be_a_directory!(ThinRepair); test_corrupted_input_data!(ThinRepair); test_missing_output_option!(ThinRepair); //----------------------------------------- // test output to a small file // TODO: share with thin_restore #[test] fn dont_repair_xml() -> Result<()> { let mut td = TestDir::new()?; let md = mk_zeroed_md(&mut td)?; let xml = mk_valid_xml(&mut td)?; run_fail(THIN_REPAIR, args!["-i", &xml, "-o", &md])?; Ok(()) } //----------------------------------------- // TODO: share with thin_dump fn override_thing(flag: &str, val: &str, pattern: &str) -> Result<()> { let mut td = TestDir::new()?; let md1 = mk_valid_md(&mut td)?; let md2 = mk_zeroed_md(&mut td)?; let output = run_ok_raw(THIN_REPAIR, args![flag, val, "-i", &md1, "-o", &md2])?; assert_eq!(output.stderr.len(), 0); let output = run_ok(THIN_DUMP, args![&md2])?; assert!(output.contains(pattern)); Ok(()) } #[test] fn override_transaction_id() -> Result<()> { override_thing("--transaction-id", "2345", "transaction=\"2345\"") } #[test] fn override_data_block_size() -> Result<()> { override_thing("--data-block-size", "8192", "data_block_size=\"8192\"") } #[test] fn override_nr_data_blocks() -> Result<()> { override_thing("--nr-data-blocks", "234500", "nr_data_blocks=\"234500\"") } // FIXME: that's repair_superblock in thin_dump.rs #[test] fn superblock_succeeds() -> Result<()> { let mut td = TestDir::new()?; let md1 = mk_valid_md(&mut td)?; let original = run_ok_raw( THIN_DUMP, args![ "--transaction-id=5", "--data-block-size=128", "--nr-data-blocks=4096000", &md1 ], )?; assert_eq!(original.stderr.len(), 0); damage_superblock(&md1)?; let md2 = mk_zeroed_md(&mut td)?; run_ok( THIN_REPAIR, args![ "--transaction-id=5", "--data-block-size=128", "--nr-data-blocks=4096000", "-i", &md1, "-o", &md2 ], )?; let repaired = run_ok_raw(THIN_DUMP, args![&md2])?; assert_eq!(repaired.stderr.len(), 0); assert_eq!(original.stdout, repaired.stdout); Ok(()) } //----------------------------------------- // TODO: share with thin_dump fn missing_thing(flag1: &str, flag2: &str, pattern: &str) -> Result<()> { let mut td = TestDir::new()?; let md1 = mk_valid_md(&mut td)?; damage_superblock(&md1)?; let md2 = mk_zeroed_md(&mut td)?; let stderr = run_fail(THIN_REPAIR, args![flag1, flag2, "-i", &md1, "-o", &md2])?; assert!(stderr.contains(pattern)); Ok(()) } #[test] fn missing_transaction_id() -> Result<()> { missing_thing( "--data-block-size=128", "--nr-data-blocks=4096000", "transaction id", ) } #[test] fn missing_data_block_size() -> Result<()> { missing_thing( "--transaction-id=5", "--nr-data-blocks=4096000", "data block size", ) } #[test] fn missing_nr_data_blocks() -> Result<()> { missing_thing( "--transaction-id=5", "--data-block-size=128", "nr data blocks", ) } //-----------------------------------------