timeline: add test for poll edit in relations overriding pending poll edit

Author: bnjbvr

crates/matrix-sdk-ui/src/timeline/tests/edit.rs

@@ -217,7 +217,7 @@ async fn test_edit_updates_encryption_info() {
 }
 
 #[async_test]
-async fn test_relations_edit_overrides_pending_edit() {
+async fn test_relations_edit_overrides_pending_edit_msg() {
     let timeline = TestTimeline::new();
     let mut stream = timeline.subscribe().await;
 
@@ -278,3 +278,81 @@ async fn test_relations_edit_overrides_pending_edit() {
 
     assert_pending!(stream);
 }
+
+#[async_test]
+async fn test_relations_edit_overrides_pending_edit_poll() {
+    let timeline = TestTimeline::new();
+    let mut stream = timeline.subscribe().await;
+
+    let f = &timeline.factory;
+
+    let original_event_id = event_id!("$original");
+    let edit1_event_id = event_id!("$edit1");
+    let edit2_event_id = event_id!("$edit2");
+
+    // Pending edit is stashed, nothing comes from the stream.
+    timeline
+        .handle_live_event(
+            f.poll_edit(
+                original_event_id,
+                "Can the fake slim shady please stand up?",
+                vec!["Excuse me?"],
+            )
+            .sender(*ALICE)
+            .event_id(edit1_event_id),
+        )
+        .await;
+    assert_pending!(stream);
+
+    // Now we receive the original event, with a bundled relations group.
+    let mut relations = BundledMessageLikeRelations::new();
+    relations.replace = Some(Box::new(
+        f.poll_edit(
+            original_event_id,
+            "Can the real slim shady please stand up?",
+            vec!["Excuse me?", "Please stand up 🎵", "Please stand up 🎶"],
+        )
+        .sender(*ALICE)
+        .event_id(edit2_event_id)
+        .into(),
+    ));
+
+    let ev = f
+        .poll_start(
+            "Can the fake slim shady please stand down?\nExcuse me?",
+            "Can the fake slim shady please stand down?",
+            vec!["Excuse me?"],
+        )
+        .sender(*ALICE)
+        .event_id(original_event_id)
+        .bundled_relations(relations);
+
+    timeline.handle_live_event(ev).await;
+
+    let item = assert_next_matches!(stream, VectorDiff::PushBack { value } => value);
+
+    // We receive the latest edit, not the pending one.
+    let event = item.as_event().unwrap();
+    assert_eq!(
+        event
+            .latest_edit_json()
+            .expect("we should have an edit json")
+            .deserialize()
+            .unwrap()
+            .event_id(),
+        edit2_event_id
+    );
+
+    let poll = event.content().as_poll().unwrap();
+    assert!(poll.has_been_edited);
+    assert_eq!(
+        poll.start_event_content.poll_start.question.text,
+        "Can the real slim shady please stand up?"
+    );
+    assert_eq!(poll.start_event_content.poll_start.answers.len(), 3);
+
+    let day_divider = assert_next_matches!(stream, VectorDiff::PushFront { value } => value);
+    assert!(day_divider.is_day_divider());
+
+    assert_pending!(stream);
+}