Prem Pillai
72 lines
2.1 KiB
Rust
72 lines
2.1 KiB
Rust
use super::{KeyringBackend, KeyringError};
|
|
use anyhow::Result;
|
|
use std::collections::HashMap;
|
|
use std::sync::{Arc, RwLock};
|
|
|
|
#[derive(Clone, Default)]
|
|
pub struct MockKeyringBackend {
|
|
storage: Arc<RwLock<HashMap<String, String>>>,
|
|
}
|
|
|
|
impl MockKeyringBackend {
|
|
pub fn new() -> Self {
|
|
Self::default()
|
|
}
|
|
|
|
pub fn clear(&self) {
|
|
self.storage
|
|
.write()
|
|
.expect("Mock keyring lock poisoned")
|
|
.clear();
|
|
}
|
|
|
|
pub fn contains(&self, service: &str, username: &str) -> bool {
|
|
let key = format!("{}:{}", service, username);
|
|
self.storage
|
|
.read()
|
|
.expect("Mock keyring lock poisoned")
|
|
.contains_key(&key)
|
|
}
|
|
|
|
fn make_key(service: &str, username: &str) -> String {
|
|
format!("{}:{}", service, username)
|
|
}
|
|
}
|
|
|
|
impl KeyringBackend for MockKeyringBackend {
|
|
fn get_password(&self, service: &str, username: &str) -> Result<String> {
|
|
let key = Self::make_key(service, username);
|
|
let storage = self
|
|
.storage
|
|
.read()
|
|
.map_err(|e| KeyringError::Backend(format!("Mock keyring lock poisoned: {}", e)))?;
|
|
|
|
storage
|
|
.get(&key)
|
|
.cloned()
|
|
.ok_or_else(|| KeyringError::NotFound {
|
|
service: service.to_string(),
|
|
username: username.to_string(),
|
|
})
|
|
.map_err(anyhow::Error::from)
|
|
}
|
|
|
|
fn set_password(&self, service: &str, username: &str, password: &str) -> Result<()> {
|
|
let key = Self::make_key(service, username);
|
|
self.storage
|
|
.write()
|
|
.map_err(|e| KeyringError::Backend(format!("Mock keyring lock poisoned: {}", e)))?
|
|
.insert(key, password.to_string());
|
|
Ok(())
|
|
}
|
|
|
|
fn delete_password(&self, service: &str, username: &str) -> Result<()> {
|
|
let key = Self::make_key(service, username);
|
|
self.storage
|
|
.write()
|
|
.map_err(|e| KeyringError::Backend(format!("Mock keyring lock poisoned: {}", e)))?
|
|
.remove(&key);
|
|
Ok(())
|
|
}
|
|
}
|