1 files changed, 13 insertions, 0 deletions
diff --git a/nitrocli/src/tests/mod.rs b/nitrocli/src/tests/mod.rs
index 4e25091..ef8d2bd 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 pin;
 mod run;
 mod status;
 
@@ -92,6 +93,14 @@ impl Nitrocli {
     result
   }
 
+  pub fn user_pin(&mut self, pin: impl Into<ffi::OsString>) {
+    self.user_pin = Some(pin.into())
+  }
+
+  pub fn new_user_pin(&mut self, pin: impl Into<ffi::OsString>) {
+    self.new_user_pin = Some(pin.into())
+  }
+
   fn model_to_arg(model: nitrokey::Model) -> &'static str {
     match model {
       nitrokey::Model::Pro => "--model=pro",
@@ -136,4 +145,8 @@ impl Nitrocli {
     let (res, out, _) = self.do_run(args, |c, a| crate::args::handle_arguments(c, a));
     res.map(|_| String::from_utf8_lossy(&out).into_owned())
   }
+
+  pub fn model(&self) -> Option<nitrokey::Model> {
+    self.model
+  }
 }