diff options
Diffstat (limited to 'tests/src/com/android/customization/testing')
3 files changed, 325 insertions, 143 deletions
diff --git a/tests/src/com/android/customization/testing/TestCustomizationInjector.java b/tests/src/com/android/customization/testing/TestCustomizationInjector.java deleted file mode 100644 index d6093356..00000000 --- a/tests/src/com/android/customization/testing/TestCustomizationInjector.java +++ /dev/null @@ -1,143 +0,0 @@ -package com.android.customization.testing; - -import android.content.Context; - -import androidx.fragment.app.FragmentActivity; - -import com.android.customization.model.theme.OverlayManagerCompat; -import com.android.customization.model.theme.ThemeBundleProvider; -import com.android.customization.model.theme.ThemeManager; -import com.android.customization.module.CustomizationInjector; -import com.android.customization.module.CustomizationPreferences; -import com.android.customization.module.ThemesUserEventLogger; -import com.android.customization.picker.quickaffordance.data.repository.KeyguardQuickAffordancePickerRepository; -import com.android.customization.picker.quickaffordance.domain.interactor.KeyguardQuickAffordancePickerInteractor; -import com.android.customization.picker.quickaffordance.domain.interactor.KeyguardQuickAffordanceSnapshotRestorer; -import com.android.systemui.shared.customization.data.content.CustomizationProviderClient; -import com.android.systemui.shared.customization.data.content.CustomizationProviderClientImpl; -import com.android.wallpaper.config.BaseFlags; -import com.android.wallpaper.module.DrawableLayerResolver; -import com.android.wallpaper.module.PackageStatusNotifier; -import com.android.wallpaper.module.UserEventLogger; -import com.android.wallpaper.picker.undo.domain.interactor.SnapshotRestorer; -import com.android.wallpaper.testing.TestInjector; - -import java.util.HashMap; -import java.util.Map; - -import kotlinx.coroutines.Dispatchers; - -/** - * Test implementation of the dependency injector. - */ -public class TestCustomizationInjector extends TestInjector implements CustomizationInjector { - private CustomizationPreferences mCustomizationPreferences; - private ThemeManager mThemeManager; - private PackageStatusNotifier mPackageStatusNotifier; - private DrawableLayerResolver mDrawableLayerResolver; - private UserEventLogger mUserEventLogger; - private KeyguardQuickAffordancePickerInteractor mKeyguardQuickAffordancePickerInteractor; - private BaseFlags mFlags; - private CustomizationProviderClient mCustomizationProviderClient; - private KeyguardQuickAffordanceSnapshotRestorer mKeyguardQuickAffordanceSnapshotRestorer; - - @Override - public CustomizationPreferences getCustomizationPreferences(Context context) { - if (mCustomizationPreferences == null) { - mCustomizationPreferences = new TestDefaultCustomizationPreferences(context); - } - return mCustomizationPreferences; - } - - @Override - public ThemeManager getThemeManager( - ThemeBundleProvider provider, - FragmentActivity activity, - OverlayManagerCompat overlayManagerCompat, - ThemesUserEventLogger logger) { - if (mThemeManager == null) { - mThemeManager = new TestThemeManager(provider, activity, overlayManagerCompat, logger); - } - return mThemeManager; - } - - @Override - public PackageStatusNotifier getPackageStatusNotifier(Context context) { - if (mPackageStatusNotifier == null) { - mPackageStatusNotifier = new TestPackageStatusNotifier(); - } - return mPackageStatusNotifier; - } - - @Override - public DrawableLayerResolver getDrawableLayerResolver() { - if (mDrawableLayerResolver == null) { - mDrawableLayerResolver = new TestDrawableLayerResolver(); - } - return mDrawableLayerResolver; - } - - @Override - public UserEventLogger getUserEventLogger(Context unused) { - if (mUserEventLogger == null) { - mUserEventLogger = new TestThemesUserEventLogger(); - } - return mUserEventLogger; - } - - @Override - public KeyguardQuickAffordancePickerInteractor getKeyguardQuickAffordancePickerInteractor( - Context context) { - if (mKeyguardQuickAffordancePickerInteractor == null) { - final CustomizationProviderClient client = - new CustomizationProviderClientImpl(context, Dispatchers.getIO()); - mKeyguardQuickAffordancePickerInteractor = new KeyguardQuickAffordancePickerInteractor( - new KeyguardQuickAffordancePickerRepository(client, Dispatchers.getIO()), - client, - () -> getKeyguardQuickAffordanceSnapshotRestorer(context)); - } - return mKeyguardQuickAffordancePickerInteractor; - } - - @Override - public BaseFlags getFlags() { - if (mFlags == null) { - mFlags = new BaseFlags() {}; - } - - return mFlags; - } - - @Override - public Map<Integer, SnapshotRestorer> getSnapshotRestorers(Context context) { - final Map<Integer, SnapshotRestorer> restorers = new HashMap<>(); - restorers.put( - KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER, - getKeyguardQuickAffordanceSnapshotRestorer(context)); - return restorers; - } - - /** Returns the {@link CustomizationProviderClient}. */ - private CustomizationProviderClient getKeyguardQuickAffordancePickerProviderClient( - Context context) { - if (mCustomizationProviderClient == null) { - mCustomizationProviderClient = - new CustomizationProviderClientImpl(context, Dispatchers.getIO()); - } - - return mCustomizationProviderClient; - } - - private KeyguardQuickAffordanceSnapshotRestorer getKeyguardQuickAffordanceSnapshotRestorer( - Context context) { - if (mKeyguardQuickAffordanceSnapshotRestorer == null) { - mKeyguardQuickAffordanceSnapshotRestorer = new KeyguardQuickAffordanceSnapshotRestorer( - getKeyguardQuickAffordancePickerInteractor(context), - getKeyguardQuickAffordancePickerProviderClient(context)); - } - - return mKeyguardQuickAffordanceSnapshotRestorer; - } - - private static final int KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER = 1; -} diff --git a/tests/src/com/android/customization/testing/TestCustomizationInjector.kt b/tests/src/com/android/customization/testing/TestCustomizationInjector.kt new file mode 100644 index 00000000..b49e6546 --- /dev/null +++ b/tests/src/com/android/customization/testing/TestCustomizationInjector.kt @@ -0,0 +1,289 @@ +package com.android.customization.testing + +import android.app.Activity +import android.content.Context +import android.text.TextUtils +import androidx.fragment.app.FragmentActivity +import com.android.customization.model.color.ColorCustomizationManager +import com.android.customization.model.color.ColorOptionsProvider +import com.android.customization.model.theme.OverlayManagerCompat +import com.android.customization.model.theme.ThemeBundleProvider +import com.android.customization.model.theme.ThemeManager +import com.android.customization.module.CustomizationInjector +import com.android.customization.module.CustomizationPreferences +import com.android.customization.module.ThemesUserEventLogger +import com.android.customization.picker.clock.data.repository.ClockRegistryProvider +import com.android.customization.picker.clock.data.repository.FakeClockPickerRepository +import com.android.customization.picker.clock.domain.interactor.ClockPickerInteractor +import com.android.customization.picker.clock.ui.view.ClockViewFactory +import com.android.customization.picker.clock.ui.viewmodel.ClockCarouselViewModel +import com.android.customization.picker.clock.ui.viewmodel.ClockSectionViewModel +import com.android.customization.picker.clock.ui.viewmodel.ClockSettingsViewModel +import com.android.customization.picker.color.data.repository.ColorPickerRepository +import com.android.customization.picker.color.data.repository.ColorPickerRepositoryImpl +import com.android.customization.picker.color.domain.interactor.ColorPickerInteractor +import com.android.customization.picker.color.domain.interactor.ColorPickerSnapshotRestorer +import com.android.customization.picker.color.ui.viewmodel.ColorPickerViewModel +import com.android.customization.picker.quickaffordance.data.repository.KeyguardQuickAffordancePickerRepository +import com.android.customization.picker.quickaffordance.domain.interactor.KeyguardQuickAffordancePickerInteractor +import com.android.customization.picker.quickaffordance.domain.interactor.KeyguardQuickAffordanceSnapshotRestorer +import com.android.systemui.shared.clocks.ClockRegistry +import com.android.systemui.shared.customization.data.content.CustomizationProviderClient +import com.android.systemui.shared.customization.data.content.CustomizationProviderClientImpl +import com.android.wallpaper.config.BaseFlags +import com.android.wallpaper.model.WallpaperColorsViewModel +import com.android.wallpaper.module.DrawableLayerResolver +import com.android.wallpaper.module.PackageStatusNotifier +import com.android.wallpaper.module.UserEventLogger +import com.android.wallpaper.picker.customization.data.content.WallpaperClientImpl +import com.android.wallpaper.picker.customization.data.repository.WallpaperRepository +import com.android.wallpaper.picker.customization.domain.interactor.WallpaperInteractor +import com.android.wallpaper.picker.undo.domain.interactor.SnapshotRestorer +import com.android.wallpaper.testing.TestInjector +import kotlinx.coroutines.Dispatchers +import kotlinx.coroutines.GlobalScope + +/** Test implementation of the dependency injector. */ +class TestCustomizationInjector : TestInjector(), CustomizationInjector { + private var customizationPreferences: CustomizationPreferences? = null + private var themeManager: ThemeManager? = null + private var packageStatusNotifier: PackageStatusNotifier? = null + private var drawableLayerResolver: DrawableLayerResolver? = null + private var userEventLogger: UserEventLogger? = null + private var wallpaperInteractor: WallpaperInteractor? = null + private var keyguardQuickAffordancePickerInteractor: KeyguardQuickAffordancePickerInteractor? = + null + private var flags: BaseFlags? = null + private var customizationProviderClient: CustomizationProviderClient? = null + private var keyguardQuickAffordanceSnapshotRestorer: KeyguardQuickAffordanceSnapshotRestorer? = + null + private var clockRegistry: ClockRegistry? = null + private var clockPickerInteractor: ClockPickerInteractor? = null + private var clockSectionViewModel: ClockSectionViewModel? = null + private var clockViewFactory: ClockViewFactory? = null + private var colorPickerRepository: ColorPickerRepository? = null + private var colorPickerInteractor: ColorPickerInteractor? = null + private var colorPickerViewModelFactory: ColorPickerViewModel.Factory? = null + private var colorPickerSnapshotRestorer: ColorPickerSnapshotRestorer? = null + private var colorCustomizationManager: ColorCustomizationManager? = null + private var clockCarouselViewModel: ClockCarouselViewModel? = null + private var clockSettingsViewModelFactory: ClockSettingsViewModel.Factory? = null + + override fun getCustomizationPreferences(context: Context): CustomizationPreferences { + return customizationPreferences + ?: TestDefaultCustomizationPreferences(context).also { customizationPreferences = it } + } + + override fun getThemeManager( + provider: ThemeBundleProvider, + activity: FragmentActivity, + overlayManagerCompat: OverlayManagerCompat, + logger: ThemesUserEventLogger + ): ThemeManager { + return themeManager + ?: TestThemeManager(provider, activity, overlayManagerCompat, logger).also { + themeManager = it + } + } + + override fun getPackageStatusNotifier(context: Context): PackageStatusNotifier { + return packageStatusNotifier + ?: TestPackageStatusNotifier().also { + packageStatusNotifier = TestPackageStatusNotifier() + } + } + + override fun getDrawableLayerResolver(): DrawableLayerResolver { + return drawableLayerResolver + ?: TestDrawableLayerResolver().also { drawableLayerResolver = it } + } + + override fun getUserEventLogger(context: Context): UserEventLogger { + return userEventLogger ?: TestThemesUserEventLogger().also { userEventLogger = it } + } + + override fun getWallpaperInteractor(context: Context): WallpaperInteractor { + return wallpaperInteractor + ?: WallpaperInteractor( + repository = + WallpaperRepository( + scope = GlobalScope, + client = WallpaperClientImpl(context = context), + backgroundDispatcher = Dispatchers.IO, + ), + shouldHandleReload = { + TextUtils.equals( + getColorCustomizationManager(context).currentColorSource, + ColorOptionsProvider.COLOR_SOURCE_PRESET + ) + } + ) + .also { wallpaperInteractor = it } + } + + override fun getKeyguardQuickAffordancePickerInteractor( + context: Context + ): KeyguardQuickAffordancePickerInteractor { + return keyguardQuickAffordancePickerInteractor + ?: createCustomizationProviderClient(context).also { + keyguardQuickAffordancePickerInteractor = it + } + } + + private fun createCustomizationProviderClient( + context: Context + ): KeyguardQuickAffordancePickerInteractor { + val client: CustomizationProviderClient = + CustomizationProviderClientImpl(context, Dispatchers.IO) + return KeyguardQuickAffordancePickerInteractor( + KeyguardQuickAffordancePickerRepository(client, Dispatchers.IO), + client + ) { + getKeyguardQuickAffordanceSnapshotRestorer(context) + } + } + + override fun getFlags(): BaseFlags { + return flags ?: object : BaseFlags() {}.also { flags = it } + } + + override fun getSnapshotRestorers(context: Context): Map<Int, SnapshotRestorer> { + val restorers: MutableMap<Int, SnapshotRestorer> = HashMap() + restorers[KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER] = + getKeyguardQuickAffordanceSnapshotRestorer(context) + restorers[KEY_COLOR_PICKER_SNAPSHOT_RESTORER] = + getColorPickerSnapshotRestorer(context, getWallpaperColorsViewModel()) + return restorers + } + + /** Returns the [CustomizationProviderClient]. */ + private fun getKeyguardQuickAffordancePickerProviderClient( + context: Context + ): CustomizationProviderClient { + return customizationProviderClient + ?: CustomizationProviderClientImpl(context, Dispatchers.IO).also { + customizationProviderClient = it + } + } + + private fun getKeyguardQuickAffordanceSnapshotRestorer( + context: Context + ): KeyguardQuickAffordanceSnapshotRestorer { + return keyguardQuickAffordanceSnapshotRestorer + ?: KeyguardQuickAffordanceSnapshotRestorer( + getKeyguardQuickAffordancePickerInteractor(context), + getKeyguardQuickAffordancePickerProviderClient(context) + ) + .also { keyguardQuickAffordanceSnapshotRestorer = it } + } + + override fun getClockRegistry(context: Context): ClockRegistry { + return clockRegistry + ?: ClockRegistryProvider(context, GlobalScope, Dispatchers.Main, Dispatchers.IO) + .get() + .also { clockRegistry = it } + } + + override fun getClockPickerInteractor(context: Context): ClockPickerInteractor { + return clockPickerInteractor + ?: ClockPickerInteractor(FakeClockPickerRepository()).also { + clockPickerInteractor = it + } + } + + override fun getClockSectionViewModel(context: Context): ClockSectionViewModel { + return clockSectionViewModel + ?: ClockSectionViewModel(context, getClockPickerInteractor(context)).also { + clockSectionViewModel = it + } + } + + private fun getColorPickerRepository( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ColorPickerRepository { + return colorPickerRepository + ?: ColorPickerRepositoryImpl( + wallpaperColorsViewModel, + getColorCustomizationManager(context) + ) + } + + override fun getColorPickerInteractor( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ColorPickerInteractor { + return colorPickerInteractor + ?: ColorPickerInteractor( + repository = getColorPickerRepository(context, wallpaperColorsViewModel), + snapshotRestorer = { + getColorPickerSnapshotRestorer(context, wallpaperColorsViewModel) + }, + ) + .also { colorPickerInteractor = it } + } + + override fun getColorPickerViewModelFactory( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ColorPickerViewModel.Factory { + return colorPickerViewModelFactory + ?: ColorPickerViewModel.Factory( + context, + getColorPickerInteractor(context, wallpaperColorsViewModel), + ) + .also { colorPickerViewModelFactory = it } + } + + private fun getColorPickerSnapshotRestorer( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel + ): ColorPickerSnapshotRestorer { + return colorPickerSnapshotRestorer + ?: ColorPickerSnapshotRestorer( + getColorPickerInteractor(context, wallpaperColorsViewModel) + ) + .also { colorPickerSnapshotRestorer = it } + } + + private fun getColorCustomizationManager(context: Context): ColorCustomizationManager { + return colorCustomizationManager + ?: ColorCustomizationManager.getInstance(context, OverlayManagerCompat(context)).also { + colorCustomizationManager = it + } + } + + override fun getClockCarouselViewModel(context: Context): ClockCarouselViewModel { + return clockCarouselViewModel + ?: ClockCarouselViewModel(getClockPickerInteractor(context)).also { + clockCarouselViewModel = it + } + } + + override fun getClockViewFactory(activity: Activity): ClockViewFactory { + return clockViewFactory + ?: ClockViewFactory(activity, getClockRegistry(activity)).also { clockViewFactory = it } + } + + override fun getClockSettingsViewModelFactory( + context: Context, + wallpaperColorsViewModel: WallpaperColorsViewModel, + ): ClockSettingsViewModel.Factory { + return clockSettingsViewModelFactory + ?: ClockSettingsViewModel.Factory( + context, + getClockPickerInteractor(context), + getColorPickerInteractor( + context, + wallpaperColorsViewModel, + ), + ) + .also { clockSettingsViewModelFactory = it } + } + + companion object { + private const val KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER = 1 + private const val KEY_COLOR_PICKER_SNAPSHOT_RESTORER = + KEY_QUICK_AFFORDANCE_SNAPSHOT_RESTORER + 1 + } +} diff --git a/tests/src/com/android/customization/testing/TestPluginManager.kt b/tests/src/com/android/customization/testing/TestPluginManager.kt new file mode 100644 index 00000000..167d8ddc --- /dev/null +++ b/tests/src/com/android/customization/testing/TestPluginManager.kt @@ -0,0 +1,36 @@ +package com.android.customization.testing + +import com.android.systemui.plugins.Plugin +import com.android.systemui.plugins.PluginListener +import com.android.systemui.plugins.PluginManager + +class TestPluginManager : PluginManager { + override fun getPrivilegedPlugins(): Array<String> { + return emptyArray() + } + + override fun <T : Plugin?> addPluginListener(listener: PluginListener<T>, cls: Class<T>) {} + override fun <T : Plugin?> addPluginListener( + listener: PluginListener<T>, + cls: Class<T>, + allowMultiple: Boolean + ) {} + + override fun <T : Plugin?> addPluginListener( + action: String, + listener: PluginListener<T>, + cls: Class<T> + ) {} + + override fun <T : Plugin?> addPluginListener( + action: String, + listener: PluginListener<T>, + cls: Class<T>, + allowMultiple: Boolean + ) {} + + override fun removePluginListener(listener: PluginListener<*>?) {} + override fun <T> dependsOn(p: Plugin, cls: Class<T>): Boolean { + return false + } +} |