matrix_sdk_test/sync_builder/
knocked_room.rs

1use ruma::{
2    api::client::sync::sync_events::v3::KnockedRoom, events::AnyStrippedStateEvent, serde::Raw,
3    OwnedRoomId, RoomId,
4};
5
6use super::StrippedStateTestEvent;
7use crate::DEFAULT_TEST_ROOM_ID;
8
9pub struct KnockedRoomBuilder {
10    pub(super) room_id: OwnedRoomId,
11    pub(super) inner: KnockedRoom,
12}
13
14impl KnockedRoomBuilder {
15    /// Create a new `KnockedRoomBuilder` for the given room ID.
16    ///
17    /// If the room ID is [`DEFAULT_TEST_ROOM_ID`],
18    /// [`KnockedRoomBuilder::default()`] can be used instead.
19    pub fn new(room_id: &RoomId) -> Self {
20        Self { room_id: room_id.to_owned(), inner: Default::default() }
21    }
22
23    /// Get the room ID of this [`KnockedRoomBuilder`].
24    pub fn room_id(&self) -> &RoomId {
25        &self.room_id
26    }
27
28    /// Add an event to the state.
29    pub fn add_state_event(mut self, event: StrippedStateTestEvent) -> Self {
30        self.inner.knock_state.events.push(event.into_raw_event());
31        self
32    }
33
34    /// Add events to the state in bulk.
35    pub fn add_state_bulk<I>(mut self, events: I) -> Self
36    where
37        I: IntoIterator<Item = Raw<AnyStrippedStateEvent>>,
38    {
39        self.inner.knock_state.events.extend(events);
40        self
41    }
42}
43
44impl Default for KnockedRoomBuilder {
45    fn default() -> Self {
46        Self::new(&DEFAULT_TEST_ROOM_ID)
47    }
48}