Closes #26995: add sorting of wallpapers on wallpapers onboarding tool
This commit is contained in:
parent
dafe5dbdd6
commit
4972b70018
|
@ -28,6 +28,7 @@ import org.mozilla.fenix.components.FenixSnackbar
|
|||
import org.mozilla.fenix.ext.components
|
||||
import org.mozilla.fenix.ext.requireComponents
|
||||
import org.mozilla.fenix.ext.settings
|
||||
import org.mozilla.fenix.settings.wallpaper.getWallpapersForOnboarding
|
||||
import org.mozilla.fenix.theme.FirefoxTheme
|
||||
import org.mozilla.fenix.wallpapers.Wallpaper
|
||||
import org.mozilla.fenix.wallpapers.WallpaperOnboarding
|
||||
|
@ -84,7 +85,7 @@ class WallpaperOnboardingDialogFragment : BottomSheetDialogFragment() {
|
|||
setContent {
|
||||
FirefoxTheme {
|
||||
val wallpapers = appStore.observeAsComposableState { state ->
|
||||
state.wallpaperState.availableWallpapers.take(THUMBNAILS_SELECTION_COUNT)
|
||||
state.wallpaperState.availableWallpapers.getWallpapersForOnboarding()
|
||||
}.value ?: listOf()
|
||||
val currentWallpaper = appStore.observeAsComposableState { state ->
|
||||
state.wallpaperState.currentWallpaper
|
||||
|
@ -147,6 +148,13 @@ class WallpaperOnboardingDialogFragment : BottomSheetDialogFragment() {
|
|||
}
|
||||
|
||||
companion object {
|
||||
// The number of wallpaper thumbnails to display.
|
||||
const val THUMBNAILS_SELECTION_COUNT = 6
|
||||
|
||||
// The desired amount of seasonal wallpapers inside of the selector.
|
||||
const val SEASONAL_WALLPAPERS_COUNT = 3
|
||||
|
||||
// The desired amount of seasonal wallpapers inside of the selector.
|
||||
const val CLASSIC_WALLPAPERS_COUNT = 2
|
||||
}
|
||||
}
|
||||
|
|
|
@ -4,24 +4,65 @@
|
|||
|
||||
package org.mozilla.fenix.settings.wallpaper
|
||||
|
||||
import org.mozilla.fenix.onboarding.WallpaperOnboardingDialogFragment.Companion.CLASSIC_WALLPAPERS_COUNT
|
||||
import org.mozilla.fenix.onboarding.WallpaperOnboardingDialogFragment.Companion.SEASONAL_WALLPAPERS_COUNT
|
||||
import org.mozilla.fenix.onboarding.WallpaperOnboardingDialogFragment.Companion.THUMBNAILS_SELECTION_COUNT
|
||||
import org.mozilla.fenix.wallpapers.Wallpaper
|
||||
|
||||
/**
|
||||
* The extension function to group wallpapers according to their name.
|
||||
**/
|
||||
fun List<Wallpaper>.groupByDisplayableCollection(): Map<Wallpaper.Collection, List<Wallpaper>> = groupBy {
|
||||
it.collection
|
||||
}.filter {
|
||||
it.key.name != "default"
|
||||
}.map {
|
||||
val wallpapers = it.value.filter { wallpaper ->
|
||||
wallpaper.thumbnailFileState == Wallpaper.ImageFileState.Downloaded
|
||||
fun List<Wallpaper>.groupByDisplayableCollection(): Map<Wallpaper.Collection, List<Wallpaper>> =
|
||||
groupBy {
|
||||
it.collection
|
||||
}.filter {
|
||||
it.key.name != "default"
|
||||
}.map {
|
||||
val wallpapers = it.value.filter { wallpaper ->
|
||||
wallpaper.thumbnailFileState == Wallpaper.ImageFileState.Downloaded
|
||||
}
|
||||
if (it.key.name == "classic-firefox") {
|
||||
it.key to listOf(Wallpaper.Default) + wallpapers
|
||||
} else {
|
||||
it.key to wallpapers
|
||||
}
|
||||
}.toMap().takeIf {
|
||||
it.isNotEmpty()
|
||||
} ?: mapOf(Wallpaper.DefaultCollection to listOf(Wallpaper.Default))
|
||||
|
||||
/**
|
||||
* Returns a list of wallpapers to display in the wallpaper onboarding.
|
||||
*
|
||||
* The ideal scenario is to return a list of wallpaper in the following order: 1 default, 3 seasonal and
|
||||
* 2 classic wallpapers, but in case where there are less than 3 seasonal wallpapers, the remaining
|
||||
* wallpapers are filled by classic wallpapers. If we have less than 6 wallpapers, return all the available
|
||||
* seasonal and classic wallpapers.
|
||||
*/
|
||||
fun List<Wallpaper>.getWallpapersForOnboarding(): List<Wallpaper> {
|
||||
val result = mutableListOf(Wallpaper.Default)
|
||||
val classicWallpapers = mutableListOf<Wallpaper>()
|
||||
val seasonalWallpapers = mutableListOf<Wallpaper>()
|
||||
|
||||
for (wallpaper in this) {
|
||||
if (wallpaper == Wallpaper.Default) continue
|
||||
|
||||
if (wallpaper.collection.name == "classic-firefox") {
|
||||
classicWallpapers.add(wallpaper)
|
||||
} else {
|
||||
seasonalWallpapers.add(wallpaper)
|
||||
}
|
||||
}
|
||||
if (it.key.name == "classic-firefox") {
|
||||
it.key to listOf(Wallpaper.Default) + wallpapers
|
||||
|
||||
if (seasonalWallpapers.size < SEASONAL_WALLPAPERS_COUNT) {
|
||||
result.addAll(seasonalWallpapers)
|
||||
result.addAll(classicWallpapers.take((THUMBNAILS_SELECTION_COUNT - 1) - seasonalWallpapers.size))
|
||||
} else if (classicWallpapers.size < CLASSIC_WALLPAPERS_COUNT) {
|
||||
result.addAll(seasonalWallpapers.take((THUMBNAILS_SELECTION_COUNT - 1) - classicWallpapers.size))
|
||||
result.addAll(classicWallpapers)
|
||||
} else {
|
||||
it.key to wallpapers
|
||||
result.addAll(seasonalWallpapers.take(SEASONAL_WALLPAPERS_COUNT))
|
||||
result.addAll(classicWallpapers.take(CLASSIC_WALLPAPERS_COUNT))
|
||||
}
|
||||
}.toMap().takeIf {
|
||||
it.isNotEmpty()
|
||||
} ?: mapOf(Wallpaper.DefaultCollection to listOf(Wallpaper.Default))
|
||||
|
||||
return result
|
||||
}
|
||||
|
|
|
@ -5,6 +5,7 @@
|
|||
package org.mozilla.fenix.settings.wallpaper
|
||||
|
||||
import org.junit.Assert.assertEquals
|
||||
import org.junit.Assert.assertTrue
|
||||
import org.junit.Test
|
||||
import org.mozilla.fenix.wallpapers.Wallpaper
|
||||
|
||||
|
@ -54,6 +55,102 @@ class ExtensionsTest {
|
|||
assertEquals(downloadedSeasonalWallpapers, result[seasonalCollection])
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN two collections of appropriate size WHEN fetched for onboarding THEN result contains 3 seasonal and 2 classic`() {
|
||||
val seasonalCollection = getSeasonalCollection("finally fall")
|
||||
val seasonalWallpapers = (0..5).map { generateSeasonalWallpaperCollection("${seasonalCollection.name}$it", seasonalCollection.name) }
|
||||
val classicFirefoxWallpapers = (0..5).map { generateClassicFirefoxWallpaper("firefox$it") }
|
||||
val allWallpapers = listOf(Wallpaper.Default) + classicFirefoxWallpapers + seasonalWallpapers
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(3, result.count { it.collection.name == "finally fall" })
|
||||
assertEquals(2, result.count { it.collection.name == classicCollection.name })
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN five collections of insufficient size WHEN fetched for onboarding THEN result contains 3 seasonal and 2 classic`() {
|
||||
val seasonalCollectionA = getSeasonalCollection("finally winter")
|
||||
val seasonalWallpapers = generateSeasonalWallpaperCollection("${seasonalCollectionA.name}$0", seasonalCollectionA.name)
|
||||
val seasonalCollectionB = getSeasonalCollection("finally spring")
|
||||
val seasonalWallpaperB = generateSeasonalWallpaperCollection("${seasonalCollectionB.name}$0", seasonalCollectionB.name)
|
||||
val seasonalCollectionC = getSeasonalCollection("finally summer")
|
||||
val seasonalWallpapersC = generateSeasonalWallpaperCollection("${seasonalCollectionC.name}$0", seasonalCollectionC.name)
|
||||
val seasonalCollectionD = getSeasonalCollection("finally autumn")
|
||||
val seasonalWallpaperD = generateSeasonalWallpaperCollection("${seasonalCollectionD.name}$0", seasonalCollectionD.name)
|
||||
val seasonalCollectionE = getSeasonalCollection("finally vacation")
|
||||
val seasonalWallpapersE = generateSeasonalWallpaperCollection("${seasonalCollectionE.name}$0", seasonalCollectionE.name)
|
||||
|
||||
val classicFirefoxWallpapers = (0..5).map { generateClassicFirefoxWallpaper("firefox$it") }
|
||||
val allWallpapers = listOf(Wallpaper.Default) + classicFirefoxWallpapers + seasonalWallpapers +
|
||||
seasonalWallpaperB + seasonalWallpapersC + seasonalWallpaperD + seasonalWallpapersE
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(3, result.count { it.collection.name != classicCollection.name && it != Wallpaper.Default })
|
||||
assertEquals(2, result.count { it.collection.name == classicCollection.name })
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN seasonal collection of insufficient size WHEN grouped for onboarding THEN result contains all seasonal and the rest is classic`() {
|
||||
val seasonalCollection = getSeasonalCollection("finally fall")
|
||||
val seasonalWallpapers = generateSeasonalWallpaperCollection("${seasonalCollection.name}$0", seasonalCollection.name)
|
||||
val classicFirefoxWallpapers = (0..5).map { generateClassicFirefoxWallpaper("firefox$it") }
|
||||
val allWallpapers = listOf(Wallpaper.Default) + classicFirefoxWallpapers + seasonalWallpapers
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(1, result.count { it.collection.name == "finally fall" })
|
||||
assertEquals(4, result.count { it.collection.name == classicCollection.name })
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN no seasonal collection WHEN grouped for onboarding THEN result contains all classic`() {
|
||||
val classicFirefoxWallpapers = (0..5).map { generateClassicFirefoxWallpaper("firefox$it") }
|
||||
val allWallpapers = listOf(Wallpaper.Default) + classicFirefoxWallpapers
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(5, result.count { it.collection.name == classicCollection.name })
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN insufficient items in classic collection WHEN grouped for onboarding THEN result contains all classic`() {
|
||||
val classicFirefoxWallpapers = (0..2).map { generateClassicFirefoxWallpaper("firefox$it") }
|
||||
val allWallpapers = listOf(Wallpaper.Default) + classicFirefoxWallpapers
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(3, result.count { it.collection.name == classicCollection.name })
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN no items in classic collection and some seasonal WHEN grouped for onboarding THEN result contains all seasonal`() {
|
||||
val seasonalCollection = getSeasonalCollection("finally fall")
|
||||
val seasonalWallpapers = (0..5).map { generateSeasonalWallpaperCollection("${seasonalCollection.name}$it", seasonalCollection.name) }
|
||||
val allWallpapers = listOf(Wallpaper.Default) + seasonalWallpapers
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(5, result.count { it.collection.name == "finally fall" })
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
@Test
|
||||
fun `GIVEN no items WHEN grouped for onboarding THEN result contains the default option`() {
|
||||
val allWallpapers = listOf(Wallpaper.Default)
|
||||
|
||||
val result = allWallpapers.getWallpapersForOnboarding()
|
||||
|
||||
assertEquals(1, result.size)
|
||||
assertTrue(result.contains(Wallpaper.Default))
|
||||
}
|
||||
|
||||
private fun generateClassicFirefoxWallpaper(name: String) = Wallpaper(
|
||||
name = name,
|
||||
textColor = 0L,
|
||||
|
|
Loading…
Reference in New Issue