From e49d689fe870d87d28b311ef2e7c7634cb2a4e1d Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Tue, 8 Oct 2024 11:23:11 +0200
Subject: [PATCH] Improve code.

---
 .../android/features/messages/impl/MessagesStateProvider.kt   | 4 +++-
 .../crypto/identity/MessagesViewWithIdentityChangePreview.kt  | 2 +-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesStateProvider.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesStateProvider.kt
index f42af8231c..c8a8ee6f1f 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesStateProvider.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/MessagesStateProvider.kt
@@ -10,6 +10,7 @@ package io.element.android.features.messages.impl
 import androidx.compose.ui.tooling.preview.PreviewParameterProvider
 import io.element.android.features.messages.impl.actionlist.ActionListState
 import io.element.android.features.messages.impl.actionlist.anActionListState
+import io.element.android.features.messages.impl.crypto.identity.IdentityChangeState
 import io.element.android.features.messages.impl.crypto.identity.anIdentityChangeState
 import io.element.android.features.messages.impl.messagecomposer.AttachmentsState
 import io.element.android.features.messages.impl.messagecomposer.MessageComposerState
@@ -107,6 +108,7 @@ fun aMessagesState(
         focusedEventIndex = 2,
     ),
     timelineProtectionState: TimelineProtectionState = aTimelineProtectionState(),
+    identityChangeState: IdentityChangeState = anIdentityChangeState(),
     readReceiptBottomSheetState: ReadReceiptBottomSheetState = aReadReceiptBottomSheetState(),
     actionListState: ActionListState = anActionListState(),
     customReactionState: CustomReactionState = aCustomReactionState(),
@@ -126,7 +128,7 @@ fun aMessagesState(
     composerState = composerState,
     voiceMessageComposerState = voiceMessageComposerState,
     timelineProtectionState = timelineProtectionState,
-    identityChangeState = anIdentityChangeState(),
+    identityChangeState = identityChangeState,
     timelineState = timelineState,
     readReceiptBottomSheetState = readReceiptBottomSheetState,
     actionListState = actionListState,
diff --git a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/crypto/identity/MessagesViewWithIdentityChangePreview.kt b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/crypto/identity/MessagesViewWithIdentityChangePreview.kt
index c484e4e8c7..a97de3f663 100644
--- a/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/crypto/identity/MessagesViewWithIdentityChangePreview.kt
+++ b/features/messages/impl/src/main/kotlin/io/element/android/features/messages/impl/crypto/identity/MessagesViewWithIdentityChangePreview.kt
@@ -20,7 +20,7 @@ internal fun MessagesViewWithIdentityChangePreview(
     @PreviewParameter(IdentityChangeStateProvider::class) identityChangeState: IdentityChangeState
 ) = ElementPreview {
     MessagesView(
-        state = aMessagesState().copy(identityChangeState = identityChangeState),
+        state = aMessagesState(identityChangeState = identityChangeState),
         onBackClick = {},
         onRoomDetailsClick = {},
         onEventClick = { false },