From 45cf32dfab790527efe5199ab090e7c03c3e050d Mon Sep 17 00:00:00 2001
From: Jonas Platte <jplatte@matrix.org>
Date: Tue, 8 Nov 2022 19:26:31 +0100
Subject: [PATCH] test(sdk): Add a test for an undecryptable event

---
 crates/matrix-sdk/src/room/timeline/tests.rs | 35 +++++++++++++++++++-
 1 file changed, 34 insertions(+), 1 deletion(-)

diff --git a/crates/matrix-sdk/src/room/timeline/tests.rs b/crates/matrix-sdk/src/room/timeline/tests.rs
index 5a857195f..54ec70a1e 100644
--- a/crates/matrix-sdk/src/room/timeline/tests.rs
+++ b/crates/matrix-sdk/src/room/timeline/tests.rs
@@ -30,6 +30,9 @@ use ruma::{
     events::{
         reaction::{self, ReactionEventContent},
         room::{
+            encrypted::{
+                EncryptedEventScheme, MegolmV1AesSha2ContentInit, RoomEncryptedEventContent,
+            },
             message::{self, Replacement, RoomMessageEventContent},
             redaction::OriginalSyncRoomRedactionEvent,
         },
@@ -40,7 +43,7 @@ use ruma::{
 };
 use serde_json::{json, Value as JsonValue};
 
-use super::{TimelineInner, TimelineItem};
+use super::{EncryptedMessage, TimelineInner, TimelineItem, TimelineItemContent};
 
 static ALICE: Lazy<&UserId> = Lazy::new(|| user_id!("@alice:server.name"));
 static BOB: Lazy<&UserId> = Lazy::new(|| user_id!("@bob:other.server"));
@@ -140,6 +143,36 @@ async fn edit_redacted() {
     assert_eq!(timeline.inner.items.lock_ref().len(), 1);
 }
 
+#[async_test]
+async fn unable_to_decrypt() {
+    let timeline = TestTimeline::new(&ALICE);
+    timeline.handle_live_message_event(
+        &BOB,
+        RoomEncryptedEventContent::new(
+            EncryptedEventScheme::MegolmV1AesSha2(
+                MegolmV1AesSha2ContentInit {
+                    ciphertext: "This can't be decrypted".to_owned(),
+                    sender_key: "whatever".to_owned(),
+                    device_id: "MyDevice".into(),
+                    session_id: "MySession".into(),
+                }
+                .into(),
+            ),
+            None,
+        ),
+    );
+    let timeline_items = timeline.inner.items.lock_ref();
+    assert_eq!(timeline_items.len(), 1);
+    let event = timeline_items[0].as_event().unwrap();
+    let session_id = assert_matches!(
+        event.content(),
+        TimelineItemContent::UnableToDecrypt(
+            EncryptedMessage::MegolmV1AesSha2 { session_id, .. },
+        ) => session_id
+    );
+    assert_eq!(session_id, "MySession");
+}
+
 struct TestTimeline {
     own_user_id: OwnedUserId,
     inner: TimelineInner,