matrix-rust-sdk/matrix_sdk_test/src/lib.rs

use std::collections::HashMap;
use std::convert::TryFrom;
use std::panic;

use http::Response;

use matrix_sdk_common::api::r0::sync::sync_events::Response as SyncResponse;
use matrix_sdk_common::events::{
    collections::{
        all::{RoomEvent, StateEvent},
        only::Event,
    },
    presence::PresenceEvent,
    stripped::AnyStrippedStateEvent,
    EventJson, TryFromRaw,
};
use matrix_sdk_common::identifiers::RoomId;

pub use matrix_sdk_test_macros::async_test;

/// Embedded event files
#[derive(Debug)]
pub enum EventsFile {
    Alias,
    Aliases,
    Create,
    FullyRead,
    HistoryVisibility,
    JoinRules,
    Member,
    MessageEmote,
    MessageNotice,
    MessageText,
    Name,
    PowerLevels,
    Presence,
    RedactedInvalid,
    RedactedState,
    Redacted,
    Redaction,
    RoomAvatar,
    Tag,
    Topic,
    Typing,
}

/// Easily create events to stream into either a Client or a `Room` for testing.
#[derive(Default)]
pub struct EventBuilder {
    /// The events that determine the state of a `Room`.
    joined_room_events: HashMap<RoomId, Vec<RoomEvent>>,
    /// The events that determine the state of a `Room`.
    invited_room_events: HashMap<RoomId, Vec<AnyStrippedStateEvent>>,
    /// The events that determine the state of a `Room`.
    left_room_events: HashMap<RoomId, Vec<RoomEvent>>,
    /// The presence events that determine the presence state of a `RoomMember`.
    presence_events: Vec<PresenceEvent>,
    /// The state events that determine the state of a `Room`.
    state_events: Vec<StateEvent>,
    /// The ephemeral room events that determine the state of a `Room`.
    ephemeral: Vec<Event>,
    /// The account data events that determine the state of a `Room`.
    account_data: Vec<Event>,
}

impl EventBuilder {
    /// Add an event to the room events `Vec`.
    pub fn add_ephemeral<Ev: TryFromRaw>(
        mut self,
        file: EventsFile,
        variant: fn(Ev) -> Event,
    ) -> Self {
        let val: &str = match file {
            EventsFile::Typing => include_str!("../test_data/events/typing.json"),
            _ => panic!("unknown ephemeral event file {:?}", file),
        };

        let event = serde_json::from_str::<EventJson<Ev>>(&val)
            .unwrap()
            .deserialize()
            .unwrap();
        self.ephemeral.push(variant(event));
        self
    }

    /// Add an event to the room events `Vec`.
    #[allow(clippy::match_single_binding, unused)]
    pub fn add_account<Ev: TryFromRaw>(
        mut self,
        file: EventsFile,
        variant: fn(Ev) -> Event,
    ) -> Self {
        let val: &str = match file {
            _ => panic!("unknown account event file {:?}", file),
        };

        let event = serde_json::from_str::<EventJson<Ev>>(&val)
            .unwrap()
            .deserialize()
            .unwrap();
        self.account_data.push(variant(event));
        self
    }

    /// Add an event to the room events `Vec`.
    pub fn add_room_event<Ev: TryFromRaw>(
        mut self,
        file: EventsFile,
        variant: fn(Ev) -> RoomEvent,
    ) -> Self {
        let val = match file {
            EventsFile::Member => include_str!("../test_data/events/member.json"),
            EventsFile::PowerLevels => include_str!("../test_data/events/power_levels.json"),
            _ => panic!("unknown room event file {:?}", file),
        };

        let event = serde_json::from_str::<EventJson<Ev>>(&val)
            .unwrap()
            .deserialize()
            .unwrap();
        self.add_joined_event(
            &RoomId::try_from("!SVkFJHzfwvuaIEawgC:localhost").unwrap(),
            variant(event),
        );
        self
    }

    pub fn add_custom_joined_event<Ev: TryFromRaw>(
        mut self,
        room_id: &RoomId,
        event: serde_json::Value,
        variant: fn(Ev) -> RoomEvent,
    ) -> Self {
        let event = serde_json::from_value::<EventJson<Ev>>(event)
            .unwrap()
            .deserialize()
            .unwrap();
        self.add_joined_event(room_id, variant(event));
        self
    }

    fn add_joined_event(&mut self, room_id: &RoomId, event: RoomEvent) {
        self.joined_room_events
            .entry(room_id.clone())
            .or_insert_with(Vec::new)
            .push(event);
    }

    pub fn add_custom_invited_event<Ev: TryFromRaw>(
        mut self,
        room_id: &RoomId,
        event: serde_json::Value,
        variant: fn(Ev) -> AnyStrippedStateEvent,
    ) -> Self {
        let event = serde_json::from_value::<EventJson<Ev>>(event)
            .unwrap()
            .deserialize()
            .unwrap();
        self.invited_room_events
            .entry(room_id.clone())
            .or_insert_with(Vec::new)
            .push(variant(event));
        self
    }

