base: Remove a bunch of stuff and add sled.
This commit is contained in:
parent
92bedb4571
commit
cd3d90df3f
5 changed files with 39 additions and 2159 deletions
|
@ -36,6 +36,7 @@ matrix-sdk-crypto = { version = "0.1.0", path = "../matrix_sdk_crypto", optional
|
|||
|
||||
# Misc dependencies
|
||||
thiserror = "1.0.21"
|
||||
sled = "*"
|
||||
|
||||
[target.'cfg(not(target_arch = "wasm32"))'.dependencies.tokio]
|
||||
version = "0.2.22"
|
||||
|
|
File diff suppressed because it is too large
Load diff
|
@ -28,7 +28,6 @@
|
|||
//! of Synapse in compliance with the Matrix API specification.
|
||||
#![deny(
|
||||
missing_debug_implementations,
|
||||
dead_code,
|
||||
missing_docs,
|
||||
trivial_casts,
|
||||
trivial_numeric_casts,
|
||||
|
@ -46,24 +45,10 @@ pub use matrix_sdk_common::*;
|
|||
|
||||
mod client;
|
||||
mod error;
|
||||
mod event_emitter;
|
||||
mod models;
|
||||
mod session;
|
||||
mod state;
|
||||
|
||||
pub use client::{BaseClient, BaseClientConfig, RoomState, RoomStateType};
|
||||
pub use event_emitter::{CustomEvent, EventEmitter, SyncRoom};
|
||||
pub use models::{Room, RoomMember};
|
||||
pub use state::{AllRooms, ClientState};
|
||||
|
||||
#[cfg(feature = "encryption")]
|
||||
#[cfg_attr(feature = "docs", doc(cfg(encryption)))]
|
||||
pub use matrix_sdk_crypto as crypto;
|
||||
|
||||
#[cfg(feature = "messages")]
|
||||
#[cfg_attr(feature = "docs", doc(cfg(messages)))]
|
||||
pub use models::{MessageQueue, PossiblyRedactedExt};
|
||||
|
||||
#[cfg(not(target_arch = "wasm32"))]
|
||||
pub use state::JsonStore;
|
||||
pub use state::StateStore;
|
||||
|
|
|
@ -214,190 +214,4 @@ impl StateStore for JsonStore {
|
|||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod test {
|
||||
use super::*;
|
||||
|
||||
use std::path::PathBuf;
|
||||
|
||||
use tempfile::tempdir;
|
||||
|
||||
use crate::{
|
||||
identifiers::{room_id, user_id},
|
||||
push::Ruleset,
|
||||
BaseClient, BaseClientConfig, Session,
|
||||
};
|
||||
|
||||
use matrix_sdk_test::{sync_response, SyncResponseFile};
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_store_client_state() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
|
||||
let user = user_id!("@example:example.com");
|
||||
|
||||
let sess = Session {
|
||||
access_token: "32nj9zu034btz90".to_string(),
|
||||
user_id: user.clone(),
|
||||
device_id: "Tester".into(),
|
||||
};
|
||||
|
||||
let state = ClientState {
|
||||
sync_token: Some("hello".into()),
|
||||
ignored_users: vec![user],
|
||||
push_ruleset: None::<Ruleset>,
|
||||
};
|
||||
|
||||
let mut path_with_user = PathBuf::from(path);
|
||||
path_with_user.push(sess.user_id.localpart());
|
||||
// we have to set the path since `JsonStore::store_client_state()` doesn't append to the path
|
||||
let store = JsonStore::open(path_with_user).unwrap();
|
||||
store.store_client_state(state.clone()).await.unwrap();
|
||||
|
||||
// the newly loaded store sets it own user_id local part when `load_client_state`
|
||||
let store = JsonStore::open(path).unwrap();
|
||||
let loaded = store.load_client_state(&sess).await.unwrap();
|
||||
assert_eq!(loaded, Some(state));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_store_load_joined_room_state() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
let store = JsonStore::open(path).unwrap();
|
||||
|
||||
let id = room_id!("!roomid:example.com");
|
||||
let user = user_id!("@example:example.com");
|
||||
|
||||
let room = Room::new(&id, &user);
|
||||
store
|
||||
.store_room_state(RoomState::Joined(&room))
|
||||
.await
|
||||
.unwrap();
|
||||
let AllRooms { joined, .. } = store.load_all_rooms().await.unwrap();
|
||||
assert_eq!(joined.get(&id), Some(&Room::new(&id, &user)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_store_load_left_room_state() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
let store = JsonStore::open(path).unwrap();
|
||||
|
||||
let id = room_id!("!roomid:example.com");
|
||||
let user = user_id!("@example:example.com");
|
||||
|
||||
let room = Room::new(&id, &user);
|
||||
store
|
||||
.store_room_state(RoomState::Left(&room))
|
||||
.await
|
||||
.unwrap();
|
||||
let AllRooms { left, .. } = store.load_all_rooms().await.unwrap();
|
||||
assert_eq!(left.get(&id), Some(&Room::new(&id, &user)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_store_load_invited_room_state() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
let store = JsonStore::open(path).unwrap();
|
||||
|
||||
let id = room_id!("!roomid:example.com");
|
||||
let user = user_id!("@example:example.com");
|
||||
|
||||
let room = Room::new(&id, &user);
|
||||
store
|
||||
.store_room_state(RoomState::Invited(&room))
|
||||
.await
|
||||
.unwrap();
|
||||
let AllRooms { invited, .. } = store.load_all_rooms().await.unwrap();
|
||||
assert_eq!(invited.get(&id), Some(&Room::new(&id, &user)));
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_store_load_join_leave_room_state() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
let store = JsonStore::open(path).unwrap();
|
||||
|
||||
let id = room_id!("!roomid:example.com");
|
||||
let user = user_id!("@example:example.com");
|
||||
|
||||
let room = Room::new(&id, &user);
|
||||
store
|
||||
.store_room_state(RoomState::Joined(&room))
|
||||
.await
|
||||
.unwrap();
|
||||
assert!(store
|
||||
.delete_room_state(RoomState::Joined(&id))
|
||||
.await
|
||||
.is_ok());
|
||||
let AllRooms { joined, .. } = store.load_all_rooms().await.unwrap();
|
||||
|
||||
// test that we have removed the correct room
|
||||
assert!(joined.is_empty());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_store_load_invite_join_room_state() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
let store = JsonStore::open(path).unwrap();
|
||||
|
||||
let id = room_id!("!roomid:example.com");
|
||||
let user = user_id!("@example:example.com");
|
||||
|
||||
let room = Room::new(&id, &user);
|
||||
store
|
||||
.store_room_state(RoomState::Invited(&room))
|
||||
.await
|
||||
.unwrap();
|
||||
assert!(store
|
||||
.delete_room_state(RoomState::Invited(&id))
|
||||
.await
|
||||
.is_ok());
|
||||
let AllRooms { invited, .. } = store.load_all_rooms().await.unwrap();
|
||||
// test that we have removed the correct room
|
||||
assert!(invited.is_empty());
|
||||
}
|
||||
|
||||
#[tokio::test]
|
||||
async fn test_client_sync_store() {
|
||||
let dir = tempdir().unwrap();
|
||||
let path: &Path = dir.path();
|
||||
|
||||
let session = Session {
|
||||
access_token: "1234".to_owned(),
|
||||
user_id: user_id!("@cheeky_monkey:matrix.org"),
|
||||
device_id: "DEVICEID".into(),
|
||||
};
|
||||
|
||||
// a sync response to populate our JSON store
|
||||
let store = Box::new(JsonStore::open(path).unwrap());
|
||||
let client =
|
||||
BaseClient::new_with_config(BaseClientConfig::new().state_store(store)).unwrap();
|
||||
client.restore_login(session.clone()).await.unwrap();
|
||||
|
||||
let mut response = sync_response(SyncResponseFile::Default);
|
||||
|
||||
// gather state to save to the db, the first time through loading will be skipped
|
||||
client.receive_sync_response(&mut response).await.unwrap();
|
||||
|
||||
// now syncing the client will update from the state store
|
||||
let store = Box::new(JsonStore::open(path).unwrap());
|
||||
let client =
|
||||
BaseClient::new_with_config(BaseClientConfig::new().state_store(store)).unwrap();
|
||||
client.restore_login(session.clone()).await.unwrap();
|
||||
|
||||
// assert the synced client and the logged in client are equal
|
||||
assert_eq!(*client.session().read().await, Some(session));
|
||||
assert_eq!(
|
||||
client.sync_token().await,
|
||||
Some("s526_47314_0_7_1_1_1_11444_1".to_string())
|
||||
);
|
||||
assert_eq!(
|
||||
*client.ignored_users.read().await,
|
||||
vec![user_id!("@someone:example.org")]
|
||||
);
|
||||
}
|
||||
}
|
||||
mod test {}
|
||||
|
|
|
@ -56,27 +56,6 @@ impl PartialEq for ClientState {
|
|||
}
|
||||
}
|
||||
|
||||
impl ClientState {
|
||||
/// Create a JSON serialize-able `ClientState`.
|
||||
///
|
||||
/// This enables non sensitive information to be saved by `JsonStore`.
|
||||
#[allow(clippy::eval_order_dependence)]
|
||||
// TODO is this ok ^^^?? https://github.com/rust-lang/rust-clippy/issues/4637
|
||||
pub async fn from_base_client(client: &BaseClient) -> ClientState {
|
||||
let BaseClient {
|
||||
sync_token,
|
||||
ignored_users,
|
||||
push_ruleset,
|
||||
..
|
||||
} = client;
|
||||
Self {
|
||||
sync_token: sync_token.read().await.clone(),
|
||||
ignored_users: ignored_users.read().await.clone(),
|
||||
push_ruleset: push_ruleset.read().await.clone(),
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/// `JsonStore::load_all_rooms` returns `AllRooms`.
|
||||
///
|
||||
/// `AllRooms` is made of the `joined`, `invited` and `left` room maps.
|
||||
|
|
Loading…
Reference in a new issue