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

Commit d7dc6ba3 authored by Brian Isganitis's avatar Brian Isganitis
Browse files

Add tests for TaskbarAllAppsViewController. 

Test: TaskbarAllAppsViewControllerTest
Flag: TEST_ONLY
Bug: 346394506
Change-Id: Ie1b6609b243326eb11d03d2ea7b0e98bd60c3a71
parent bd90d3aa

quickstep/src/com/android/launcher3/appprediction/AppsDividerView.java

+6 −0
Original line number Diff line number Diff line
@@ -31,6 +31,7 @@ import android.view.View; 
import android.view.accessibility.AccessibilityManager;



import androidx.annotation.ColorInt;

import androidx.annotation.VisibleForTesting;



import com.android.launcher3.R;

import com.android.launcher3.Utilities;
@@ -253,4 +254,9 @@ public class AppsDividerView extends View implements FloatingHeaderRow { 
    public View getFocusedChild() {

        return null;

    }



    @VisibleForTesting

    public DividerType getDividerType() {

        return mDividerType;

    }

}

quickstep/tests/multivalentTests/src/com/android/launcher3/taskbar/allapps/TaskbarAllAppsControllerTest.kt

+3 −3
Original line number Diff line number Diff line
@@ -199,8 +199,8 @@ class TaskbarAllAppsControllerTest { 
        assertThat(editText?.hasFocus()).isTrue()

    }



    private companion object {

        private val TEST_APPS =

    companion object {

        val TEST_APPS =

            Array(16) {

                AppInfo(

                    ComponentName(
@@ -213,6 +213,6 @@ class TaskbarAllAppsControllerTest { 
                )

            }



        private val TEST_PREDICTED_APPS = TEST_APPS.take(4).map { WorkspaceItemInfo(it) }

        val TEST_PREDICTED_APPS = TEST_APPS.take(4).map { WorkspaceItemInfo(it) }

    }

}

quickstep/tests/multivalentTests/src/com/android/launcher3/taskbar/allapps/TaskbarAllAppsViewControllerTest.kt

0 → 100644
+156 −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.launcher3.taskbar.allapps



import androidx.test.platform.app.InstrumentationRegistry.getInstrumentation

import com.android.launcher3.R

import com.android.launcher3.appprediction.AppsDividerView

import com.android.launcher3.appprediction.AppsDividerView.DividerType

import com.android.launcher3.appprediction.PredictionRowView

import com.android.launcher3.taskbar.TaskbarStashController

import com.android.launcher3.taskbar.TaskbarStashController.FLAG_STASHED_IN_APP_AUTO

import com.android.launcher3.taskbar.allapps.TaskbarAllAppsControllerTest.Companion.TEST_PREDICTED_APPS

import com.android.launcher3.taskbar.overlay.TaskbarOverlayController

import com.android.launcher3.taskbar.rules.TaskbarModeRule

import com.android.launcher3.taskbar.rules.TaskbarModeRule.Mode.PINNED

import com.android.launcher3.taskbar.rules.TaskbarModeRule.Mode.TRANSIENT

import com.android.launcher3.taskbar.rules.TaskbarModeRule.TaskbarMode

import com.android.launcher3.taskbar.rules.TaskbarPreferenceRule

import com.android.launcher3.taskbar.rules.TaskbarUnitTestRule

import com.android.launcher3.taskbar.rules.TaskbarUnitTestRule.InjectController

import com.android.launcher3.taskbar.rules.TaskbarWindowSandboxContext

import com.android.launcher3.util.LauncherMultivalentJUnit

import com.android.launcher3.util.LauncherMultivalentJUnit.EmulatedDevices

import com.android.launcher3.util.OnboardingPrefs.ALL_APPS_VISITED_COUNT

import com.android.launcher3.util.TestUtil

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

import org.junit.After

import org.junit.Rule

import org.junit.Test

import org.junit.runner.RunWith



@RunWith(LauncherMultivalentJUnit::class)

@EmulatedDevices(["pixelFoldable2023"])

class TaskbarAllAppsViewControllerTest {



    private val context = TaskbarWindowSandboxContext.create(getInstrumentation().targetContext)



    @get:Rule(order = 0) val taskbarModeRule = TaskbarModeRule(context)

    @get:Rule(order = 1)

    val allAppsVisitedPreferenceRule =

        TaskbarPreferenceRule(context, ALL_APPS_VISITED_COUNT.prefItem)

    @get:Rule(order = 2) val taskbarUnitTestRule = TaskbarUnitTestRule(this, context)



    @InjectController lateinit var overlayController: TaskbarOverlayController

    @InjectController lateinit var stashController: TaskbarStashController



    private val searchSessionController =

        TestUtil.getOnUiThread { TaskbarSearchSessionController.newInstance(context) }



    @After

    fun cleanUpSearchSessionController() {

        getInstrumentation().runOnMainSync { searchSessionController.onDestroy() }

    }



    @Test

    @TaskbarMode(TRANSIENT)

    fun testShow_transientMode_stashesTaskbar() {

        getInstrumentation().runOnMainSync {

            stashController.updateStateForFlag(FLAG_STASHED_IN_APP_AUTO.toLong(), false)

            stashController.applyState(0)

        }



        val viewController = createViewController()

        getInstrumentation().runOnMainSync { viewController.show(false) }

        assertThat(stashController.isStashed).isTrue()

    }



    @Test

    @TaskbarMode(PINNED)

    fun testShow_pinnedMode_taskbarDoesNotStash() {

        val viewController = createViewController()

        getInstrumentation().runOnMainSync { viewController.show(false) }

        assertThat(stashController.isStashed).isFalse()

    }



    @Test

    @TaskbarMode(TRANSIENT)

    fun testHide_transientMode_unstashesTaskbar() {

        getInstrumentation().runOnMainSync {

            stashController.updateStateForFlag(FLAG_STASHED_IN_APP_AUTO.toLong(), false)

            stashController.applyState(0)

        }



        val viewController = createViewController()

        getInstrumentation().runOnMainSync { viewController.show(false) }

        getInstrumentation().runOnMainSync { viewController.close(false) }

        assertThat(stashController.isStashed).isFalse()

    }



    @Test

    fun testShow_firstAllAppsVisit_hasAllAppsTextDivider() {

        allAppsVisitedPreferenceRule.value = 0

        val viewController = createViewController()

        getInstrumentation().runOnMainSync { viewController.show(false) }



        val appsView = overlayController.requestWindow().appsView

        getInstrumentation().runOnMainSync {

            appsView.floatingHeaderView

                .findFixedRowByType(PredictionRowView::class.java)

                .setPredictedApps(TEST_PREDICTED_APPS)

        }



        val dividerView =

            appsView.floatingHeaderView.findFixedRowByType(AppsDividerView::class.java)

        assertThat(dividerView.dividerType).isEqualTo(DividerType.ALL_APPS_LABEL)

    }



    @Test

    fun testShow_maxAllAppsVisitedCount_hasLineDivider() {

        allAppsVisitedPreferenceRule.value = ALL_APPS_VISITED_COUNT.maxCount

        val viewController = createViewController()

        getInstrumentation().runOnMainSync { viewController.show(false) }



        val appsView = overlayController.requestWindow().appsView

        getInstrumentation().runOnMainSync {

            appsView.floatingHeaderView

                .findFixedRowByType(PredictionRowView::class.java)

                .setPredictedApps(TEST_PREDICTED_APPS)

        }



        val dividerView =

            appsView.floatingHeaderView.findFixedRowByType(AppsDividerView::class.java)

        assertThat(dividerView.dividerType).isEqualTo(DividerType.LINE)

    }



    private fun createViewController(): TaskbarAllAppsViewController {

        return TestUtil.getOnUiThread {

            val overlayContext = overlayController.requestWindow()

            TaskbarAllAppsViewController(

                overlayContext,

                overlayContext.layoutInflater.inflate(

                    R.layout.taskbar_all_apps_sheet,

                    overlayContext.dragLayer,

                    false,

                ) as TaskbarAllAppsSlideInView,

                taskbarUnitTestRule.activityContext.controllers,

                searchSessionController,

                /* showKeyboard= */ false, // Covered in TaskbarAllAppsControllerTest.

            )

        }

    }

}