Donate to e Foundation | Murena handsets with /e/OS | Own a part of Murena! Learn more

Commit 0ab18584 authored by Xiaowen Lei's avatar Xiaowen Lei Committed by Automerger Merge Worker
Browse files

Merge "Add test that ensures setIsDreaming(true) is called by the Smartspace... 

Merge "Add test that ensures setIsDreaming(true) is called by the Smartspace on Dream." into tm-qpr-dev am: 9aa858ed

Original change: https://googleplex-android-review.googlesource.com/c/platform/frameworks/base/+/18151731



Change-Id: I6bb0180cb4d1c912d0f38cc6658fddbea1be2865
Signed-off-by: default avatarAutomerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
parents 18518185 9aa858ed

packages/SystemUI/tests/src/com/android/systemui/smartspace/DreamSmartspaceControllerTest.kt

+22 −13
Original line number Diff line number Diff line
@@ -28,6 +28,7 @@ import androidx.test.filters.SmallTest 
import com.android.systemui.SysuiTestCase

import com.android.systemui.dreams.smartspace.DreamSmartspaceController

import com.android.systemui.plugins.BcSmartspaceDataPlugin

import com.android.systemui.plugins.BcSmartspaceDataPlugin.SmartspaceView

import com.android.systemui.plugins.FalsingManager

import com.android.systemui.smartspace.dagger.SmartspaceViewComponent

import com.android.systemui.util.concurrency.Execution
@@ -39,6 +40,7 @@ import org.junit.Before 
import org.junit.Test

import org.junit.runner.RunWith

import org.mockito.Mock

import org.mockito.Spy

import org.mockito.Mockito

import org.mockito.Mockito.`when`

import org.mockito.Mockito.verify
@@ -74,8 +76,8 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 
    @Mock

    private lateinit var precondition: SmartspacePrecondition



    @Mock

    private lateinit var smartspaceView: BcSmartspaceDataPlugin.SmartspaceView

    @Spy

    private var smartspaceView: SmartspaceView = TestView(context)



    @Mock

    private lateinit var listener: BcSmartspaceDataPlugin.SmartspaceTargetListener
@@ -83,6 +85,8 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 
    @Mock

    private lateinit var session: SmartspaceSession



    private lateinit var controller: DreamSmartspaceController



    @Before

    fun setup() {

        MockitoAnnotations.initMocks(this)
@@ -90,6 +94,9 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 
                .thenReturn(viewComponent)

        `when`(viewComponent.getView()).thenReturn(smartspaceView)

        `when`(smartspaceManager.createSmartspaceSession(any())).thenReturn(session)



        controller = DreamSmartspaceController(context, smartspaceManager, execution, uiExecutor,

                viewComponentFactory, precondition, Optional.of(targetFilter), Optional.of(plugin))

    }



    /**
@@ -97,10 +104,6 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 
     */

    @Test

    fun testConnectOnListen() {

        val controller = DreamSmartspaceController(context,

        smartspaceManager, execution, uiExecutor, viewComponentFactory, precondition,

                Optional.of(targetFilter), Optional.of(plugin))



        `when`(precondition.conditionsMet()).thenReturn(true)

        controller.addListener(listener)
@@ -130,8 +133,7 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 
     * A class which implements SmartspaceView and extends View. This is mocked to provide the right

     * object inheritance and interface implementation used in DreamSmartspaceController

     */

    private class TestView(context: Context?) : View(context),

            BcSmartspaceDataPlugin.SmartspaceView {

    private class TestView(context: Context?) : View(context), SmartspaceView {

        override fun registerDataProvider(plugin: BcSmartspaceDataPlugin?) {}



        override fun setPrimaryTextColor(color: Int) {}
@@ -158,11 +160,6 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 
     */

    @Test

    fun testConnectOnViewCreate() {

        val controller = DreamSmartspaceController(context,

                smartspaceManager, execution, uiExecutor, viewComponentFactory, precondition,

                Optional.of(targetFilter),

                Optional.of(plugin))



        `when`(precondition.conditionsMet()).thenReturn(true)

        controller.buildAndConnectView(Mockito.mock(ViewGroup::class.java))
@@ -181,4 +178,16 @@ class DreamSmartspaceControllerTest : SysuiTestCase() { 


        verify(session).close()

    }



    /**

     * Ensures setIsDreaming(true) is called when the view is built.

     */

    @Test

    fun testSetIsDreamingTrueOnViewCreate() {

        `when`(precondition.conditionsMet()).thenReturn(true)



        controller.buildAndConnectView(Mockito.mock(ViewGroup::class.java))



        verify(smartspaceView).setIsDreaming(true)

    }

}
 
 No newline at end of file