diff options
-rw-r--r-- | nitrocli/src/tests/mod.rs | 86 | ||||
-rw-r--r-- | nitrocli/src/tests/run.rs | 5 | ||||
-rw-r--r-- | nitrocli/src/tests/status.rs | 7 |
3 files changed, 39 insertions, 59 deletions
diff --git a/nitrocli/src/tests/mod.rs b/nitrocli/src/tests/mod.rs index cb0d09e..c7ff222 100644 --- a/nitrocli/src/tests/mod.rs +++ b/nitrocli/src/tests/mod.rs @@ -33,38 +33,6 @@ fn dummy() {} mod run; mod status; -/// An `Option<IntoArg>` that represents a non-present device. Rust can -/// be notoriously bad at inferring type parameters and this constant -/// alleviates the pain. -const NO_DEV: Option<nitrokey::Pro> = None; - -/// A trait for conversion of a nitrokey::Device into an argument -/// representing the device model that the program recognizes. -pub trait IntoArg { - fn into_arg(self) -> &'static str; -} - -impl IntoArg for nitrokey::Pro { - fn into_arg(self) -> &'static str { - "--model=pro" - } -} - -impl IntoArg for nitrokey::Storage { - fn into_arg(self) -> &'static str { - "--model=storage" - } -} - -impl IntoArg for nitrokey::DeviceWrapper { - fn into_arg(self) -> &'static str { - match self { - nitrokey::DeviceWrapper::Pro(x) => x.into_arg(), - nitrokey::DeviceWrapper::Storage(x) => x.into_arg(), - } - } -} - /// A trait simplifying checking for expected errors. pub trait UnwrapError { /// Unwrap an Error::Error variant. @@ -83,22 +51,42 @@ where } } -mod nitrocli { - use super::*; +struct Nitrocli { + model: Option<nitrokey::Model>, +} - use crate::args; - use crate::Result; - use crate::RunCtx; +impl Nitrocli { + pub fn new() -> Self { + Self { model: None } + } - fn do_run<F, R, I>(device: Option<I>, args: &[&'static str], f: F) -> (R, Vec<u8>, Vec<u8>) + pub fn with_dev<D>(device: D) -> Self where - F: FnOnce(&mut RunCtx<'_>, Vec<String>) -> R, - I: IntoArg, + D: nitrokey::Device, + { + let result = Self { + model: Some(device.get_model()), + }; + + drop(device); + result + } + + fn model_to_arg(model: nitrokey::Model) -> &'static str { + match model { + nitrokey::Model::Pro => "--model=pro", + nitrokey::Model::Storage => "--model=storage", + } + } + + fn do_run<F, R>(&mut self, args: &[&'static str], f: F) -> (R, Vec<u8>, Vec<u8>) + where + F: FnOnce(&mut crate::RunCtx<'_>, Vec<String>) -> R, { let args = ["nitrocli"] .into_iter() .cloned() - .chain(device.into_iter().map(IntoArg::into_arg)) + .chain(self.model.map(Self::model_to_arg)) .chain(args.into_iter().cloned()) .map(ToOwned::to_owned) .collect(); @@ -106,7 +94,7 @@ mod nitrocli { let mut stdout = Vec::new(); let mut stderr = Vec::new(); - let ctx = &mut RunCtx { + let ctx = &mut crate::RunCtx { stdout: &mut stdout, stderr: &mut stderr, }; @@ -115,19 +103,13 @@ mod nitrocli { } /// Run `nitrocli`'s `run` function. - pub fn run<I>(device: Option<I>, args: &[&'static str]) -> (i32, Vec<u8>, Vec<u8>) - where - I: IntoArg, - { - do_run(device, args, |c, a| crate::run(c, a)) + pub fn run(&mut self, args: &[&'static str]) -> (i32, Vec<u8>, Vec<u8>) { + self.do_run(args, |c, a| crate::run(c, a)) } /// Run `nitrocli`'s `handle_arguments` function. - pub fn handle<I>(device: Option<I>, args: &[&'static str]) -> Result<String> - where - I: IntoArg, - { - let (res, out, _) = do_run(device, args, |c, a| args::handle_arguments(c, a)); + pub fn handle(&mut self, args: &[&'static str]) -> crate::Result<String> { + let (res, out, _) = self.do_run(args, |c, a| crate::args::handle_arguments(c, a)); res.map(|_| String::from_utf8_lossy(&out).into_owned()) } } diff --git a/nitrocli/src/tests/run.rs b/nitrocli/src/tests/run.rs index 51c2b87..260a851 100644 --- a/nitrocli/src/tests/run.rs +++ b/nitrocli/src/tests/run.rs @@ -18,11 +18,10 @@ // ************************************************************************* use super::*; -use crate::tests::nitrocli; #[test] fn no_command_or_option() { - let (rc, out, err) = nitrocli::run(NO_DEV, &[]); + let (rc, out, err) = Nitrocli::new().run(&[]); assert_ne!(rc, 0); assert_eq!(out, b""); @@ -34,7 +33,7 @@ fn no_command_or_option() { #[test] fn help_option() { fn test(opt: &'static str) { - let (rc, out, err) = nitrocli::run(NO_DEV, &[opt]); + let (rc, out, err) = Nitrocli::new().run(&[opt]); assert_eq!(rc, 0); assert_eq!(err, b""); diff --git a/nitrocli/src/tests/status.rs b/nitrocli/src/tests/status.rs index 838ebc9..acd33dc 100644 --- a/nitrocli/src/tests/status.rs +++ b/nitrocli/src/tests/status.rs @@ -18,13 +18,12 @@ // ************************************************************************* use super::*; -use crate::tests::nitrocli; // This test acts as verification that conversion of Error::Error // variants into the proper exit code works properly. #[test_device] fn not_found_raw() { - let (rc, out, err) = nitrocli::run(NO_DEV, &["status"]); + let (rc, out, err) = Nitrocli::new().run(&["status"]); assert_ne!(rc, 0); assert_eq!(out, b""); @@ -33,7 +32,7 @@ fn not_found_raw() { #[test_device] fn not_found() { - let res = nitrocli::handle(NO_DEV, &["status"]); + let res = Nitrocli::new().handle(&["status"]); assert_eq!(res.unwrap_str_err(), "Nitrokey device not found"); } @@ -50,7 +49,7 @@ $"#, ) .unwrap(); - let out = nitrocli::handle(Some(device), &["status"])?; + let out = Nitrocli::with_dev(device).handle(&["status"])?; assert!(re.is_match(&out), out); Ok(()) } |