From 6b59c82a5ccb671d76dc5097a0d736de6fc0a8a1 Mon Sep 17 00:00:00 2001 From: Benoit Marty Date: Wed, 8 Oct 2025 10:29:15 +0200 Subject: [PATCH] Improve InMemoryAnnouncementStore --- .../impl/store/InMemoryAnnouncementStore.kt | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/features/announcement/impl/src/test/kotlin/io/element/android/features/announcement/impl/store/InMemoryAnnouncementStore.kt b/features/announcement/impl/src/test/kotlin/io/element/android/features/announcement/impl/store/InMemoryAnnouncementStore.kt index 6dfcac6f39..bac711c59c 100644 --- a/features/announcement/impl/src/test/kotlin/io/element/android/features/announcement/impl/store/InMemoryAnnouncementStore.kt +++ b/features/announcement/impl/src/test/kotlin/io/element/android/features/announcement/impl/store/InMemoryAnnouncementStore.kt @@ -20,20 +20,20 @@ class InMemoryAnnouncementStore( private val newNotificationSoundAnnouncement = MutableStateFlow(initialNewNotificationSoundAnnouncementStatus) override suspend fun setAnnouncementStatus(announcement: Announcement, status: AnnouncementStore.AnnouncementStatus) { - when (announcement) { - Announcement.Space -> spaceAnnouncement.value = status - Announcement.NewNotificationSound -> newNotificationSoundAnnouncement.value = status - } + announcement.toMutableStateFlow().value = status } override fun announcementStatusFlow(announcement: Announcement): Flow { - return when (announcement) { - Announcement.Space -> spaceAnnouncement.asStateFlow() - Announcement.NewNotificationSound -> newNotificationSoundAnnouncement.asStateFlow() - } + return announcement.toMutableStateFlow().asStateFlow() } override suspend fun reset() { spaceAnnouncement.value = AnnouncementStore.AnnouncementStatus.NeverShown + newNotificationSoundAnnouncement.value = AnnouncementStore.AnnouncementStatus.NeverShown + } + + private fun Announcement.toMutableStateFlow() = when (this) { + Announcement.Space -> spaceAnnouncement + Announcement.NewNotificationSound -> newNotificationSoundAnnouncement } }