Send LOADING to the UI model of stats screen if the state is loading

wordpress-mobile · Mar 2, 2024 · c069412 · c069412
1 parent 86e5706
commit c069412
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/WordPress/src/main/java/org/wordpress/android/ui/stats/refresh/lists/UiModelMapper.kt b/WordPress/src/main/java/org/wordpress/android/ui/stats/refresh/lists/UiModelMapper.kt
@@ -99,6 +99,10 @@ class UiModelMapper
             if (useCaseModels.isNotEmpty()) {
                 UiModel.Success(useCaseModels.mapNotNull { useCaseModel ->
                     when {
+                        useCaseModel.state == LOADING -> useCaseModel.stateData?.let {
+                            StatsBlock.Loading(useCaseModel.type, useCaseModel.stateData)
+                        }
+
                         useCaseModel.type == overViewType && useCaseModel.data != null -> StatsBlock.Success(
                             useCaseModel.type,
                             useCaseModel.data
@@ -109,10 +113,6 @@ class UiModelMapper
                             useCaseModel.data ?: listOf()
                         )
 
-                        useCaseModel.state == LOADING -> useCaseModel.stateData?.let {
-                            StatsBlock.Loading(useCaseModel.type, useCaseModel.stateData)
-                        }
-
                         useCaseModel.state == ERROR -> useCaseModel.stateData?.let {
                             StatsBlock.Error(useCaseModel.type, useCaseModel.stateData)
                         }