use anyhow::Result; mod common; use common::cache::*; 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::*; //------------------------------------------ const USAGE: &str = "Usage: cache_restore [options]\n\ Options:\n \ {-h|--help}\n \ {-i|--input} \n \ {-o|--output} \n \ {-q|--quiet}\n \ {--metadata-version} <1 or 2>\n \ {-V|--version}\n\ \n \ {--debug-override-metadata-version} \n \ {--omit-clean-shutdown}"; //------------------------------------------ struct CacheRestore; impl<'a> Program<'a> for CacheRestore { fn name() -> &'a str { "thin_restore" } fn path() -> &'a std::ffi::OsStr { CACHE_RESTORE.as_ref() } fn usage() -> &'a str { USAGE } fn arg_type() -> ArgType { ArgType::IoOptions } fn bad_option_hint(option: &str) -> String { msg::bad_option_hint(option) } } impl<'a> InputProgram<'a> for CacheRestore { fn mk_valid_input(td: &mut TestDir) -> Result { mk_valid_xml(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 { "" // we don't intent to verify error messages of XML parsing } } impl<'a> OutputProgram<'a> for CacheRestore { fn missing_output_arg() -> &'a str { msg::MISSING_OUTPUT_ARG } } impl<'a> MetadataWriter<'a> for CacheRestore { fn file_not_found() -> &'a str { msg::FILE_NOT_FOUND } } //----------------------------------------- test_accepts_help!(CacheRestore); test_accepts_version!(CacheRestore); test_missing_input_option!(CacheRestore); test_input_file_not_found!(CacheRestore); test_corrupted_input_data!(CacheRestore); test_missing_output_option!(CacheRestore); test_tiny_output_file!(CacheRestore); test_unwritable_output_file!(CacheRestore); //----------------------------------------- // TODO: share with thin_restore, era_restore fn quiet_flag(flag: &str) -> Result<()> { let mut td = TestDir::new()?; let xml = mk_valid_xml(&mut td)?; let md = mk_zeroed_md(&mut td)?; let output = run_ok_raw(CACHE_RESTORE, args!["-i", &xml, "-o", &md, flag])?; 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") } //----------------------------------------- #[test] fn successfully_restores() -> Result<()> { let mut td = TestDir::new()?; let xml = mk_valid_xml(&mut td)?; let md = mk_zeroed_md(&mut td)?; run_ok(CACHE_RESTORE, args!["-i", &xml, "-o", &md])?; Ok(()) } #[test] fn override_metadata_version() -> Result<()> { let mut td = TestDir::new()?; let xml = mk_valid_xml(&mut td)?; let md = mk_zeroed_md(&mut td)?; run_ok( CACHE_RESTORE, args![ "-i", &xml, "-o", &md, "--debug-override-metadata-version", "10298" ], )?; Ok(()) } #[test] fn accepts_omit_clean_shutdown() -> Result<()> { let mut td = TestDir::new()?; let xml = mk_valid_xml(&mut td)?; let md = mk_zeroed_md(&mut td)?; run_ok( CACHE_RESTORE, args!["-i", &xml, "-o", &md, "--omit-clean-shutdown"], )?; Ok(()) } //-----------------------------------------