Skip to content

Commit

Permalink
tests: rename RoomMessagesResponse to RoomMessagesResponseTemplate
Browse files Browse the repository at this point in the history
  • Loading branch information
bnjbvr committed Jan 13, 2025
1 parent e6dc109 commit 9514388
Show file tree
Hide file tree
Showing 5 changed files with 34 additions and 29 deletions.
4 changes: 2 additions & 2 deletions crates/matrix-sdk-ui/tests/integration/room_list_service.rs
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@ use matrix_sdk::{
config::RequestConfig,
test_utils::{
logged_in_client_with_server,
mocks::{MatrixMockServer, RoomMessagesResponse},
mocks::{MatrixMockServer, RoomMessagesResponseTemplate},
set_client_session, test_client_builder,
},
Client,
Expand Down Expand Up @@ -2846,7 +2846,7 @@ async fn test_multiple_timeline_init() {
// Send back-pagination responses with a small delay.
server
.mock_room_messages()
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("hello").into_raw_timeline()])
.delayed(Duration::from_millis(500)))
.mount()
Expand Down
4 changes: 2 additions & 2 deletions crates/matrix-sdk-ui/tests/integration/timeline/edit.rs
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ use matrix_sdk::{
room::edit::EditedContent,
test_utils::{
logged_in_client_with_server,
mocks::{MatrixMockServer, RoomMessagesResponse},
mocks::{MatrixMockServer, RoomMessagesResponseTemplate},
},
Client,
};
Expand Down Expand Up @@ -870,7 +870,7 @@ impl PendingEditHelper {
async fn handle_backpagination(&mut self, events: Vec<Raw<AnyTimelineEvent>>, batch_size: u16) {
self.server
.mock_room_messages()
.ok(RoomMessagesResponse::default().end_token("yolo").events(events))
.ok(RoomMessagesResponseTemplate::default().end_token("yolo").events(events))
.mock_once()
.mount()
.await;
Expand Down
8 changes: 4 additions & 4 deletions crates/matrix-sdk/src/test_utils/mocks.rs
Original file line number Diff line number Diff line change
Expand Up @@ -1895,7 +1895,7 @@ impl<'a> MockEndpoint<'a, RoomMessagesEndpoint> {
///
/// Note: pass `chunk` in the correct order: topological for forward
/// pagination, reverse topological for backwards pagination.
pub fn ok(self, response: RoomMessagesResponse) -> MatrixMock<'a> {
pub fn ok(self, response: RoomMessagesResponseTemplate) -> MatrixMock<'a> {
let mut template = ResponseTemplate::new(200).set_body_json(json!({
"start": response.start,
"end": response.end,
Expand All @@ -1913,7 +1913,7 @@ impl<'a> MockEndpoint<'a, RoomMessagesEndpoint> {
}

/// A response to a [`RoomMessagesEndpoint`] query.
pub struct RoomMessagesResponse {
pub struct RoomMessagesResponseTemplate {
/// The start token for this /messages query.
pub start: String,
/// The end token for this /messages query (previous batch for back
Expand All @@ -1927,7 +1927,7 @@ pub struct RoomMessagesResponse {
pub delay: Option<Duration>,
}

impl RoomMessagesResponse {
impl RoomMessagesResponseTemplate {
/// Fill the events returned as part of this response.
pub fn events(mut self, chunk: Vec<impl Into<Raw<AnyTimelineEvent>>>) -> Self {
self.chunk = chunk.into_iter().map(Into::into).collect();
Expand All @@ -1947,7 +1947,7 @@ impl RoomMessagesResponse {
}
}

impl Default for RoomMessagesResponse {
impl Default for RoomMessagesResponseTemplate {
fn default() -> Self {
Self {
start: "start-token-unused".to_owned(),
Expand Down
43 changes: 24 additions & 19 deletions crates/matrix-sdk/tests/integration/event_cache.rs
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,7 @@ use matrix_sdk::{
},
test_utils::{
assert_event_matches_msg,
mocks::{MatrixMockServer, RoomMessagesResponse},
mocks::{MatrixMockServer, RoomMessagesResponseTemplate},
},
};
use matrix_sdk_test::{
Expand Down Expand Up @@ -261,7 +261,7 @@ async fn test_backpaginate_once() {
server
.mock_room_messages()
.from("prev_batch")
.ok(RoomMessagesResponse::default().events(vec![
.ok(RoomMessagesResponseTemplate::default().events(vec![
f.text_msg("world").event_id(event_id!("$2")),
f.text_msg("hello").event_id(event_id!("$3")),
]))
Expand Down Expand Up @@ -349,7 +349,7 @@ async fn test_backpaginate_many_times_with_many_iterations() {
server
.mock_room_messages()
.from("prev_batch")
.ok(RoomMessagesResponse::default().end_token("prev_batch2").events(vec![
.ok(RoomMessagesResponseTemplate::default().end_token("prev_batch2").events(vec![
f.text_msg("world").event_id(event_id!("$2")),
f.text_msg("hello").event_id(event_id!("$3")),
]))
Expand All @@ -361,7 +361,7 @@ async fn test_backpaginate_many_times_with_many_iterations() {
server
.mock_room_messages()
.from("prev_batch2")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("oh well").event_id(event_id!("$4"))]))
.mock_once()
.mount()
Expand Down Expand Up @@ -487,7 +487,7 @@ async fn test_backpaginate_many_times_with_one_iteration() {
server
.mock_room_messages()
.from("prev_batch")
.ok(RoomMessagesResponse::default().end_token("prev_batch2").events(vec![
.ok(RoomMessagesResponseTemplate::default().end_token("prev_batch2").events(vec![
f.text_msg("world").event_id(event_id!("$2")),
f.text_msg("hello").event_id(event_id!("$3")),
]))
Expand All @@ -499,7 +499,7 @@ async fn test_backpaginate_many_times_with_one_iteration() {
server
.mock_room_messages()
.from("prev_batch2")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("oh well").event_id(event_id!("$4"))]))
.mock_once()
.mount()
Expand Down Expand Up @@ -641,7 +641,7 @@ async fn test_reset_while_backpaginating() {
server
.mock_room_messages()
.from("first_backpagination")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("lalala").into_raw_timeline()])
.delayed(Duration::from_millis(500)))
.mock_once()
Expand All @@ -653,7 +653,7 @@ async fn test_reset_while_backpaginating() {
server
.mock_room_messages()
.from("second_backpagination")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.end_token("third_backpagination")
.events(vec![f.text_msg("finally!").into_raw_timeline()]))
.mock_once()
Expand Down Expand Up @@ -764,7 +764,7 @@ async fn test_backpaginating_without_token() {

server
.mock_room_messages()
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("hi").event_id(event_id!("$2")).into_raw_timeline()]))
.mock_once()
.mount()
Expand Down Expand Up @@ -822,7 +822,7 @@ async fn test_limited_timeline_resets_pagination() {

server
.mock_room_messages()
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("hi").event_id(event_id!("$2")).into_raw_timeline()]))
.mock_once()
.mount()
Expand Down Expand Up @@ -992,7 +992,7 @@ async fn test_limited_timeline_without_storage() {
server
.mock_room_messages()
.from("prev-batch")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("oh well").event_id(event_id!("$1"))]))
.mock_once()
.mount()
Expand Down Expand Up @@ -1058,7 +1058,7 @@ async fn test_backpaginate_with_no_initial_events() {
let wait_time = Duration::from_millis(500);
server
.mock_room_messages()
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.end_token("prev_batch")
.events(vec![
f.text_msg("world").event_id(event_id!("$2")).into_raw_timeline(),
Expand All @@ -1073,7 +1073,7 @@ async fn test_backpaginate_with_no_initial_events() {
server
.mock_room_messages()
.from("prev_batch")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("oh well").event_id(event_id!("$1"))]))
.mock_once()
.mount()
Expand Down Expand Up @@ -1146,7 +1146,7 @@ async fn test_backpaginate_replace_empty_gap() {
// The first back-pagination will return a previous-batch token, but no events.
server
.mock_room_messages()
.ok(RoomMessagesResponse::default().end_token("prev_batch"))
.ok(RoomMessagesResponseTemplate::default().end_token("prev_batch"))
.mock_once()
.mount()
.await;
Expand All @@ -1155,7 +1155,7 @@ async fn test_backpaginate_replace_empty_gap() {
server
.mock_room_messages()
.from("prev_batch")
.ok(RoomMessagesResponse::default()
.ok(RoomMessagesResponseTemplate::default()
.events(vec![f.text_msg("hello").event_id(event_id!("$1"))]))
.mock_once()
.mount()
Expand Down Expand Up @@ -1218,7 +1218,12 @@ async fn test_no_gap_stored_after_deduplicated_sync() {
drop(events);

// Backpagination will return nothing.
server.mock_room_messages().ok(RoomMessagesResponse::default()).mock_once().mount().await;
server
.mock_room_messages()
.ok(RoomMessagesResponseTemplate::default())
.mock_once()
.mount()
.await;

let pagination = room_event_cache.pagination();

Expand Down Expand Up @@ -1328,7 +1333,7 @@ async fn test_no_gap_stored_after_deduplicated_backpagination() {
server
.mock_room_messages()
.from("prev-batch2")
.ok(RoomMessagesResponse::default())
.ok(RoomMessagesResponseTemplate::default())
.mock_once()
.mount()
.await;
Expand All @@ -1338,7 +1343,7 @@ async fn test_no_gap_stored_after_deduplicated_backpagination() {
server
.mock_room_messages()
.from("prev-batch")
.ok(RoomMessagesResponse::default().end_token("prev-batch3").events(vec![
.ok(RoomMessagesResponseTemplate::default().end_token("prev-batch3").events(vec![
// Items in reverse order, since this is back-pagination.
f.text_msg("world").event_id(event_id!("$2")).into_raw_timeline(),
f.text_msg("hello").event_id(event_id!("$1")).into_raw_timeline(),
Expand Down Expand Up @@ -1424,7 +1429,7 @@ async fn test_dont_delete_gap_that_wasnt_inserted() {
server
.mock_room_messages()
.from("prev-batch")
.ok(RoomMessagesResponse::default())
.ok(RoomMessagesResponseTemplate::default())
.mock_once()
.mount()
.await;
Expand Down
4 changes: 2 additions & 2 deletions crates/matrix-sdk/tests/integration/widget.rs
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ use assert_matches::assert_matches;
use async_trait::async_trait;
use futures_util::FutureExt;
use matrix_sdk::{
test_utils::mocks::{MatrixMockServer, RoomMessagesResponse},
test_utils::mocks::{MatrixMockServer, RoomMessagesResponseTemplate},
widget::{
Capabilities, CapabilitiesProvider, WidgetDriver, WidgetDriverHandle, WidgetSettings,
},
Expand Down Expand Up @@ -313,7 +313,7 @@ async fn test_read_messages_with_msgtype_capabilities() {
mock_server
.mock_room_messages()
.limit(3)
.ok(RoomMessagesResponse::default().end_token(end).events(chunk2))
.ok(RoomMessagesResponseTemplate::default().end_token(end).events(chunk2))
.mock_once()
.mount()
.await;
Expand Down

0 comments on commit 9514388

Please sign in to comment.