Document and improve EventBuilder.

EventBuilder now clears itself between `build_sync_response` calls so
that each subsequent call will return an empty response if nothing was
added.

This allows reuse of a single EventBuilder instance which is important
for correct sync token rotation.
master
Denis Kasak 2020-06-30 17:09:10 +02:00
parent c57f076375
commit 84fc662614
1 changed files with 51 additions and 2 deletions

View File

@ -47,7 +47,41 @@ pub enum EventsJson {
Typing, Typing,
} }
/// Easily create events to stream into either a Client or a `Room` for testing. /// The `EventBuilder` struct can be used to easily generate valid sync responses for testing.
/// These can be then fed into either `Client` or `Room`.
///
/// It supports generated a number of canned events, such as a member entering a room, his power
/// level and display name changing and similar. It also supports insertion of custom events in the
/// form of `EventsJson` values.
///
/// **Important** You *must* use the *same* builder when sending multiple sync responses to
/// a single client. Otherwise, the subsequent responses will be *ignored* by the client because
/// the `next_batch` sync token will not be rotated properly.
///
/// # Example usage
///
/// ```rust
/// use matrix_sdk_test::{EventBuilder, EventsJson};
/// use matrix_sdk_common::events::collections::all::RoomEvent;
///
/// let mut builder = EventBuilder::new();
///
/// // response1 now contains events that add an example member to the room and change their power
/// // level
/// let response1 = builder
/// .add_room_event(EventsJson::Member, RoomEvent::RoomMember)
/// .add_room_event(EventsJson::PowerLevels, RoomEvent::RoomPowerLevels)
/// .build_sync_response();
///
/// // response2 is now empty (nothing changed)
/// let response2 = builder.build_sync_response();
///
/// // response3 contains a display name change for member example
/// let response3 = builder
/// .add_room_event(EventsJson::MemberNameChange, RoomEvent::RoomMember)
/// .build_sync_response();
/// ```
#[derive(Default)] #[derive(Default)]
pub struct EventBuilder { pub struct EventBuilder {
/// The events that determine the state of a `Room`. /// The events that determine the state of a `Room`.
@ -226,7 +260,8 @@ impl EventBuilder {
self self
} }
/// Consumes `ResponseBuilder` and returns `SyncResponse`. /// Builds a `SyncResponse` containing the events we queued so far. The next response returned
/// by `build_sync_response` will then be empty if no further events were queued.
pub fn build_sync_response(&mut self) -> SyncResponse { pub fn build_sync_response(&mut self) -> SyncResponse {
let main_room_id = RoomId::try_from("!SVkFJHzfwvuaIEawgC:localhost").unwrap(); let main_room_id = RoomId::try_from("!SVkFJHzfwvuaIEawgC:localhost").unwrap();
@ -338,12 +373,26 @@ impl EventBuilder {
let response = Response::builder() let response = Response::builder()
.body(serde_json::to_vec(&body).unwrap()) .body(serde_json::to_vec(&body).unwrap())
.unwrap(); .unwrap();
// Clear state so that the next sync response will be empty if nothing was added.
self.clear();
SyncResponse::try_from(response).unwrap() SyncResponse::try_from(response).unwrap()
} }
fn generate_sync_token(&self) -> String { fn generate_sync_token(&self) -> String {
format!("t392-516_47314_0_7_1_1_1_11444_{}", self.batch_counter) format!("t392-516_47314_0_7_1_1_1_11444_{}", self.batch_counter)
} }
pub fn clear(&mut self) {
self.account_data.clear();
self.ephemeral.clear();
self.invited_room_events.clear();
self.joined_room_events.clear();
self.left_room_events.clear();
self.presence_events.clear();
self.state_events.clear();
}
} }
/// Embedded sync reponse files /// Embedded sync reponse files