From 79db05cffb67aa5475ef2b88fa7474cc01352b9a Mon Sep 17 00:00:00 2001
From: Jonas Platte <jplatte@matrix.org>
Date: Wed, 9 Nov 2022 09:42:55 +0100
Subject: [PATCH] test(sdk): Add a unit test for redacting a reaction

---
 crates/matrix-sdk/src/room/timeline/mod.rs   |   2 +
 crates/matrix-sdk/src/room/timeline/tests.rs | 119 +++++++++++++++++++
 2 files changed, 121 insertions(+)
 create mode 100644 crates/matrix-sdk/src/room/timeline/tests.rs

diff --git a/crates/matrix-sdk/src/room/timeline/mod.rs b/crates/matrix-sdk/src/room/timeline/mod.rs
index 8f9d4b8bd..8fcd9a46c 100644
--- a/crates/matrix-sdk/src/room/timeline/mod.rs
+++ b/crates/matrix-sdk/src/room/timeline/mod.rs
@@ -43,6 +43,8 @@ use crate::{
 
 mod event_handler;
 mod event_item;
+#[cfg(test)]
+mod tests;
 mod virtual_item;
 
 pub use self::{
diff --git a/crates/matrix-sdk/src/room/timeline/tests.rs b/crates/matrix-sdk/src/room/timeline/tests.rs
new file mode 100644
index 000000000..fce69869c
--- /dev/null
+++ b/crates/matrix-sdk/src/room/timeline/tests.rs
@@ -0,0 +1,119 @@
+// Copyright 2022 The Matrix.org Foundation C.I.C.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+//! Unit tests (based on private methods) for the timeline API.
+
+use std::sync::{
+    atomic::{AtomicU32, Ordering::SeqCst},
+    Arc,
+};
+
+use assert_matches::assert_matches;
+use futures_core::Stream;
+use futures_signals::signal_vec::{SignalVecExt, VecDiff};
+use futures_util::StreamExt;
+use matrix_sdk_test::async_test;
+use once_cell::sync::Lazy;
+use ruma::{
+    events::{
+        reaction::{self, ReactionEventContent},
+        room::{message::RoomMessageEventContent, redaction::OriginalSyncRoomRedactionEvent},
+        MessageLikeEventContent, OriginalSyncMessageLikeEvent,
+    },
+    serde::Raw,
+    server_name, user_id, EventId, MilliSecondsSinceUnixEpoch, OwnedUserId, UserId,
+};
+
+use super::{TimelineInner, TimelineItem};
+
+static ALICE: Lazy<&UserId> = Lazy::new(|| user_id!("@alice:server.name"));
+static BOB: Lazy<&UserId> = Lazy::new(|| user_id!("@bob:other.server"));
+
+#[async_test]
+async fn reaction_redaction() {
+    let timeline = TestTimeline::new(&ALICE);
+    let mut stream = timeline.stream();
+
+    timeline.handle_live_message_event(&ALICE, RoomMessageEventContent::text_plain("hi!"));
+    let item = assert_matches!(stream.next().await, Some(VecDiff::Push { value }) => value);
+    let event = item.as_event().unwrap();
+    assert_eq!(event.reactions().len(), 0);
+
+    let msg_event_id = event.event_id().unwrap();
+
+    let rel = reaction::Relation::new(msg_event_id.to_owned(), "+1".to_owned());
+    timeline.handle_live_message_event(&BOB, ReactionEventContent::new(rel));
+    let item =
+        assert_matches!(stream.next().await, Some(VecDiff::UpdateAt { index: 0, value }) => value);
+    let event = item.as_event().unwrap();
+    assert_eq!(event.reactions().len(), 1);
+
+    // TODO: After adding raw timeline items, check for one here
+
+    let reaction_event_id = event.event_id().unwrap();
+
+    timeline.handle_live_redaction(&BOB, reaction_event_id);
+    let item =
+        assert_matches!(stream.next().await, Some(VecDiff::UpdateAt { index: 0, value }) => value);
+    let event = item.as_event().unwrap();
+    assert_eq!(event.reactions().len(), 0);
+}
+
+struct TestTimeline {
+    own_user_id: OwnedUserId,
+    inner: TimelineInner,
+}
+
+impl TestTimeline {
+    fn new(own_user_id: &UserId) -> Self {
+        Self { own_user_id: own_user_id.to_owned(), inner: Default::default() }
+    }
+
+    fn stream(&self) -> impl Stream<Item = VecDiff<Arc<TimelineItem>>> {
+        self.inner.items.signal_vec_cloned().to_stream()
+    }
+
+    fn handle_live_message_event<C>(&self, sender: &UserId, content: C)
+    where
+        C: MessageLikeEventContent,
+    {
+        let ev = OriginalSyncMessageLikeEvent {
+            content,
+            event_id: EventId::new(server_name!("dummy.server")),
+            sender: sender.to_owned(),
+            origin_server_ts: next_server_ts(),
+            unsigned: Default::default(),
+        };
+        let raw = Raw::new(&ev).unwrap().cast();
+        self.inner.handle_live_event(raw, None, &self.own_user_id);
+    }
+
+    fn handle_live_redaction(&self, sender: &UserId, redacts: &EventId) {
+        let ev = OriginalSyncRoomRedactionEvent {
+            content: Default::default(),
+            redacts: redacts.to_owned(),
+            event_id: EventId::new(server_name!("dummy.server")),
+            sender: sender.to_owned(),
+            origin_server_ts: next_server_ts(),
+            unsigned: Default::default(),
+        };
+        let raw = Raw::new(&ev).unwrap().cast();
+        self.inner.handle_live_event(raw, None, &self.own_user_id);
+    }
+}
+
+fn next_server_ts() -> MilliSecondsSinceUnixEpoch {
+    static NEXT_TS: AtomicU32 = AtomicU32::new(0);
+    MilliSecondsSinceUnixEpoch(NEXT_TS.fetch_add(1, SeqCst).into())
+}