From 3a99e7417e4c4eb30df9e711077c89b75764c029 Mon Sep 17 00:00:00 2001
From: Daniel Mueller <deso@posteo.net>
Date: Wed, 9 Jan 2019 11:16:03 -0800
Subject: Report command errors properly

So far we have taken all nitrokey::CommandError objects and put them in
formatted form into the Error::Error variant. What we really should do,
though, is to preserve the original error, with the additional context
provided by the caller, and report that up the stack directly. Doing so
has at least the benefit that we are able to check for expected errors
without hard coding the textual representation as maintained by the
nitrokey create.
This change refactors the code accordingly and adds two tests for such
expected error codes.
---
 nitrocli/src/tests/mod.rs | 10 ++++++++++
 nitrocli/src/tests/otp.rs | 42 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 52 insertions(+)
 create mode 100644 nitrocli/src/tests/otp.rs

(limited to 'nitrocli/src/tests')

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
+    )
+  );
+}
-- 
cgit v1.2.1