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

Commit b25b66d9 authored by Liran Binyamin's avatar Liran Binyamin
Browse files

Initial unit tests for BubbleStackView 

Fixes: 321265700
Test: atest BubbleStackViewTest
Change-Id: I0832d53f39527c5d0b0b89e868192a3835b77d77
parent 915f5e23

libs/WindowManager/Shell/Android.bp

+2 −0
Original line number Diff line number Diff line
@@ -205,6 +205,8 @@ android_robolectric_test { 
    srcs: [

        "multivalentTests/src/**/*.kt",

    ],

    // TODO(b/323188766): Include BubbleStackViewTest once the robolectric issue is fixed.

    exclude_srcs: ["multivalentTests/src/com/android/wm/shell/bubbles/BubbleStackViewTest.kt"],

    static_libs: [

        "junit",

        "androidx.test.runner",

libs/WindowManager/Shell/multivalentTests/src/com/android/wm/shell/bubbles/BubbleStackViewTest.kt

0 → 100644
+215 −0
Original line number Diff line number Diff line 
/*

 * Copyright (C) 2024 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.wm.shell.bubbles



import android.content.Context

import android.content.Intent

import android.graphics.Color

import android.graphics.drawable.Icon

import android.os.UserHandle

import android.view.IWindowManager

import android.view.WindowManager

import android.view.WindowManagerGlobal

import androidx.test.annotation.UiThreadTest

import androidx.test.core.app.ApplicationProvider

import androidx.test.ext.junit.runners.AndroidJUnit4

import androidx.test.filters.SmallTest

import com.android.internal.logging.testing.UiEventLoggerFake

import com.android.internal.protolog.common.ProtoLog

import com.android.launcher3.icons.BubbleIconFactory

import com.android.wm.shell.R

import com.android.wm.shell.bubbles.Bubbles.SysuiProxy

import com.android.wm.shell.common.FloatingContentCoordinator

import com.android.wm.shell.common.ShellExecutor

import com.android.wm.shell.taskview.TaskView

import com.android.wm.shell.taskview.TaskViewTaskController

import com.google.common.truth.Truth.assertThat

import com.google.common.util.concurrent.MoreExecutors.directExecutor

import java.util.concurrent.Semaphore

import java.util.concurrent.TimeUnit

import java.util.function.Consumer

import org.junit.Before

import org.junit.Test

import org.junit.runner.RunWith

import org.mockito.kotlin.mock



/** Unit tests for [BubbleStackView]. */

@SmallTest

@RunWith(AndroidJUnit4::class)

class BubbleStackViewTest {



    private val context = ApplicationProvider.getApplicationContext<Context>()

    private lateinit var positioner: BubblePositioner

    private lateinit var iconFactory: BubbleIconFactory

    private lateinit var expandedViewManager: FakeBubbleExpandedViewManager

    private lateinit var bubbleStackView: BubbleStackView

    private lateinit var shellExecutor: ShellExecutor

    private lateinit var windowManager: IWindowManager

    private lateinit var bubbleTaskViewFactory: BubbleTaskViewFactory

    private lateinit var bubbleData: BubbleData



    @Before

    fun setUp() {

        // Disable protolog tool when running the tests from studio

        ProtoLog.REQUIRE_PROTOLOGTOOL = false

        windowManager = WindowManagerGlobal.getWindowManagerService()!!

        shellExecutor = TestShellExecutor()

        val windowManager = context.getSystemService(WindowManager::class.java)

        iconFactory =

            BubbleIconFactory(

                context,

                context.resources.getDimensionPixelSize(R.dimen.bubble_size),

                context.resources.getDimensionPixelSize(R.dimen.bubble_badge_size),

                Color.BLACK,

                context.resources.getDimensionPixelSize(

                    com.android.internal.R.dimen.importance_ring_stroke_width

                )

            )

        positioner = BubblePositioner(context, windowManager)

        val bubbleStackViewManager = FakeBubbleStackViewManager()

        bubbleData =

            BubbleData(

                context,

                BubbleLogger(UiEventLoggerFake()),

                positioner,

                BubbleEducationController(context),

                shellExecutor

            )



        val sysuiProxy = mock<SysuiProxy>()

        expandedViewManager = FakeBubbleExpandedViewManager()

        bubbleTaskViewFactory = FakeBubbleTaskViewFactory()

        bubbleStackView =

            BubbleStackView(

                context,

                bubbleStackViewManager,

                positioner,

                bubbleData,

                null,

                FloatingContentCoordinator(),

                { sysuiProxy },

                shellExecutor

            )

    }



    @UiThreadTest

    @Test

    fun addBubble() {

        val bubble = createAndInflateBubble()

        bubbleStackView.addBubble(bubble)

        assertThat(bubbleStackView.bubbleCount).isEqualTo(1)

    }



    @UiThreadTest

    @Test

    fun tapBubbleToExpand() {

        val bubble = createAndInflateBubble()

        bubbleStackView.addBubble(bubble)

        assertThat(bubbleStackView.bubbleCount).isEqualTo(1)



        bubble.iconView!!.performClick()

        // we're checking the expanded state in BubbleData because that's the source of truth. This

        // will eventually propagate an update back to the stack view, but setting the entire

        // pipeline is outside the scope of a unit test.

        assertThat(bubbleData.isExpanded).isTrue()

    }



    private fun createAndInflateBubble(): Bubble {

        val intent = Intent(Intent.ACTION_VIEW).setPackage(context.packageName)

        val icon = Icon.createWithResource(context.resources, R.drawable.bubble_ic_overflow_button)

        val bubble = Bubble.createAppBubble(intent, UserHandle(1), icon, directExecutor())

        bubble.setInflateSynchronously(true)

        bubbleData.notificationEntryUpdated(bubble, true, false)



        val semaphore = Semaphore(0)

        val callback: BubbleViewInfoTask.Callback =

            BubbleViewInfoTask.Callback { semaphore.release() }

        bubble.inflate(

            callback,

            context,

            expandedViewManager,

            bubbleTaskViewFactory,

            positioner,

            bubbleStackView,

            null,

            iconFactory,

            false

        )



        assertThat(semaphore.tryAcquire(5, TimeUnit.SECONDS)).isTrue()

        assertThat(bubble.isInflated).isTrue()

        return bubble

    }



    private class FakeBubbleStackViewManager : BubbleStackViewManager {



        override fun onAllBubblesAnimatedOut() {}



        override fun updateWindowFlagsForBackpress(interceptBack: Boolean) {}



        override fun checkNotificationPanelExpandedState(callback: Consumer<Boolean>) {}



        override fun hideCurrentInputMethod() {}

    }



    private class TestShellExecutor : ShellExecutor {



        override fun execute(runnable: Runnable) {

            runnable.run()

        }



        override fun executeDelayed(r: Runnable, delayMillis: Long) {

            r.run()

        }



        override fun removeCallbacks(r: Runnable) {}



        override fun hasCallback(r: Runnable): Boolean = false

    }



    private inner class FakeBubbleTaskViewFactory : BubbleTaskViewFactory {

        override fun create(): BubbleTaskView {

            val taskViewTaskController = mock<TaskViewTaskController>()

            val taskView = TaskView(context, taskViewTaskController)

            return BubbleTaskView(taskView, shellExecutor)

        }

    }



    private inner class FakeBubbleExpandedViewManager : BubbleExpandedViewManager {



        override val overflowBubbles: List<Bubble>

            get() = emptyList()



        override fun setOverflowListener(listener: BubbleData.Listener) {}



        override fun collapseStack() {}



        override fun updateWindowFlagsForBackpress(intercept: Boolean) {}



        override fun promoteBubbleFromOverflow(bubble: Bubble) {}



        override fun removeBubble(key: String, reason: Int) {}



        override fun dismissBubble(bubble: Bubble, reason: Int) {}



        override fun setAppBubbleTaskId(key: String, taskId: Int) {}



        override fun isStackExpanded(): Boolean = false



        override fun isShowingAsBubbleBar(): Boolean = false

    }

}

libs/WindowManager/Shell/tests/unittest/AndroidManifest.xml

+1 −0
Original line number Diff line number Diff line
@@ -21,6 +21,7 @@ 


    <uses-permission android:name="android.permission.READ_DEVICE_CONFIG" />

    <uses-permission android:name="android.permission.VIBRATE"/>

    <uses-permission android:name="android.permission.MANAGE_ACTIVITY_TASKS"/>



    <application android:debuggable="true" android:largeHeap="true">

        <uses-library android:name="android.test.mock" />