From e78f59801d3140fae0a1eb69725c98d26745617e Mon Sep 17 00:00:00 2001 From: ghassmo Date: Sat, 2 Oct 2021 18:43:03 +0300 Subject: [PATCH] walletdb: add test case for load and save cashier public key --- src/wallet/walletdb.rs | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/src/wallet/walletdb.rs b/src/wallet/walletdb.rs index b0033f63f..d66027ea0 100644 --- a/src/wallet/walletdb.rs +++ b/src/wallet/walletdb.rs @@ -639,4 +639,24 @@ mod tests { Ok(()) } + + #[test] + pub fn test_put_and_get_cashier_public_key() -> Result<()> { + let walletdb_path = join_config_path(&PathBuf::from("test6_wallet.db"))?; + let password: String = "darkfi".into(); + let wallet = WalletDb::new(&walletdb_path, password.clone())?; + init_db(&walletdb_path, password)?; + + let secret: jubjub::Fr = jubjub::Fr::random(&mut OsRng); + let public = zcash_primitives::constants::SPENDING_KEY_GENERATOR * secret; + + wallet.put_cashier_pub(&public)?; + let cashier_public = wallet.get_cashier_public_keys()?[0]; + + assert_eq!(cashier_public, public); + + std::fs::remove_file(walletdb_path)?; + + Ok(()) + } }