    pub fn add_custom_left_event<Ev: TryFromRaw>(
        mut self,
        room_id: &RoomId,
        event: serde_json::Value,
        variant: fn(Ev) -> RoomEvent,
    ) -> Self {
        let event = serde_json::from_value::<EventJson<Ev>>(event)
            .unwrap()
            .deserialize()
            .unwrap();
        self.left_room_events
            .entry(room_id.clone())
            .or_insert_with(Vec::new)
            .push(variant(event));
        self
    }

    /// Add a state event to the state events `Vec`.
    pub fn add_state_event<Ev: TryFromRaw>(
        mut self,
        file: EventsFile,
        variant: fn(Ev) -> StateEvent,
    ) -> Self {
        let val = match file {
            EventsFile::Alias => include_str!("../test_data/events/alias.json"),
            EventsFile::Aliases => include_str!("../test_data/events/aliases.json"),
            EventsFile::Name => include_str!("../test_data/events/name.json"),
            _ => panic!("unknown state event file {:?}", file),
        };

        let event = serde_json::from_str::<EventJson<Ev>>(&val)
            .unwrap()
            .deserialize()
            .unwrap();
        self.state_events.push(variant(event));
        self
    }

    /// Add an presence event to the presence events `Vec`.
    pub fn add_presence_event(mut self, file: EventsFile) -> Self {
        let val = match file {
            EventsFile::Presence => include_str!("../test_data/events/presence.json"),
            _ => panic!("unknown presence event file {:?}", file),
        };

        let event = serde_json::from_str::<EventJson<PresenceEvent>>(&val)
            .unwrap()
            .deserialize()
            .unwrap();
        self.presence_events.push(event);
        self
    }

    /// Consumes `ResponseBuilder and returns SyncResponse.
    pub fn build_sync_response(mut self) -> SyncResponse {
        let main_room_id = RoomId::try_from("!SVkFJHzfwvuaIEawgC:localhost").unwrap();

        // TODO generalize this.
        let joined_room = serde_json::json!({
            "summary": {},
            "account_data": {
                "events": self.account_data
            },
            "ephemeral": {
                "events": self.ephemeral
            },
            "state": {
                "events": self.state_events
            },
            "timeline": {
                "events": self.joined_room_events.remove(&main_room_id).unwrap_or_default(),
                "limited": true,
                "prev_batch": "t392-516_47314_0_7_1_1_1_11444_1"
            },
            "unread_notifications": {
                "highlight_count": 0,
                "notification_count": 11
            }
        });

        let mut joined_rooms: HashMap<RoomId, serde_json::Value> = HashMap::new();

        joined_rooms.insert(main_room_id, joined_room);

        for (room_id, events) in self.joined_room_events.drain() {
            let joined_room = serde_json::json!({
                "summary": {},
                "account_data": {
                    "events": [],
                },
                "ephemeral": {
                    "events": [],
                },
                "state": {
                    "events": [],
                },
                "timeline": {
                    "events": events,
                    "limited": true,
                    "prev_batch": "t392-516_47314_0_7_1_1_1_11444_1"
                },
                "unread_notifications": {
                    "highlight_count": 0,
                    "notification_count": 11
                }
            });
            joined_rooms.insert(room_id, joined_room);
        }

        let mut left_rooms: HashMap<RoomId, serde_json::Value> = HashMap::new();

        for (room_id, events) in self.left_room_events.drain() {
            let room = serde_json::json!({
                "state": {
                    "events": [],
                },
                "timeline": {
                    "events": events,
                    "limited": false,
                    "prev_batch": "t392-516_47314_0_7_1_1_1_11444_1"
                },
            });
            left_rooms.insert(room_id, room);
        }

        let mut invited_rooms: HashMap<RoomId, serde_json::Value> = HashMap::new();

        for (room_id, events) in self.invited_room_events.drain() {
            let room = serde_json::json!({
                "invite_state": {
                    "events": events,
                },
            });
            invited_rooms.insert(room_id, room);
        }

        let body = serde_json::json! {
            {
                "device_one_time_keys_count": {},
                "next_batch": "s526_47314_0_7_1_1_1_11444_1",
                "device_lists": {
                    "changed": [],
                    "left": []
                },
                "rooms": {
                    "invite": invited_rooms,
                    "join": joined_rooms,
                    "leave": left_rooms,
                },
                "to_device": {
                    "events": []
                },
                "presence": {
                    "events": []
                }
            }
        };
        let response = Response::builder()
            .body(serde_json::to_vec(&body).unwrap())
            .unwrap();
        SyncResponse::try_from(response).unwrap()
    }
}

/// Embedded sync reponse files
pub enum SyncResponseFile {
    Default,
    DefaultWithSummary,
    Invite,
    Leave,
}

/// Get specific API responses for testing
pub fn sync_response(kind: SyncResponseFile) -> SyncResponse {
    let data = match kind {
        SyncResponseFile::Default => include_bytes!("../test_data/sync.json").to_vec(),
        SyncResponseFile::DefaultWithSummary => {
            include_bytes!("../test_data/sync_with_summary.json").to_vec()
        }
        SyncResponseFile::Invite => include_bytes!("../test_data/invite_sync.json").to_vec(),
        SyncResponseFile::Leave => include_bytes!("../test_data/leave_sync.json").to_vec(),
    };

    let response = Response::builder().body(data.to_vec()).unwrap();
    SyncResponse::try_from(response).unwrap()
}