87 lines
2.4 KiB
Rust
87 lines
2.4 KiB
Rust
use super::{FakeTdClient, TdUpdate};
|
|
use crate::tdlib::types::FolderInfo;
|
|
use crate::tdlib::{AuthState, ChatInfo, MessageInfo, NetworkState, ProfileInfo};
|
|
use tokio::sync::mpsc;
|
|
|
|
#[allow(dead_code)]
|
|
impl FakeTdClient {
|
|
/// Create an update channel for receiving simulated TDLib events.
|
|
pub fn with_update_channel(self) -> (Self, mpsc::UnboundedReceiver<TdUpdate>) {
|
|
let (tx, rx) = mpsc::unbounded_channel();
|
|
*self.update_tx.lock().unwrap() = Some(tx);
|
|
(self, rx)
|
|
}
|
|
|
|
/// Enable simulated delays, closer to real TDLib behavior.
|
|
pub fn with_delays(mut self) -> Self {
|
|
self.simulate_delays = true;
|
|
self
|
|
}
|
|
|
|
pub fn with_chat(self, chat: ChatInfo) -> Self {
|
|
self.chats.lock().unwrap().push(chat);
|
|
self
|
|
}
|
|
|
|
pub fn with_chats(self, chats: Vec<ChatInfo>) -> Self {
|
|
self.chats.lock().unwrap().extend(chats);
|
|
self
|
|
}
|
|
|
|
pub fn with_message(self, chat_id: i64, message: MessageInfo) -> Self {
|
|
self.messages
|
|
.lock()
|
|
.unwrap()
|
|
.entry(chat_id)
|
|
.or_default()
|
|
.push(message);
|
|
self
|
|
}
|
|
|
|
pub fn with_messages(self, chat_id: i64, messages: Vec<MessageInfo>) -> Self {
|
|
self.messages.lock().unwrap().insert(chat_id, messages);
|
|
self
|
|
}
|
|
|
|
pub fn with_folder(self, id: i32, name: &str) -> Self {
|
|
self.folders
|
|
.lock()
|
|
.unwrap()
|
|
.push(FolderInfo { id, name: name.to_string() });
|
|
self
|
|
}
|
|
|
|
pub fn with_user(self, id: i64, name: &str) -> Self {
|
|
self.user_names.lock().unwrap().insert(id, name.to_string());
|
|
self
|
|
}
|
|
|
|
pub fn with_profile(self, chat_id: i64, profile: ProfileInfo) -> Self {
|
|
self.profiles.lock().unwrap().insert(chat_id, profile);
|
|
self
|
|
}
|
|
|
|
pub fn with_network_state(self, state: NetworkState) -> Self {
|
|
*self.network_state.lock().unwrap() = state;
|
|
self
|
|
}
|
|
|
|
pub fn with_auth_state(self, state: AuthState) -> Self {
|
|
*self.auth_state.lock().unwrap() = state;
|
|
self
|
|
}
|
|
|
|
pub fn with_downloaded_file(self, file_id: i32, path: &str) -> Self {
|
|
self.downloaded_files
|
|
.lock()
|
|
.unwrap()
|
|
.insert(file_id, path.to_string());
|
|
self
|
|
}
|
|
|
|
pub fn with_available_reactions(self, reactions: Vec<String>) -> Self {
|
|
*self.available_reactions.lock().unwrap() = reactions;
|
|
self
|
|
}
|
|
}
|