Parcelize CommunalWidgetContentModel (11a57e35) · Commits · e / os / android_frameworks_base

packages/SystemUI/multivalentTests/src/com/android/systemui/communal/shared/model/CommunalWidgetContentModelTest.kt

0 → 100644

+81 −0

Original line number	Diff line number	Diff line
		/*
		* Copyright (C) 2023 The Android Open Source Project
		*
		* Licensed under the Apache License, Version 2.0 (the "License");
		* you may not use this file except in compliance with the License.
		* You may obtain a copy of the License at
		*
		* http://www.apache.org/licenses/LICENSE-2.0
		*
		* Unless required by applicable law or agreed to in writing, software
		* distributed under the License is distributed on an "AS IS" BASIS,
		* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
		* See the License for the specific language governing permissions and
		* limitations under the License.
		*/

		package com.android.systemui.communal.shared.model

		import android.appwidget.AppWidgetProviderInfo
		import android.content.ComponentName
		import android.os.Parcel
		import android.os.UserHandle
		import androidx.test.ext.junit.runners.AndroidJUnit4
		import androidx.test.filters.SmallTest
		import com.android.systemui.SysuiTestCase
		import com.google.common.truth.Truth.assertThat
		import org.junit.Test
		import org.junit.runner.RunWith

		@SmallTest
		@RunWith(AndroidJUnit4::class)
		class CommunalWidgetContentModelTest : SysuiTestCase() {
		@Test
		fun testParcelizeAvailableWidget() {
		val widgetToParcelize =
		CommunalWidgetContentModel.Available(
		appWidgetId = 1,
		providerInfo =
		AppWidgetProviderInfo().apply { provider = ComponentName("pkg", "cls") },
		rank = 2,
		spanY = 3,
		)

		val parcel = Parcel.obtain()
		widgetToParcelize.writeToParcel(parcel, flags = 0)

		parcel.setDataPosition(0)

		// Only checking fields are equal and not complete equality because not all fields are
		// specified in the fake AppWidgetProviderInfo
		val widgetFromParcel =
		CommunalWidgetContentModel.createFromParcel(parcel)
		as CommunalWidgetContentModel.Available
		assertThat(widgetFromParcel.appWidgetId).isEqualTo(widgetToParcelize.appWidgetId)
		assertThat(widgetFromParcel.rank).isEqualTo(widgetToParcelize.rank)
		assertThat(widgetFromParcel.spanY).isEqualTo(widgetToParcelize.spanY)
		assertThat(widgetFromParcel.providerInfo.provider)
		.isEqualTo(widgetToParcelize.providerInfo.provider)
		}

		@Test
		fun testParcelizePendingWidget() {
		val widgetToParcelize =
		CommunalWidgetContentModel.Pending(
		appWidgetId = 2,
		rank = 3,
		componentName = ComponentName("pkg", "cls"),
		icon = null,
		user = UserHandle(0),
		spanY = 6,
		)

		val parcel = Parcel.obtain()
		widgetToParcelize.writeToParcel(parcel, flags = 0)

		parcel.setDataPosition(0)

		val widgetFromParcel = CommunalWidgetContentModel.createFromParcel(parcel)
		assertThat(widgetFromParcel).isEqualTo(widgetToParcelize)
		}
		}

packages/SystemUI/src/com/android/systemui/communal/shared/model/CommunalWidgetContentModel.kt

+98 −3

Original line number	Diff line number	Diff line
		@@ -19,21 +19,60 @@ package com.android.systemui.communal.shared.model
		import android.appwidget.AppWidgetProviderInfo
		import android.content.ComponentName
		import android.graphics.Bitmap
		import android.os.Parcel
		import android.os.Parcelable
		import android.os.UserHandle

		/** Encapsulates data for a communal widget. */
		sealed interface CommunalWidgetContentModel {
		sealed interface CommunalWidgetContentModel : Parcelable {
		val appWidgetId: Int
		val rank: Int
		val spanY: Int

		// Used for distinguishing subtypes when reading from a parcel.
		val type: Int

		/** Widget is ready to display */
		data class Available(
		override val appWidgetId: Int,
		val providerInfo: AppWidgetProviderInfo,
		override val rank: Int,
		override val spanY: Int,
		) : CommunalWidgetContentModel
		) : CommunalWidgetContentModel {

		override val type = TYPE_AVAILABLE

		constructor(
		parcel: Parcel
		) : this(
		parcel.readInt(),
		requireNotNull(parcel.readTypedObject(AppWidgetProviderInfo.CREATOR)),
		parcel.readInt(),
		parcel.readInt(),
		)

		override fun writeToParcel(parcel: Parcel, flags: Int) {
		parcel.writeInt(type)
		parcel.writeInt(appWidgetId)
		parcel.writeTypedObject(providerInfo, flags)
		parcel.writeInt(rank)
		parcel.writeInt(spanY)
		}

		override fun describeContents(): Int {
		return 0
		}

		companion object CREATOR : Parcelable.Creator<Available> {
		override fun createFromParcel(parcel: Parcel): Available {
		return Available(parcel)
		}

		override fun newArray(size: Int): Array<Available?> {
		return arrayOfNulls(size)
		}
		}
		}

		/** Widget is pending installation */
		data class Pending(
		@@ -43,5 +82,61 @@ sealed interface CommunalWidgetContentModel {
		val icon: Bitmap?,
		val user: UserHandle,
		override val spanY: Int,
		) : CommunalWidgetContentModel
		) : CommunalWidgetContentModel {

		override val type = TYPE_PENDING

		constructor(
		parcel: Parcel
		) : this(
		parcel.readInt(),
		parcel.readInt(),
		requireNotNull(parcel.readTypedObject(ComponentName.CREATOR)),
		parcel.readTypedObject(Bitmap.CREATOR),
		requireNotNull(parcel.readTypedObject(UserHandle.CREATOR)),
		parcel.readInt(),
		)

		override fun writeToParcel(parcel: Parcel, flags: Int) {
		parcel.writeInt(type)
		parcel.writeInt(appWidgetId)
		parcel.writeInt(rank)
		parcel.writeTypedObject(componentName, flags)
		parcel.writeTypedObject(icon, flags)
		parcel.writeTypedObject(user, flags)
		parcel.writeInt(spanY)
		}

		override fun describeContents(): Int {
		return 0
		}

		companion object CREATOR : Parcelable.Creator<Pending> {
		override fun createFromParcel(parcel: Parcel): Pending {
		return Pending(parcel)
		}

		override fun newArray(size: Int): Array<Pending?> {
		return arrayOfNulls(size)
		}
		}
		}

		// Used for distinguishing subtypes when reading from a parcel.
		companion object CREATOR : Parcelable.Creator<CommunalWidgetContentModel> {
		private const val TYPE_AVAILABLE = 0
		private const val TYPE_PENDING = 1

		override fun createFromParcel(parcel: Parcel): CommunalWidgetContentModel {
		return when (val type = parcel.readInt()) {
		TYPE_AVAILABLE -> Available(parcel)
		TYPE_PENDING -> Pending(parcel)
		else -> throw IllegalArgumentException("Unknown type: $type")
		}
		}

		override fun newArray(size: Int): Array<CommunalWidgetContentModel?> {
		return arrayOfNulls(size)
		}
		}
		}