From 82926d6f089873e12bcfcd64faaddea776d2564d Mon Sep 17 00:00:00 2001
From: Benjamin Bouvier <benjamin@bouvier.cc>
Date: Tue, 8 Jul 2025 15:53:17 +0200
Subject: [PATCH] test(timeline): allow setting the timeline focus in
 `TestTimeline`

---
 crates/matrix-sdk-ui/src/timeline/tests/mod.rs | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/crates/matrix-sdk-ui/src/timeline/tests/mod.rs b/crates/matrix-sdk-ui/src/timeline/tests/mod.rs
index b6785934f..f351a9c77 100644
--- a/crates/matrix-sdk-ui/src/timeline/tests/mod.rs
+++ b/crates/matrix-sdk-ui/src/timeline/tests/mod.rs
@@ -82,6 +82,7 @@ mod virt;
 #[derive(Default)]
 struct TestTimelineBuilder {
     provider: Option<TestRoomDataProvider>,
+    focus: Option<TimelineFocus>,
     internal_id_prefix: Option<String>,
     utd_hook: Option<Arc<UtdHookManager>>,
     is_room_encrypted: bool,
@@ -120,10 +121,15 @@ impl TestTimelineBuilder {
         self
     }
 
+    fn focus(mut self, focus: TimelineFocus) -> Self {
+        self.focus = Some(focus);
+        self
+    }
+
     fn build(self) -> TestTimeline {
         let controller = TimelineController::new(
             self.provider.unwrap_or_default(),
-            TimelineFocus::Live { hide_threaded_events: false },
+            self.focus.unwrap_or(TimelineFocus::Live { hide_threaded_events: false }),
             self.internal_id_prefix,
             self.utd_hook,
             self.is_room_encrypted,