108 lines
2.4 KiB
Rust
108 lines
2.4 KiB
Rust
use anyhow::Result;
|
|
|
|
mod common;
|
|
|
|
use common::common_args::*;
|
|
use common::process::*;
|
|
use common::program::*;
|
|
use common::target::*;
|
|
use common::test_dir::*;
|
|
use common::thin::*;
|
|
|
|
//------------------------------------------
|
|
|
|
const USAGE: &str = "Usage: thin_rmap [options] {device|file}\n\
|
|
Options:\n \
|
|
{-h|--help}\n \
|
|
{-V|--version}\n \
|
|
{--region <block range>}*\n\
|
|
Where:\n \
|
|
<block range> is of the form <begin>..<one-past-the-end>\n \
|
|
for example 5..45 denotes blocks 5 to 44 inclusive, but not block 45";
|
|
|
|
//------------------------------------------
|
|
|
|
struct ThinRmap;
|
|
|
|
impl<'a> Program<'a> for ThinRmap {
|
|
fn name() -> &'a str {
|
|
"thin_rmap"
|
|
}
|
|
|
|
fn cmd<I>(args: I) -> Command
|
|
where
|
|
I: IntoIterator,
|
|
I::Item: Into<std::ffi::OsString>,
|
|
{
|
|
thin_rmap_cmd(args)
|
|
}
|
|
|
|
fn usage() -> &'a str {
|
|
USAGE
|
|
}
|
|
|
|
fn arg_type() -> ArgType {
|
|
ArgType::InputArg
|
|
}
|
|
|
|
fn bad_option_hint(option: &str) -> String {
|
|
msg::bad_option_hint(option)
|
|
}
|
|
}
|
|
|
|
//------------------------------------------
|
|
|
|
test_accepts_help!(ThinRmap);
|
|
test_accepts_version!(ThinRmap);
|
|
test_rejects_bad_option!(ThinRmap);
|
|
|
|
//------------------------------------------
|
|
|
|
#[test]
|
|
fn valid_region_format_should_pass() -> Result<()> {
|
|
let mut td = TestDir::new()?;
|
|
let md = mk_valid_md(&mut td)?;
|
|
run_ok(thin_rmap_cmd(args!["--region", "23..7890", &md]))?;
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn invalid_regions_should_fail() -> Result<()> {
|
|
let invalid_regions = [
|
|
"23,7890",
|
|
"23..six",
|
|
"found..7890",
|
|
"89..88",
|
|
"89..89",
|
|
"89..",
|
|
"",
|
|
"89...99",
|
|
];
|
|
for r in &invalid_regions {
|
|
let mut td = TestDir::new()?;
|
|
let md = mk_valid_md(&mut td)?;
|
|
run_fail(thin_rmap_cmd(args![r, &md]))?;
|
|
}
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn multiple_regions_should_pass() -> Result<()> {
|
|
let mut td = TestDir::new()?;
|
|
let md = mk_valid_md(&mut td)?;
|
|
run_ok(thin_rmap_cmd(args![
|
|
"--region", "1..23", "--region", "45..78", &md
|
|
]))?;
|
|
Ok(())
|
|
}
|
|
|
|
#[test]
|
|
fn junk_input() -> Result<()> {
|
|
let mut td = TestDir::new()?;
|
|
let xml = mk_valid_xml(&mut td)?;
|
|
run_fail(thin_rmap_cmd(args!["--region", "0..-1", &xml]))?;
|
|
Ok(())
|
|
}
|
|
|
|
//------------------------------------------
|