diff options
Diffstat (limited to 'nitrocli/src/tests/mod.rs')
-rw-r--r-- | nitrocli/src/tests/mod.rs | 10 |
1 files changed, 10 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 { |