Skip to content

Commit 41ce2bb

Browse files
committed
test(sdk,ui): Update tests to the new API.
1 parent 88f2cba commit 41ce2bb

2 files changed

Lines changed: 43 additions & 34 deletions

File tree

crates/matrix-sdk-ui/tests/integration/timeline/pinned_event.rs

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -771,10 +771,10 @@ async fn test_redacted_events_are_reflected_in_sync() {
771771
let room_id = room_id!("!test:localhost");
772772

773773
let f = EventFactory::new().room(room_id).sender(*BOB);
774-
let event_id = event_id!("$1");
774+
let pinned_event_id = event_id!("$1");
775775
let pinned_event = f
776776
.text_msg("in the end")
777-
.event_id(event_id!("$1"))
777+
.event_id(pinned_event_id)
778778
.server_ts(MilliSecondsSinceUnixEpoch::now())
779779
.into_raw_sync();
780780

@@ -784,7 +784,7 @@ async fn test_redacted_events_are_reflected_in_sync() {
784784
// Mock /relations for pinned timeline event.
785785
server
786786
.mock_room_relations()
787-
.match_target_event(event_id.to_owned())
787+
.match_target_event(pinned_event_id.to_owned())
788788
.match_limit(256)
789789
.ok(RoomRelationsResponseTemplate::default().events(Vec::<Raw<AnyTimelineEvent>>::new()))
790790
.mock_once()
@@ -794,7 +794,7 @@ async fn test_redacted_events_are_reflected_in_sync() {
794794
// Load initial timeline items: a text message and a `m.room.pinned_events` with
795795
// event $1
796796
let room = PinnedEventsSync::new(room_id)
797-
.with_pinned_event_ids(vec!["$1"])
797+
.with_pinned_event_ids(vec![pinned_event_id.as_str()])
798798
.mock_and_sync(&client, &server)
799799
.await
800800
.expect("Sync failed");
@@ -818,12 +818,13 @@ async fn test_redacted_events_are_reflected_in_sync() {
818818

819819
assert_eq!(items.len(), 1 + 1); // event item + a date divider
820820
assert!(items[0].is_date_divider());
821-
assert_eq!(items[1].as_event().unwrap().content().as_message().unwrap().body(), "in the end");
821+
assert_eq!(items[1].as_event().unwrap().event_id(), Some(pinned_event_id));
822822
assert_pending!(timeline_stream);
823823

824+
let redaction_event_id = event_id!("$2");
824825
let redaction_event = f
825-
.redaction(event_id!("$1"))
826-
.event_id(event_id!("$2"))
826+
.redaction(pinned_event_id)
827+
.event_id(redaction_event_id)
827828
.server_ts(MilliSecondsSinceUnixEpoch::now())
828829
.into_raw_sync();
829830

crates/matrix-sdk/tests/integration/room/pinned_events.rs

Lines changed: 35 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -171,8 +171,10 @@ async fn test_pinned_events_are_loaded_from_network_then_are_reloaded_from_stora
171171
.build()
172172
.await;
173173

174+
let event_cache = client.event_cache();
175+
174176
// Subscribe the event cache to sync updates.
175-
client.event_cache().subscribe().unwrap();
177+
event_cache.subscribe().unwrap();
176178

177179
// Sync the room with the pinned event ID in the room state.
178180
//
@@ -181,19 +183,20 @@ async fn test_pinned_events_are_loaded_from_network_then_are_reloaded_from_stora
181183
// events.
182184
let pinned_events_state = f.room_pinned_events(vec![pinned_event_id.to_owned()]);
183185

184-
let room = server
186+
let _room = server
185187
.sync_room(
186188
&client,
187189
JoinedRoomBuilder::new(room_id).add_state_bulk(vec![pinned_events_state.into()]),
188190
)
189191
.await;
190192

191-
// Get the room event cache and subscribe to pinned events.
192-
let (room_event_cache, _drop_handles) = room.event_cache().await.unwrap();
193+
// Get the pinned events cache and subscribe to it.
194+
let (pinned_events_cache, _drop_handles) =
195+
event_cache.pinned_events(room_id).await.unwrap();
193196

194-
// Subscribe to pinned events - this triggers PinnedEventsCache::new() which
195-
// spawns a task that calls reload_from_storage() first.
196-
let (events, mut subscriber) = room_event_cache.subscribe_to_pinned_events().await.unwrap();
197+
// Getting the pinned events cache triggers `PinnedEventsCache::new()` which
198+
// spawns a task that calls `reload_from_storage()` first.
199+
let (events, mut subscriber) = pinned_events_cache.subscribe().await.unwrap();
197200
let mut events = events.into();
198201

199202
// Wait for the background task to reload the events.
@@ -229,17 +232,19 @@ async fn test_pinned_events_are_loaded_from_network_then_are_reloaded_from_stora
229232
.build()
230233
.await;
231234

235+
let event_cache = client.event_cache();
236+
232237
// Subscribe the event cache to sync updates.
233-
client.event_cache().subscribe().unwrap();
238+
event_cache.subscribe().unwrap();
234239

235-
let room = client.get_room(room_id).unwrap();
240+
let _room = client.get_room(room_id).unwrap();
236241

237-
// Get the room event cache and subscribe to pinned events.
238-
let (room_event_cache, _drop_handles) = room.event_cache().await.unwrap();
242+
// Get the pinned events cache and subscribe to it.
243+
let (pinned_events_cache, _drop_handles) = event_cache.pinned_events(room_id).await.unwrap();
239244

240-
// Subscribe to pinned events - this triggers PinnedEventsCache::new() which
241-
// spawns a task that calls reload_from_storage() first.
242-
let (events, mut subscriber) = room_event_cache.subscribe_to_pinned_events().await.unwrap();
245+
// Getting the pinned events cache triggers `PinnedEventsCache::new()` which
246+
// spawns a task that calls `reload_from_storage()` first.
247+
let (events, mut subscriber) = pinned_events_cache.subscribe().await.unwrap();
243248
let mut events = events.into();
244249

245250
// Wait for the background task to reload the events from storage.
@@ -277,8 +282,10 @@ async fn test_pinned_events_are_reloaded_from_storage_from_many_chunks() {
277282
let server = MatrixMockServer::new().await;
278283
let client = server.client_builder().build().await;
279284

285+
let event_cache = client.event_cache();
286+
280287
// Create a non empty Event Cache store containing two chunks.
281-
client.event_cache().subscribe().unwrap();
288+
event_cache.subscribe().unwrap();
282289
client
283290
.event_cache_store()
284291
.lock()
@@ -308,14 +315,14 @@ async fn test_pinned_events_are_reloaded_from_storage_from_many_chunks() {
308315
.await
309316
.unwrap();
310317

311-
let room = server.sync_room(&client, JoinedRoomBuilder::new(room_id)).await;
318+
let _room = server.sync_room(&client, JoinedRoomBuilder::new(room_id)).await;
312319

313-
// Get the room event cache and subscribe to pinned events.
314-
let (room_event_cache, _drop_handles) = room.event_cache().await.unwrap();
320+
// Get the pinned events cache and subscribe to it.
321+
let (pinned_events_cache, _drop_handles) = event_cache.pinned_events(room_id).await.unwrap();
315322

316-
// Subscribe to pinned events - this triggers PinnedEventsCache::new() which
317-
// spawns a task that calls reload_from_storage() first.
318-
let (events, mut subscriber) = room_event_cache.subscribe_to_pinned_events().await.unwrap();
323+
// Getting the pinned events cache triggers `PinnedEventsCache::new()` which
324+
// spawns a task that calls `reload_from_storage()` first.
325+
let (events, mut subscriber) = pinned_events_cache.subscribe().await.unwrap();
319326
let mut events = events.into();
320327

321328
// Wait for the background task to reload the events.
@@ -368,9 +375,10 @@ async fn test_pinned_events_dont_include_thread_responses() {
368375
.await;
369376

370377
let client = server.client_builder().build().await;
378+
let event_cache = client.event_cache();
371379

372380
// Subscribe the event cache to sync updates.
373-
client.event_cache().subscribe().unwrap();
381+
event_cache.subscribe().unwrap();
374382

375383
// Sync the room with the pinned event ID in the room state.
376384
//
@@ -379,19 +387,19 @@ async fn test_pinned_events_dont_include_thread_responses() {
379387
// events.
380388
let pinned_events_state = f.room_pinned_events(vec![pinned_event_id.to_owned()]);
381389

382-
let room = server
390+
let _room = server
383391
.sync_room(
384392
&client,
385393
JoinedRoomBuilder::new(room_id).add_state_bulk(vec![pinned_events_state.into()]),
386394
)
387395
.await;
388396

389397
// Get the room event cache and subscribe to pinned events.
390-
let (room_event_cache, _drop_handles) = room.event_cache().await.unwrap();
398+
let (pinned_events_cache, _drop_handles) = event_cache.pinned_events(room_id).await.unwrap();
391399

392-
// Subscribe to pinned events - this triggers PinnedEventsCache::new() which
393-
// spawns a task that calls reload_from_storage() first.
394-
let (events, mut subscriber) = room_event_cache.subscribe_to_pinned_events().await.unwrap();
400+
// Getting the pinned events cache triggers `PinnedEventsCache::new()` which
401+
// spawns a task that calls `reload_from_storage()` first.
402+
let (events, mut subscriber) = pinned_events_cache.subscribe().await.unwrap();
395403
let mut events = events.into();
396404

397405
// Wait for the background task to reload the events.

0 commit comments

Comments
 (0)