From f7fe898e662478bcd6210bbd600988ecfc295cf7 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Thu, 22 Aug 2024 15:22:14 +0200
Subject: [PATCH] Add test on extension MatrixRoom.getCurrentRoomMember

---
 .../matrix/ui/room/MatrixRoomMembersTest.kt   | 35 +++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/libraries/matrixui/src/test/kotlin/io/element/android/libraries/matrix/ui/room/MatrixRoomMembersTest.kt b/libraries/matrixui/src/test/kotlin/io/element/android/libraries/matrix/ui/room/MatrixRoomMembersTest.kt
index 246f4f4fd5..1c3d6a82ad 100644
--- a/libraries/matrixui/src/test/kotlin/io/element/android/libraries/matrix/ui/room/MatrixRoomMembersTest.kt
+++ b/libraries/matrixui/src/test/kotlin/io/element/android/libraries/matrix/ui/room/MatrixRoomMembersTest.kt
@@ -158,4 +158,39 @@ class MatrixRoomMembersTest {
             assertThat(awaitItem().value).isEqualTo(roomMember2)
         }
     }
+
+    @Test
+    fun `getCurrentRoomMember returns the current user`() = runTest {
+        val matrixRoom = FakeMatrixRoom(sessionId = A_USER_ID)
+        moleculeFlow(RecompositionMode.Immediate) {
+            matrixRoom.getCurrentRoomMember(
+                MatrixRoomMembersState.Ready(
+                    persistentListOf(
+                        roomMember1,
+                        roomMember2,
+                        roomMember3,
+                    )
+                )
+            )
+        }.test {
+            assertThat(awaitItem().value).isEqualTo(roomMember1)
+        }
+    }
+
+    @Test
+    fun `getCurrentRoomMember returns null if the member is not found`() = runTest {
+        val matrixRoom = FakeMatrixRoom(sessionId = A_USER_ID)
+        moleculeFlow(RecompositionMode.Immediate) {
+            matrixRoom.getCurrentRoomMember(
+                MatrixRoomMembersState.Ready(
+                    persistentListOf(
+                        roomMember2,
+                        roomMember3,
+                    )
+                )
+            )
+        }.test {
+            assertThat(awaitItem().value).isNull()
+        }
+    }
 }