Skip to content

Commit 664fb74

Browse files
committed
bench: fix pinned events benchmark
1 parent df81723 commit 664fb74

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

benchmarks/benches/room_bench.rs

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@ use matrix_sdk_base::{
1010
store::StoreConfig, BaseClient, RoomInfo, RoomState, SessionMeta, StateChanges, StateStore,
1111
};
1212
use matrix_sdk_sqlite::SqliteStateStore;
13-
use matrix_sdk_test::{EventBuilder, JoinedRoomBuilder, StateTestEvent, SyncResponseBuilder};
13+
use matrix_sdk_test::{EventBuilder, JoinedRoomBuilder, StateTestEvent, SyncResponseBuilder, ALICE};
1414
use matrix_sdk_ui::{timeline::TimelineFocus, Timeline};
1515
use ruma::{
1616
api::client::membership::get_member_events,
@@ -20,6 +20,8 @@ use ruma::{
2020
serde::Raw,
2121
user_id, EventId, MilliSecondsSinceUnixEpoch, OwnedEventId, OwnedUserId,
2222
};
23+
use ruma::events::AnyStateEvent;
24+
use ruma::events::room::member::MembershipState;
2325
use serde::Serialize;
2426
use serde_json::json;
2527
use tokio::runtime::Builder;
@@ -153,11 +155,10 @@ pub fn load_pinned_events_benchmark(c: &mut Criterion) {
153155

154156
runtime.block_on(
155157
Mock::given(method("GET"))
156-
.and(path_regex(r"/_matrix/client/r0/rooms/.*/event/.*"))
158+
.and(path_regex(r"/_matrix/client/r0/rooms/.*/context/.*"))
157159
.respond_with(move |r: &Request| {
158160
let segments: Vec<&str> = r.url.path_segments().expect("Invalid path").collect();
159161
let event_id_str = segments[6];
160-
// let f = EventFactory::new().room(&room_id)
161162
let event_id = EventId::parse(event_id_str).expect("Invalid event id in response");
162163
let event = f
163164
.text_msg(format!("Message {event_id_str}"))
@@ -166,13 +167,16 @@ pub fn load_pinned_events_benchmark(c: &mut Criterion) {
166167
.into_raw_sync();
167168
ResponseTemplate::new(200)
168169
.set_delay(Duration::from_millis(50))
169-
.set_body_json(event.json())
170+
.set_body_json(json!({
171+
"events_before": [],
172+
"event": event.json(),
173+
"events_after": [],
174+
"state": [],
175+
})
176+
)
170177
})
171178
.mount(&server),
172179
);
173-
// runtime.block_on(server.reset());
174-
175-
client.event_cache().subscribe().unwrap();
176180

177181
let room = client.get_room(&room_id).expect("Room not found");
178182
assert!(!room.pinned_event_ids().is_empty());
@@ -186,6 +190,8 @@ pub fn load_pinned_events_benchmark(c: &mut Criterion) {
186190

187191
group.bench_function(BenchmarkId::new("load_pinned_events", name), |b| {
188192
b.to_async(&runtime).iter(|| async {
193+
client.event_cache().subscribe().unwrap();
194+
189195
assert!(!room.pinned_event_ids().is_empty());
190196
assert_eq!(room.pinned_event_ids().len(), PINNED_EVENTS_COUNT);
191197

0 commit comments

Comments
 (0)