fix readme test

Author: lonerapier

README.md

@@ -116,25 +116,39 @@ cargo run --bin uniffi-bindgen generate --library target/debug/libmfkdf2.dylib -
 
 ```rust
 # use std::collections::HashMap;
-use mfkdf2::{derive, setup};
-use mfkdf2::setup::{factors::hotp::HOTPOptions, password::PasswordOptions, key::MFKDF2Options};
-use mfkdf2::derive::factors::{hotp::HOTPOptions, password::PasswordOptions};
-
-// 1. Define factors
-let password_factor = setup::password("my-super-secret-password", PasswordOptions::default())?;
-let totp_factor = setup::hotp("base32-encoded-secret", HOTPOptions::default())?;
-
-// 2. Set up the key with the policy
-let key = setup::key(vec![password_factor, totp_factor], MFKDF2Options::default())?;
-
-println!("Key: {:?}", key);
-
-let factors = HashMap::from([
-  ("password".to_string(), derive::factors::password("my-super-secret-password")?),
-  "hotp".to_string(), derive::factors::hotp("123456")?),
-]);
-// 3. Derive the key using user inputs
-let derived_key = derive::key(&key.policy, factors, true, false)?;
+use mfkdf2::setup::factors::{totp::TOTPOptions, password::PasswordOptions};
+use mfkdf2::definitions::MFKDF2Options;
+
+let setup = mfkdf2::setup::key(
+  &[
+    mfkdf2::setup::factors::password("password1", PasswordOptions::default())?,
+    mfkdf2::setup::factors::totp(TOTPOptions {
+      secret: Some(b"abcdefghijklmnopqrst".to_vec()),
+      time: Some(1),
+      ..Default::default()
+    })?,
+  ],
+  MFKDF2Options::default(),
+)?;
+
+let derived_key = mfkdf2::derive::key(
+  &setup.policy,
+  HashMap::from([
+    ("password".to_string(), mfkdf2::derive::factors::password("password1")?),
+    (
+      "totp".to_string(),
+      mfkdf2::derive::factors::totp(
+        241063,
+        Some(mfkdf2::derive::factors::totp::TOTPDeriveOptions {
+          time: Some(30001),
+          ..Default::default()
+        }),
+      )?,
+    ),
+  ]),
+  true,
+  false,
+)?;
 
 println!("Derived Key: {:?}", derived_key);
 

mfkdf2/README.md

@@ -142,6 +142,8 @@ After you have setup a key policy, you can derive the key from the policy and th
 
 ## Examples
 
+### Password + HOTP + HMACSHA1
+
 Derive a composite key with password, hmacsha1 and hotp factors. Derive returns the
 [MFKDF2DerivedKey](`crate::definitions::MFKDF2DerivedKey`) and updated [Policy](`crate::policy::Policy`).
 
@@ -222,6 +224,50 @@ let derived_key = derive::key(
 # Ok::<(), mfkdf2::error::MFKDF2Error>(())
 ```
 
+### Password + TOTP
+
+```rust
+# use std::collections::HashMap;
+use mfkdf2::setup::factors::{totp::TOTPOptions, password::PasswordOptions};
+use mfkdf2::definitions::MFKDF2Options;
+
+let setup = mfkdf2::setup::key(
+  &[
+    mfkdf2::setup::factors::password("password1", PasswordOptions::default())?,
+    mfkdf2::setup::factors::totp(TOTPOptions {
+      secret: Some(b"abcdefghijklmnopqrst".to_vec()),
+      time: Some(1),
+      ..Default::default()
+    })?,
+  ],
+  MFKDF2Options::default(),
+)?;
+
+let derived_key = mfkdf2::derive::key(
+  &setup.policy,
+  HashMap::from([
+    ("password".to_string(), mfkdf2::derive::factors::password("password1")?),
+    (
+      "totp".to_string(),
+      mfkdf2::derive::factors::totp(
+        241063,
+        Some(mfkdf2::derive::factors::totp::TOTPDeriveOptions {
+          time: Some(30001),
+          ..Default::default()
+        }),
+      )?,
+    ),
+  ]),
+  true,
+  false,
+)?;
+
+println!("Derived Key: {:?}", derived_key);
+
+# assert_eq!(setup.key, derived_key.key);
+# Ok::<(), mfkdf2::error::MFKDF2Error>(())
+```
+
 # Threshold Recovery
 
 Threshold recovery generalizes a multi‑factor policy from “all factors required” to a