From 83063599f4ab1bcbbd9be9166e738a13ae4e4cc6 Mon Sep 17 00:00:00 2001 From: Robin Krahl Date: Sat, 5 Jan 2019 09:52:29 +0000 Subject: Fix example for GenerateOtp::get_totp_code --- src/otp.rs | 22 +++++++++++----------- tests/otp.rs | 5 ++++- 2 files changed, 15 insertions(+), 12 deletions(-) diff --git a/src/otp.rs b/src/otp.rs index 9f0a388..901bef9 100644 --- a/src/otp.rs +++ b/src/otp.rs @@ -155,7 +155,7 @@ pub trait GenerateOtp { /// /// `time` is the number of seconds since January 1st, 1970 (Unix timestamp). Unless `force` /// is set to `true`, this command fails if the timestamp on the device is larger than the - /// given timestamp or if it is zero. + /// given timestamp or if it is zero. /// /// The time is used for TOTP generation (see [`get_totp_code`][]). /// @@ -297,21 +297,21 @@ pub trait GenerateOtp { /// /// # Example /// - /// ```ignore - /// extern crate chrono; - /// + /// ```no_run + /// use std::time; /// use nitrokey::GenerateOtp; /// # use nitrokey::CommandError; /// /// # fn try_main() -> Result<(), CommandError> { /// let device = nitrokey::connect()?; - /// let time = Utc::now().timestamp(); - /// if time < 0 { - /// println!("Timestamps before 1970-01-01 are not supported!"); - /// } else { - /// device.set_time(time as u64); - /// let code = device.get_totp_code(1)?; - /// println!("Generated TOTP code on slot 1: {}", code); + /// let time = time::SystemTime::now().duration_since(time::UNIX_EPOCH); + /// match time { + /// Ok(time) => { + /// device.set_time(time.as_secs(), false)?; + /// let code = device.get_totp_code(1)?; + /// println!("Generated TOTP code on slot 1: {}", code); + /// }, + /// Err(_) => println!("Timestamps before 1970-01-01 are not supported!"), /// } /// # Ok(()) /// # } diff --git a/tests/otp.rs b/tests/otp.rs index 8124c76..2b46088 100644 --- a/tests/otp.rs +++ b/tests/otp.rs @@ -63,7 +63,10 @@ fn check_hotp_codes(device: &GenerateOtp, offset: u8) { fn set_time(device: DeviceWrapper) { assert_eq!(Ok(()), device.set_time(1546385382, true)); assert_eq!(Ok(()), device.set_time(1546385392, false)); - assert_eq!(Err(CommandError::Timestamp), device.set_time(1546385292, false)); + assert_eq!( + Err(CommandError::Timestamp), + device.set_time(1546385292, false) + ); assert_eq!(Ok(()), device.set_time(1546385382, true)); } -- cgit v1.2.1