char
/
matrix-rust-sdk
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

crypto: Test the keys info for a device.

master
Damir Jelić 2020-04-15 18:55:15 +02:00
parent 14f25bf66a
commit bedb788432
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
src/crypto/device.rs
View File

@ -33,12 +33,6 @@ pub struct Device {
trust_state: Arc<Atomic<TrustState>>, trust_state: Arc<Atomic<TrustState>>,
} }
impl Device {
pub fn keys(&self, algorithm: &KeyAlgorithm) -> Option<&String> {
self.keys.get(algorithm)
}
}
#[derive(Debug, Clone, Copy)] #[derive(Debug, Clone, Copy)]
pub enum TrustState { pub enum TrustState {
Verified, Verified,
@ -55,6 +49,10 @@ impl Device {
pub fn user_id(&self) -> &UserId { pub fn user_id(&self) -> &UserId {
&self.user_id &self.user_id
} }
pub fn keys(&self, algorithm: &KeyAlgorithm) -> Option<&String> {
self.keys.get(algorithm)
}
} }
impl From<&DeviceKeys> for Device { impl From<&DeviceKeys> for Device {
@ -94,7 +92,7 @@ pub(crate) mod test {
use serde_json::json; use serde_json::json;
use std::convert::{From, TryFrom}; use std::convert::{From, TryFrom};
use crate::api::r0::keys::DeviceKeys; use crate::api::r0::keys::{DeviceKeys, KeyAlgorithm};
use crate::crypto::device::Device; use crate::crypto::device::Device;
use crate::identifiers::UserId; use crate::identifiers::UserId;
@ -138,5 +136,13 @@ pub(crate) mod test {
assert_eq!(&user_id, device.user_id()); assert_eq!(&user_id, device.user_id());
assert_eq!(device_id, device.device_id()); assert_eq!(device_id, device.device_id());
assert_eq!(device.algorithms.len(), 2); assert_eq!(device.algorithms.len(), 2);
assert_eq!(
device.keys(&KeyAlgorithm::Curve25519).unwrap(),
"wjLpTLRqbqBzLs63aYaEv2Boi6cFEbbM/sSRQ2oAKk4"
);
assert_eq!(
device.keys(&KeyAlgorithm::Ed25519).unwrap(),
"nE6W2fCblxDcOFmeEtCHNl8/l8bXcu7GKyAswA4r3mM"
);
} }
} }