From cf38553701160c76828fd7056bf03f58335d81c8 Mon Sep 17 00:00:00 2001
From: Benoit Marty <benoit@matrix.org>
Date: Thu, 22 Feb 2024 14:03:59 +0100
Subject: [PATCH] Add more states in DirectLogoutStateProvider

---
 .../features/logout/api/direct/DirectLogoutStateProvider.kt | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/features/logout/api/src/main/kotlin/io/element/android/features/logout/api/direct/DirectLogoutStateProvider.kt b/features/logout/api/src/main/kotlin/io/element/android/features/logout/api/direct/DirectLogoutStateProvider.kt
index 48f1a292f0..585e7cb570 100644
--- a/features/logout/api/src/main/kotlin/io/element/android/features/logout/api/direct/DirectLogoutStateProvider.kt
+++ b/features/logout/api/src/main/kotlin/io/element/android/features/logout/api/direct/DirectLogoutStateProvider.kt
@@ -22,7 +22,11 @@ import io.element.android.libraries.architecture.AsyncAction
 open class DirectLogoutStateProvider : PreviewParameterProvider<DirectLogoutState> {
     override val values: Sequence<DirectLogoutState>
         get() = sequenceOf(
-            aDirectLogoutState()
+            aDirectLogoutState(),
+            aDirectLogoutState(logoutAction = AsyncAction.Confirming),
+            aDirectLogoutState(logoutAction = AsyncAction.Loading),
+            aDirectLogoutState(logoutAction = AsyncAction.Failure(Exception("Error"))),
+            aDirectLogoutState(logoutAction = AsyncAction.Success("success")),
         )
 }