From 0a29ad01851eb60f9695c7f0150ee8f0db5855ce Mon Sep 17 00:00:00 2001
From: TommyDL-Infomaniak <tommy.diluna@infomaniak.com>
Date: Mon, 24 Feb 2025 09:50:11 +0100
Subject: [PATCH] fix: Update AvatarView state when AvatarDisplayType is
 USER_AVATAR

---
 app/src/main/java/com/infomaniak/mail/views/AvatarView.kt | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/com/infomaniak/mail/views/AvatarView.kt b/app/src/main/java/com/infomaniak/mail/views/AvatarView.kt
index 299fcac7c3..dfa1d6bdba 100644
--- a/app/src/main/java/com/infomaniak/mail/views/AvatarView.kt
+++ b/app/src/main/java/com/infomaniak/mail/views/AvatarView.kt
@@ -196,7 +196,10 @@ class AvatarView @JvmOverloads constructor(
     ) {
         when (avatarDisplayType) {
             AvatarDisplayType.UNKNOWN_CORRESPONDENT -> loadUnknownUserAvatar()
-            AvatarDisplayType.USER_AVATAR -> AccountUtils.currentUser?.let { loadAvatar(user = it) }
+            AvatarDisplayType.USER_AVATAR -> {
+                state.update(correspondent, bimi)
+                AccountUtils.currentUser?.let { loadAvatar(user = it) }
+            }
             AvatarDisplayType.CUSTOM_AVATAR,
             AvatarDisplayType.INITIALS -> loadAvatarUsingDictionary(correspondent!!, contacts, bimi)
             AvatarDisplayType.BIMI -> loadBimiAvatar(correspondent!!, bimi!!)