diff options
author | Robin Krahl <robin.krahl@ireas.org> | 2020-09-07 20:47:41 +0200 |
---|---|---|
committer | Daniel Mueller <deso@posteo.net> | 2020-09-07 16:42:56 -0700 |
commit | 16f6b3ba0c3535efd1b9288ea1980cdd281b6565 (patch) | |
tree | 07e9605ab3cf81a91a860f3e9d61fccc254f0b47 /src | |
parent | a4dc74f23ed1ac45a5db497499c2ac334ab295ce (diff) | |
download | nitrocli-16f6b3ba0c3535efd1b9288ea1980cdd281b6565.tar.gz nitrocli-16f6b3ba0c3535efd1b9288ea1980cdd281b6565.tar.bz2 |
Add test cases for new connection logic
This patch adds test cases for the new connection logic:
- connect_multiple checks that nitrocli aborts if more than one Nitrokey
device is connected and no filter is set.
- connect_serial_number checks that we can use the --serial-number
option to connect to all supported Nitrokey devices.
- connect_wrong_serial_number checks that nitrocli aborts if it can’t
find a device with the requested serial number.
- connect_model checks that we can use the --model option to select the
Nitrokey device to connect to, and that nitrocli aborts if it finds no
matching device or multiple matching devices.
Diffstat (limited to 'src')
-rw-r--r-- | src/tests/run.rs | 73 |
1 files changed, 73 insertions, 0 deletions
diff --git a/src/tests/run.rs b/src/tests/run.rs index 4d470f6..4f53608 100644 --- a/src/tests/run.rs +++ b/src/tests/run.rs @@ -3,6 +3,7 @@ // Copyright (C) 2019-2020 The Nitrocli Developers // SPDX-License-Identifier: GPL-3.0-or-later +use std::collections; use std::path; use super::*; @@ -106,3 +107,75 @@ fn config_file() { assert_eq!(true, config.no_cache); assert_eq!(2, config.verbosity); } + +#[test_device] +fn connect_multiple(_model: nitrokey::Model) -> anyhow::Result<()> { + let devices = nitrokey::list_devices()?; + if devices.len() > 1 { + let res = Nitrocli::new().handle(&["status"]); + let err = res.unwrap_err().to_string(); + assert_eq!( + err, + "Multiple Nitrokey devices found. Use the --model and --serial-number options to select one" + ); + } + Ok(()) +} + +#[test_device] +fn connect_serial_number(_model: nitrokey::Model) -> anyhow::Result<()> { + let devices = nitrokey::list_devices()?; + for serial_number in devices.iter().filter_map(|d| d.serial_number) { + let res = Nitrocli::new().handle(&["status", &format!("--serial-number={}", serial_number)])?; + assert!(res.contains(&format!("serial number: {}\n", serial_number))); + } + Ok(()) +} + +#[test_device] +fn connect_wrong_serial_number(_model: nitrokey::Model) { + let res = Nitrocli::new().handle(&["status", "--serial-number=0xdeadbeef"]); + let err = res.unwrap_err().to_string(); + assert_eq!( + err, + "Nitrokey device not found (filter: serial number in [0xdeadbeef])" + ); +} + +#[test_device] +fn connect_model(_model: nitrokey::Model) -> anyhow::Result<()> { + let devices = nitrokey::list_devices()?; + let mut model_counts = collections::BTreeMap::new(); + let _ = model_counts.insert(nitrokey::Model::Pro.to_string(), 0); + let _ = model_counts.insert(nitrokey::Model::Storage.to_string(), 0); + for model in devices.iter().filter_map(|d| d.model) { + *model_counts.entry(model.to_string()).or_default() += 1; + } + + for (model, count) in model_counts { + let res = Nitrocli::new().handle(&["status", &format!("--model={}", model.to_lowercase())]); + if count == 0 { + let err = res.unwrap_err().to_string(); + assert_eq!( + err, + format!( + "Nitrokey device not found (filter: model={})", + model.to_lowercase() + ) + ); + } else if count == 1 { + assert!(res?.contains(&format!("model: {}\n", model))); + } else { + let err = res.unwrap_err().to_string(); + assert_eq!( + err, + format!( + "Multiple Nitrokey devices found (filter: model={}). ", + model.to_lowercase() + ) + "Use the --model and --serial-number options to select one" + ); + } + } + + Ok(()) +} |