Fixing tests

This commit is contained in:
Tomasz Drwięga 2016-11-30 16:21:57 +01:00
parent c028f106b1
commit 022ccb5bce
3 changed files with 9 additions and 5 deletions

View File

@ -81,10 +81,11 @@ impl KeyDirectory for NullDir {
}
fn insert(&self, account: SafeAccount) -> Result<SafeAccount, SSError> {
self.accounts.write()
.entry(account.address.clone())
.or_insert_with(Vec::new)
.push(account.clone());
let mut lock = self.accounts.write();
let mut accounts = lock.entry(account.address.clone()).or_insert_with(Vec::new);
// If the filename is the same we just need to replace the entry
accounts.retain(|acc| acc.filename != account.filename);
accounts.push(account.clone());
Ok(account)
}

View File

@ -220,6 +220,9 @@ impl EthMultiStore {
// update cache
let mut cache = self.cache.write();
let mut accounts = cache.entry(account.address.clone()).or_insert_with(Vec::new);
// TODO [ToDr] That is crappy way of overcoming set_name, set_meta, etc.
// Avoid cloning instead!
accounts.retain(|acc| acc.filename != account.filename);
accounts.push(account);
Ok(())
}

View File

@ -267,7 +267,7 @@ mod tests {
// when
let res = serde_json::to_string(&response);
let expected = r#"{"result":"0x0000000000000000000000000000000000000000","token":"test-token"}"#;
let expected = r#"{"result":"0x0000000000000000000000000000000000000000000000000000000000000000","token":"test-token"}"#;
// then
assert_eq!(res.unwrap(), expected.to_owned());