Merge "Update the VolumePanelViewModelTest with the new test APIs" into main (014118ee) · Commits · e / os / android_frameworks_base

packages/SystemUI/multivalentTests/src/com/android/systemui/volume/panel/ui/viewmodel/VolumePanelViewModelTest.kt

+65 −93

Original line number	Diff line number	Diff line
		@@ -19,6 +19,8 @@ package com.android.systemui.volume.panel.ui.viewmodel
		import android.content.Intent
		import android.content.applicationContext
		import android.content.res.Configuration
		import android.content.testableContext
		import android.testing.TestableResources
		import androidx.test.ext.junit.runners.AndroidJUnit4
		import androidx.test.filters.SmallTest
		import com.android.systemui.SysuiTestCase
		@@ -27,11 +29,12 @@ import com.android.systemui.coroutines.collectLastValue
		import com.android.systemui.dump.DumpManager
		import com.android.systemui.kosmos.Kosmos
		import com.android.systemui.kosmos.applicationCoroutineScope
		import com.android.systemui.kosmos.testScope
		import com.android.systemui.kosmos.collectLastValue
		import com.android.systemui.kosmos.runTest
		import com.android.systemui.res.R
		import com.android.systemui.statusbar.policy.configurationController
		import com.android.systemui.statusbar.policy.fakeConfigurationController
		import com.android.systemui.testKosmos
		import com.android.systemui.testKosmosNew
		import com.android.systemui.volume.panel.dagger.factory.volumePanelComponentFactory
		import com.android.systemui.volume.panel.data.repository.volumePanelGlobalStateRepository
		import com.android.systemui.volume.panel.domain.interactor.criteriaByKey
		@@ -47,8 +50,7 @@ import com.google.common.truth.Truth.assertThat
		import java.io.PrintWriter
		import java.io.StringWriter
		import javax.inject.Provider
		import kotlinx.coroutines.launch
		import kotlinx.coroutines.test.runCurrent
		import kotlinx.coroutines.cancel
		import kotlinx.coroutines.test.runTest
		import org.junit.Test
		import org.junit.runner.RunWith
		@@ -58,29 +60,41 @@ import org.junit.runner.RunWith
		class VolumePanelViewModelTest : SysuiTestCase() {

		private val kosmos =
		testKosmos().apply {
		testKosmosNew().apply {
		componentsLayoutManager = DefaultComponentsLayoutManager(BOTTOM_BAR)
		volumePanelGlobalStateRepository.updateVolumePanelState { it.copy(isVisible = true) }
		}

		private val realDumpManager = DumpManager()
		private val testableResources = context.orCreateTestableResources
		private val Kosmos.testableResources: TestableResources
		get() = testableContext.orCreateTestableResources

		private lateinit var underTest: VolumePanelViewModel
		private val underTest: VolumePanelViewModel by lazy {
		with(kosmos) {
		VolumePanelViewModel(
		testableContext.orCreateTestableResources.resources,
		applicationCoroutineScope,
		volumePanelComponentFactory,
		configurationController,
		broadcastDispatcher,
		realDumpManager,
		volumePanelLogger,
		volumePanelGlobalStateInteractor,
		)
		}
		}

		@Test
		fun dismissingPanel_changesVisibility() = test {
		testScope.runTest {
		fun dismissingPanel_changesVisibility() =
		kosmos.runTest {
		underTest.dismissPanel()
		runCurrent()

		assertThat(volumePanelGlobalStateRepository.globalState.value.isVisible).isFalse()
		}
		}

		@Test
		fun orientationChanges_panelOrientationChanges() = test {
		testScope.runTest {
		fun orientationChanges_panelOrientationChanges() =
		kosmos.runTest {
		val volumePanelState by collectLastValue(underTest.volumePanelState)
		testableResources.overrideConfiguration(
		Configuration().apply { orientation = Configuration.ORIENTATION_PORTRAIT }
		@@ -90,16 +104,14 @@ class VolumePanelViewModelTest : SysuiTestCase() {
		fakeConfigurationController.onConfigurationChanged(
		Configuration().apply { orientation = Configuration.ORIENTATION_LANDSCAPE }
		)
		runCurrent()

		assertThat(volumePanelState!!.orientation)
		.isEqualTo(Configuration.ORIENTATION_LANDSCAPE)
		}
		}

		@Test
		fun components_areReturned() =
		test({
		kosmos.runTest {
		componentByKey =
		mapOf(
		COMPONENT_1 to mockVolumePanelUiComponentProvider,
		@@ -107,10 +119,8 @@ class VolumePanelViewModelTest : SysuiTestCase() {
		BOTTOM_BAR to mockVolumePanelUiComponentProvider,
		)
		criteriaByKey = mapOf(COMPONENT_2 to Provider { unavailableCriteria })
		}) {
		testScope.runTest {

		val componentsLayout by collectLastValue(underTest.componentsLayout)
		runCurrent()

		assertThat(componentsLayout!!.contentComponents).hasSize(2)
		assertThat(componentsLayout!!.contentComponents[0].key).isEqualTo(COMPONENT_1)
		@@ -120,42 +130,30 @@ class VolumePanelViewModelTest : SysuiTestCase() {
		assertThat(componentsLayout!!.bottomBarComponent.key).isEqualTo(BOTTOM_BAR)
		assertThat(componentsLayout!!.bottomBarComponent.isVisible).isTrue()
		}
		}

		@Test
		fun dismissPanel_dismissesPanel() = test {
		testScope.runTest {
		fun dismissPanel_dismissesPanel() =
		kosmos.runTest {
		underTest.dismissPanel()
		runCurrent()

		assertThat(volumePanelGlobalStateRepository.globalState.value.isVisible).isFalse()
		}
		}

		@Test
		fun testDumpableRegister_unregister() =
		with(kosmos) {
		testScope.runTest {
		val job = launch {
		applicationCoroutineScope = this
		underTest = createViewModel()
		kosmos.runTest {
		underTest // access the field to instantiate it

		runCurrent()

		assertThat(realDumpManager.getDumpables().any { it.name == DUMPABLE_NAME })
		.isTrue()
		}
		assertThat(realDumpManager.getDumpables().any { it.name == DUMPABLE_NAME }).isTrue()

		runCurrent()
		job.cancel()
		applicationCoroutineScope.cancel()

		assertThat(realDumpManager.getDumpables().any { it.name == DUMPABLE_NAME }).isTrue()
		}
		assertThat(realDumpManager.getDumpables().any { it.name == DUMPABLE_NAME }).isFalse()
		}

		@Test
		fun testDumpingState() =
		test({
		kosmos.runTest {
		testableResources.addOverride(R.bool.volume_panel_is_large_screen, false)
		testableResources.overrideConfiguration(
		Configuration().apply { orientation = Configuration.ORIENTATION_PORTRAIT }
		@@ -167,9 +165,6 @@ class VolumePanelViewModelTest : SysuiTestCase() {
		BOTTOM_BAR to mockVolumePanelUiComponentProvider,
		)
		criteriaByKey = mapOf(COMPONENT_2 to Provider { unavailableCriteria })
		}) {
		testScope.runTest {
		runCurrent()

		StringWriter().use {
		underTest.dump(PrintWriter(it), emptyArray())
		@@ -188,41 +183,18 @@ class VolumePanelViewModelTest : SysuiTestCase() {
		)
		}
		}
		}

		@Test
		fun dismissBroadcast_dismissesPanel() = test {
		testScope.runTest {
		runCurrent() // run the flows to let allow the receiver to be registered
		fun dismissBroadcast_dismissesPanel() =
		kosmos.runTest {
		underTest // instantiate the View Model
		broadcastDispatcher.sendIntentToMatchingReceiversOnly(
		applicationContext,
		Intent(DISMISS_ACTION),
		)
		runCurrent()

		assertThat(volumePanelGlobalStateRepository.globalState.value.isVisible).isFalse()
		}
		}

		private fun test(setup: Kosmos.() -> Unit = {}, test: Kosmos.() -> Unit) =
		with(kosmos) {
		setup()
		underTest = createViewModel()

		test()
		}

		private fun Kosmos.createViewModel(): VolumePanelViewModel =
		VolumePanelViewModel(
		context.orCreateTestableResources.resources,
		applicationCoroutineScope,
		volumePanelComponentFactory,
		configurationController,
		broadcastDispatcher,
		realDumpManager,
		volumePanelLogger,
		volumePanelGlobalStateInteractor,
		)

		private companion object {
		const val DUMPABLE_NAME = "VolumePanelViewModel"