github/darkfi
1
1
Fork 0
mirror of https://github.com/darkrenaissance/darkfi.git synced 2026-04-28 03:00:18 -04:00
Code Issues Packages Projects Releases Wiki Activity

walletdb: enabled test_get_balances

Browse Source
This commit is contained in:
lunar-mining
2021-10-07 16:43:31 +02:00
parent 1c1816e725
commit 7742f71a0e
1 changed files with 42 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files

85
src/wallet/walletdb.rs
View File

@@ -377,7 +377,8 @@ mod tests {
use super::*;
use crate::crypto::{coin::Coin, OwnCoin};
use crate::util::join_config_path;
use crate::serial::{deserialize, serialize};
use crate::util::{join_config_path, DrkTokenList, SolTokenList};
use ff::PrimeField;
pub fn init_db(path: &PathBuf, password: String) -> Result<()> {
@@ -452,61 +453,59 @@ mod tests {
Ok(())
}
//#[test]
//pub fn test_get_token_table() -> Result<()> {
// let walletdb_path = join_config_path(&PathBuf::from("test2_wallet.db"))?;
// let password: String = "darkfi".into();
// let wallet = WalletDb::new(&walletdb_path, password.clone())?;
// init_db(&walletdb_path, password)?;
#[test]
pub fn test_get_balances() -> Result<()> {
let walletdb_path = join_config_path(&PathBuf::from("test2_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;
// let key_public = serial::serialize(&public);
// let key_private = serial::serialize(&secret);
let secret: jubjub::Fr = jubjub::Fr::random(&mut OsRng);
let public = zcash_primitives::constants::SPENDING_KEY_GENERATOR * secret;
let key_public = serial::serialize(&public);
let key_private = serial::serialize(&secret);
// wallet.put_keypair(key_public, key_private)?;
wallet.put_keypair(key_public, key_private)?;
// let asset_id = jubjub::Fr::random(&mut OsRng);
let asset_id = jubjub::Fr::random(&mut OsRng);
// let note = Note {
// serial: jubjub::Fr::random(&mut OsRng),
// value: 110,
// asset_id,
// coin_blind: jubjub::Fr::random(&mut OsRng),
// valcom_blind: jubjub::Fr::random(&mut OsRng),
// };
let note = Note {
serial: jubjub::Fr::random(&mut OsRng),
value: 110,
asset_id,
coin_blind: jubjub::Fr::random(&mut OsRng),
valcom_blind: jubjub::Fr::random(&mut OsRng),
};
// let coin = Coin::new(bls12_381::Scalar::random(&mut OsRng).to_repr());
let coin = Coin::new(bls12_381::Scalar::random(&mut OsRng).to_repr());
// let mut tree = crate::crypto::merkle::CommitmentTree::empty();
// tree.append(MerkleNode::from_coin(&coin))?;
let mut tree = crate::crypto::merkle::CommitmentTree::empty();
tree.append(MerkleNode::from_coin(&coin))?;
// let witness = IncrementalWitness::from_tree(&tree);
let witness = IncrementalWitness::from_tree(&tree);
// let own_coin = OwnCoin {
// coin,
// note: note.clone(),
// secret,
// witness: witness.clone(),
// };
let own_coin = OwnCoin {
coin,
note: note.clone(),
secret,
witness: witness.clone(),
};
// wallet.put_own_coins(own_coin.clone())?;
// wallet.put_own_coins(own_coin.clone())?;
// wallet.put_own_coins(own_coin.clone())?;
// wallet.put_own_coins(own_coin.clone())?;
wallet.put_own_coins(own_coin.clone())?;
wallet.put_own_coins(own_coin.clone())?;
wallet.put_own_coins(own_coin.clone())?;
wallet.put_own_coins(own_coin.clone())?;
// let table_vec = wallet.get_balances()?;
let balances = wallet.get_balances()?;
// assert_eq!(table_vec.len(), 4);
// assert_eq!(table_vec[0].value, 110);
// assert_eq!(table_vec[0].token_id, asset_id);
// assert_eq!(table_vec[2].value, 110);
// assert_eq!(table_vec[2].token_id, asset_id);
let asset_id = serialize(&asset_id);
// std::fs::remove_file(walletdb_path)?;
assert_eq!(balances[&asset_id], 110);
// Ok(())
//}
std::fs::remove_file(walletdb_path)?;
Ok(())
}
#[test]
pub fn test_save_and_load_keypair() -> Result<()> {