diff options
Diffstat (limited to 'nitrocli/src/tests')
-rw-r--r-- | nitrocli/src/tests/mod.rs | 10 | ||||
-rw-r--r-- | nitrocli/src/tests/otp.rs | 42 |
2 files changed, 52 insertions, 0 deletions
diff --git a/nitrocli/src/tests/mod.rs b/nitrocli/src/tests/mod.rs index ef8d2bd..4a4ab09 100644 --- a/nitrocli/src/tests/mod.rs +++ b/nitrocli/src/tests/mod.rs @@ -36,6 +36,7 @@ const NITROKEY_DEFAULT_USER_PIN: &str = "123456"; #[test_device] fn dummy() {} +mod otp; mod pin; mod run; mod status; @@ -44,6 +45,8 @@ mod status; pub trait UnwrapError { /// Unwrap an Error::Error variant. fn unwrap_str_err(self) -> String; + /// Unwrap a Error::CommandError variant. + fn unwrap_cmd_err(self) -> (Option<&'static str>, nitrokey::CommandError); } impl<T> UnwrapError for crate::Result<T> @@ -56,6 +59,13 @@ where err => panic!("Unexpected error variant found: {:?}", err), } } + + fn unwrap_cmd_err(self) -> (Option<&'static str>, nitrokey::CommandError) { + match self.unwrap_err() { + crate::Error::CommandError(ctx, err) => (ctx, err), + err => panic!("Unexpected error variant found: {:?}", err), + } + } } struct Nitrocli { diff --git a/nitrocli/src/tests/otp.rs b/nitrocli/src/tests/otp.rs new file mode 100644 index 0000000..403cc80 --- /dev/null +++ b/nitrocli/src/tests/otp.rs @@ -0,0 +1,42 @@ +// otp.rs + +// ************************************************************************* +// * Copyright (C) 2019 Daniel Mueller (deso@posteo.net) * +// * * +// * This program is free software: you can redistribute it and/or modify * +// * it under the terms of the GNU General Public License as published by * +// * the Free Software Foundation, either version 3 of the License, or * +// * (at your option) any later version. * +// * * +// * This program is distributed in the hope that it will be useful, * +// * but WITHOUT ANY WARRANTY; without even the implied warranty of * +// * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * +// * GNU General Public License for more details. * +// * * +// * You should have received a copy of the GNU General Public License * +// * along with this program. If not, see <http://www.gnu.org/licenses/>. * +// ************************************************************************* + +use super::*; + +#[test_device] +fn set_invalid_slot_raw(device: nitrokey::DeviceWrapper) { + let (rc, out, err) = Nitrocli::with_dev(device).run(&["otp", "set", "100", "name", "1234"]); + + assert_ne!(rc, 0); + assert_eq!(out, b""); + assert_eq!(&err[..24], b"Could not write OTP slot"); +} + +#[test_device] +fn set_invalid_slot(device: nitrokey::DeviceWrapper) { + let res = Nitrocli::with_dev(device).handle(&["otp", "set", "100", "name", "1234"]); + + assert_eq!( + res.unwrap_cmd_err(), + ( + Some("Could not write OTP slot"), + nitrokey::CommandError::InvalidSlot + ) + ); +} |