Refactor test cases

1 files changed, 19 insertions, 31 deletions

diff --git a/src/tests/pro.rs b/src/tests/pro.rs
index 76d68a0..8394e47 100644
--- a/src/tests/pro.rs
+++ b/src/tests/pro.rs
@@ -24,6 +24,15 @@ static TOTP_CODES: &[(u64, &str)] = &[
     (20000000000, "65353130"),
 ];
 
+fn get_test_device() -> UnauthenticatedDevice {
+    set_debug(false);
+    ::connect_model(Model::Pro).expect("Could not connect to the Nitrokey Pro.")
+}
+
+fn get_admin_test_device() -> AdminAuthenticatedDevice {
+    get_test_device().authenticate_admin(ADMIN_PASSWORD).expect("Could not login as admin.")
+}
+
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn connect() {
@@ -49,8 +58,7 @@ fn disconnect() {
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn get_serial_number() {
-    set_debug(false);
-    let device = ::connect().unwrap();
+    let device = get_test_device();
     let result = device.get_serial_number();
     assert!(result.is_ok());
     let serial_number = result.unwrap();
@@ -76,9 +84,7 @@ where
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn hotp() {
-    set_debug(false);
-    let device = ::connect().unwrap();
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
+    let admin = get_admin_test_device();
     let config = Config::new(None, None, None, false);
     assert_eq!(CommandStatus::Success, admin.write_config(config));
 
@@ -92,9 +98,7 @@ fn hotp() {
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn hotp_pin() {
-    set_debug(false);
-    let device = ::connect().unwrap();
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
+    let admin = get_admin_test_device();
     let config = Config::new(None, None, None, true);
     assert_eq!(CommandStatus::Success, admin.write_config(config));
 
@@ -109,10 +113,7 @@ fn hotp_pin() {
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn hotp_slot_name() {
-    set_debug(false);
-    let device = ::connect().unwrap();
-
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
+    let admin = get_admin_test_device();
     let slot_data = OtpSlotData::new(1, "test-hotp", HOTP_SECRET, OtpMode::SixDigits);
     assert_eq!(CommandStatus::Success, admin.write_hotp_slot(slot_data, 0));
 
@@ -146,9 +147,7 @@ where
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn totp() {
     // TODO: this test may fail due to bad timing --> find solution
-    set_debug(false);
-    let device = ::connect().unwrap();
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
+    let admin = get_admin_test_device();
     let config = Config::new(None, None, None, false);
     assert_eq!(CommandStatus::Success, admin.write_config(config));
 
@@ -163,9 +162,7 @@ fn totp() {
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn totp_pin() {
     // TODO: this test may fail due to bad timing --> find solution
-    set_debug(false);
-    let device = ::connect().unwrap();
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
+    let admin = get_admin_test_device();
     let config = Config::new(None, None, None, true);
     assert_eq!(CommandStatus::Success, admin.write_config(config));
 
@@ -180,10 +177,7 @@ fn totp_pin() {
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn totp_slot_name() {
-    set_debug(false);
-    let device = ::connect().unwrap();
-
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
+    let admin = get_admin_test_device();
     let slot_data = OtpSlotData::new(1, "test-totp", TOTP_SECRET, OtpMode::EightDigits);
     assert_eq!(CommandStatus::Success, admin.write_totp_slot(slot_data, 0));
 
@@ -195,8 +189,7 @@ fn totp_slot_name() {
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn get_firmware_version() {
-    set_debug(false);
-    let device = ::connect().unwrap();
+    let device = get_test_device();
     assert_eq!(0, device.get_major_firmware_version());
     let minor = device.get_minor_firmware_version();
     assert!(minor == 7 || minor == 8);
@@ -225,8 +218,7 @@ fn user_retry(device: UnauthenticatedDevice, suffix: &str, count: u8) -> Unauthe
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn get_retry_count() {
-    set_debug(false);
-    let device = ::connect().unwrap();
+    let device = get_test_device();
 
     let device = admin_retry(device, "", 3);
     let device = admin_retry(device, "123", 2);
@@ -242,11 +234,7 @@ fn get_retry_count() {
 #[test]
 #[cfg_attr(not(feature = "test-pro"), ignore)]
 fn read_write_config() {
-    set_debug(false);
-    let device = ::connect().unwrap();
-
-    let admin = device.authenticate_admin(ADMIN_PASSWORD).unwrap();
-
+    let admin = get_admin_test_device();
     let config = Config::new(None, None, None, true);
     assert_eq!(CommandStatus::Success, admin.write_config(config));
     let get_config = admin.get_config().unwrap();