Sunny Goyal
57a195b790
BgDataModel holds a mutable handle of the data, which ensures that all edits go through this class and there is a single place to keep the repository updated Bug: 390572144 Test: Updated existing model tests Flag: EXEMPT refactor Change-Id: I991b707dd21801d1abd00510f676549947285012
142 lines
6.3 KiB
Kotlin
142 lines
6.3 KiB
Kotlin
package com.android.launcher3.util
|
|
|
|
import android.content.ContentValues
|
|
import android.database.sqlite.SQLiteDatabase
|
|
import android.os.Process
|
|
import android.util.SparseArray
|
|
import androidx.test.platform.app.InstrumentationRegistry
|
|
import com.android.launcher3.Flags
|
|
import com.android.launcher3.LauncherModel
|
|
import com.android.launcher3.LauncherSettings.Favorites.APPWIDGET_ID
|
|
import com.android.launcher3.LauncherSettings.Favorites.APPWIDGET_PROVIDER
|
|
import com.android.launcher3.LauncherSettings.Favorites.APPWIDGET_SOURCE
|
|
import com.android.launcher3.LauncherSettings.Favorites.CELLX
|
|
import com.android.launcher3.LauncherSettings.Favorites.CELLY
|
|
import com.android.launcher3.LauncherSettings.Favorites.CONTAINER
|
|
import com.android.launcher3.LauncherSettings.Favorites.CONTAINER_DESKTOP
|
|
import com.android.launcher3.LauncherSettings.Favorites.CONTAINER_HOTSEAT
|
|
import com.android.launcher3.LauncherSettings.Favorites.INTENT
|
|
import com.android.launcher3.LauncherSettings.Favorites.ITEM_TYPE
|
|
import com.android.launcher3.LauncherSettings.Favorites.ITEM_TYPE_APPLICATION
|
|
import com.android.launcher3.LauncherSettings.Favorites.PROFILE_ID
|
|
import com.android.launcher3.LauncherSettings.Favorites.RESTORED
|
|
import com.android.launcher3.LauncherSettings.Favorites.SCREEN
|
|
import com.android.launcher3.LauncherSettings.Favorites.SPANX
|
|
import com.android.launcher3.LauncherSettings.Favorites.SPANY
|
|
import com.android.launcher3.LauncherSettings.Favorites.TITLE
|
|
import com.android.launcher3.LauncherSettings.Favorites._ID
|
|
import com.android.launcher3.model.BgDataModel
|
|
import com.android.launcher3.model.ModelDbController
|
|
import com.android.launcher3.model.data.ItemInfo
|
|
import com.android.launcher3.util.Executors.MODEL_EXECUTOR
|
|
import java.io.BufferedReader
|
|
import java.io.InputStreamReader
|
|
|
|
object ModelTestExtensions {
|
|
/** Clears and reloads Launcher db to cleanup the workspace */
|
|
fun LauncherModel.clearModelDb() {
|
|
// Load the model once so that there is no pending migration:
|
|
loadModelSync()
|
|
TestUtil.runOnExecutorSync(Executors.MODEL_EXECUTOR) {
|
|
modelDbController.run {
|
|
if (Flags.gridMigrationRefactor())
|
|
attemptMigrateDb(null /* restoreEventLogger */, modelDelegate)
|
|
else tryMigrateDB(null /* restoreEventLogger */, modelDelegate)
|
|
createEmptyDB()
|
|
clearEmptyDbFlag()
|
|
}
|
|
}
|
|
// Reload model
|
|
TestUtil.runOnExecutorSync(Executors.MAIN_EXECUTOR) { forceReload() }
|
|
loadModelSync()
|
|
}
|
|
|
|
/** Loads the model in memory synchronously */
|
|
fun LauncherModel.loadModelSync() {
|
|
val mockCb: BgDataModel.Callbacks = object : BgDataModel.Callbacks {}
|
|
TestUtil.runOnExecutorSync(Executors.MAIN_EXECUTOR) { addCallbacksAndLoad(mockCb) }
|
|
TestUtil.runOnExecutorSync(Executors.MODEL_EXECUTOR) {}
|
|
TestUtil.runOnExecutorSync(Executors.MAIN_EXECUTOR) {}
|
|
TestUtil.runOnExecutorSync(Executors.MAIN_EXECUTOR) { removeCallbacks(mockCb) }
|
|
}
|
|
|
|
/** Adds and commits a new item to Launcher.db */
|
|
fun LauncherModel.addItem(
|
|
title: String = "LauncherTestApp",
|
|
intent: String =
|
|
"#Intent;action=android.intent.action.MAIN;category=android.intent.category.LAUNCHER;component=com.google.android.apps.nexuslauncher.tests/com.android.launcher3.testcomponent.BaseTestingActivity;launchFlags=0x10200000;end",
|
|
type: Int = ITEM_TYPE_APPLICATION,
|
|
restoreFlags: Int = 0,
|
|
screen: Int = 0,
|
|
container: Int = CONTAINER_DESKTOP,
|
|
x: Int,
|
|
y: Int,
|
|
spanX: Int = 1,
|
|
spanY: Int = 1,
|
|
id: Int = 0,
|
|
profileId: Int = Process.myUserHandle().identifier,
|
|
appWidgetId: Int = -1,
|
|
appWidgetSource: Int = -1,
|
|
appWidgetProvider: String? = null,
|
|
) {
|
|
loadModelSync()
|
|
TestUtil.runOnExecutorSync(Executors.MODEL_EXECUTOR) {
|
|
val controller: ModelDbController = modelDbController
|
|
controller.attemptMigrateDb(null /* restoreEventLogger */, modelDelegate)
|
|
modelDbController.newTransaction().use { transaction ->
|
|
val values =
|
|
ContentValues().apply {
|
|
values[_ID] = id
|
|
values[TITLE] = title
|
|
values[PROFILE_ID] = profileId
|
|
values[CONTAINER] = container
|
|
values[SCREEN] = screen
|
|
values[CELLX] = x
|
|
values[CELLY] = y
|
|
values[SPANX] = spanX
|
|
values[SPANY] = spanY
|
|
values[ITEM_TYPE] = type
|
|
values[RESTORED] = restoreFlags
|
|
values[INTENT] = intent
|
|
values[APPWIDGET_ID] = appWidgetId
|
|
values[APPWIDGET_SOURCE] = appWidgetSource
|
|
values[APPWIDGET_PROVIDER] = appWidgetProvider
|
|
}
|
|
// Migrate any previous data so that the DB state is correct
|
|
controller.insert(values)
|
|
transaction.commit()
|
|
}
|
|
}
|
|
}
|
|
|
|
@JvmStatic
|
|
val LauncherModel.bgDataModel: BgDataModel
|
|
get() {
|
|
var data: BgDataModel? = null
|
|
enqueueModelUpdateTask { _, dataModel, _ -> data = dataModel }
|
|
TestUtil.runOnExecutorSync(MODEL_EXECUTOR) {}
|
|
return data!!
|
|
}
|
|
|
|
/** Total number of items belonging to a non-predicted container */
|
|
@JvmStatic
|
|
fun Iterable<ItemInfo>.nonPredictedItemCount() = count { it.container >= CONTAINER_HOTSEAT }
|
|
|
|
/** Creates an in-memory sqlite DB and initializes with the data in [insertFile] */
|
|
fun createInMemoryDb(insertFile: String): SQLiteDatabase =
|
|
SQLiteDatabase.createInMemory(SQLiteDatabase.OpenParams.Builder().build()).also { db ->
|
|
BufferedReader(
|
|
InputStreamReader(
|
|
InstrumentationRegistry.getInstrumentation().context.assets.open(insertFile)
|
|
)
|
|
)
|
|
.lines()
|
|
.forEach { sqlStatement -> db.execSQL(sqlStatement) }
|
|
}
|
|
|
|
/** Initializes [BgDataModel.itemsIdMap] with provided [items] */
|
|
fun BgDataModel.initItems(vararg items: ItemInfo) {
|
|
dataLoadComplete(SparseArray<ItemInfo>().apply { items.forEach { this[it.id] = it } })
|
|
}
|
|
}
|