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

Commit 330b2a6e authored by Grace Cheng's avatar Grace Cheng
Browse files

Migrate AuthContainerViewTest to Kosmos 

Migrate AuthContainerViewTest and dependencies to use Kosmos

Flag: TEST_ONLY
Fixes: 335757373
Test: atest AuthContainerViewTest
Change-Id: I023b09bb5f5f71dce2b8a1e5cefd1a2683300e84
parent 0d2f511b

packages/SystemUI/multivalentTests/src/com/android/systemui/biometrics/AuthContainerViewTest.kt

+35 −131
Original line number Diff line number Diff line
@@ -15,9 +15,9 @@ 
 */

package com.android.systemui.biometrics



import android.app.ActivityTaskManager

import android.content.pm.PackageManager

import android.content.packageManager

import android.content.res.Configuration

import android.content.testableContext

import android.hardware.biometrics.BiometricAuthenticator

import android.hardware.biometrics.BiometricConstants

import android.hardware.biometrics.BiometricManager
@@ -26,54 +26,36 @@ import android.hardware.biometrics.PromptContentViewWithMoreOptionsButton 
import android.hardware.biometrics.PromptInfo

import android.hardware.biometrics.PromptVerticalListContentView

import android.hardware.face.FaceSensorPropertiesInternal

import android.hardware.fingerprint.FingerprintManager

import android.hardware.fingerprint.FingerprintSensorPropertiesInternal

import android.os.IBinder

import android.os.UserManager

import android.os.userManager

import android.testing.TestableLooper

import android.testing.TestableLooper.RunWithLooper

import android.testing.ViewUtils

import android.view.View

import android.view.WindowInsets

import android.view.WindowManager

import android.view.accessibility.AccessibilityManager

import android.widget.ScrollView

import androidx.constraintlayout.widget.ConstraintLayout

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

import androidx.test.filters.SmallTest

import com.android.internal.jank.InteractionJankMonitor

import com.android.internal.widget.LockPatternUtils

import com.android.internal.widget.LockPatternUtils.CREDENTIAL_TYPE_PATTERN

import com.android.internal.widget.LockPatternUtils.CREDENTIAL_TYPE_PIN

import com.android.launcher3.icons.IconProvider

import com.android.internal.widget.lockPatternUtils

import com.android.systemui.SysuiTestCase

import com.android.systemui.biometrics.data.repository.FakeBiometricStatusRepository

import com.android.systemui.biometrics.data.repository.FakeDisplayStateRepository

import com.android.systemui.biometrics.data.repository.FakeFingerprintPropertyRepository

import com.android.systemui.biometrics.data.repository.FakePromptRepository

import com.android.systemui.biometrics.domain.interactor.BiometricStatusInteractor

import com.android.systemui.biometrics.domain.interactor.BiometricStatusInteractorImpl

import com.android.systemui.biometrics.domain.interactor.DisplayStateInteractor

import com.android.systemui.biometrics.domain.interactor.DisplayStateInteractorImpl

import com.android.systemui.biometrics.domain.interactor.FakeCredentialInteractor

import com.android.systemui.biometrics.domain.interactor.PromptCredentialInteractor

import com.android.systemui.biometrics.domain.interactor.PromptSelectorInteractorImpl

import com.android.systemui.biometrics.domain.interactor.UdfpsOverlayInteractor

import com.android.systemui.biometrics.ui.viewmodel.CredentialViewModel

import com.android.systemui.biometrics.ui.viewmodel.PromptViewModel

import com.android.systemui.display.data.repository.FakeDisplayRepository

import com.android.systemui.biometrics.domain.interactor.promptSelectorInteractor

import com.android.systemui.biometrics.ui.viewmodel.credentialViewModel

import com.android.systemui.biometrics.ui.viewmodel.promptViewModel

import com.android.systemui.concurrency.fakeExecutor

import com.android.systemui.haptics.msdl.msdlPlayer

import com.android.systemui.keyguard.WakefulnessLifecycle

import com.android.systemui.haptics.vibratorHelper

import com.android.systemui.jank.interactionJankMonitor

import com.android.systemui.keyguard.wakefulnessLifecycle

import com.android.systemui.kosmos.testScope

import com.android.systemui.res.R

import com.android.systemui.statusbar.VibratorHelper

import com.android.systemui.testKosmos

import com.android.systemui.user.domain.interactor.SelectedUserInteractor

import com.android.systemui.util.concurrency.FakeExecutor

import com.android.systemui.util.time.FakeSystemClock

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

import kotlinx.coroutines.Dispatchers

import kotlinx.coroutines.test.StandardTestDispatcher

import kotlinx.coroutines.test.TestScope

import kotlinx.coroutines.test.runCurrent

import org.junit.After

import org.junit.Before
@@ -98,88 +80,26 @@ private const val OP_PACKAGE_NAME = "biometric.testapp" 
@RunWithLooper(setAsMainLooper = true)

@SmallTest

open class AuthContainerViewTest : SysuiTestCase() {



    @JvmField @Rule var mockitoRule = MockitoJUnit.rule()



    @Mock lateinit var callback: AuthDialogCallback

    @Mock lateinit var userManager: UserManager

    @Mock lateinit var fingerprintManager: FingerprintManager

    @Mock lateinit var lockPatternUtils: LockPatternUtils

    @Mock lateinit var wakefulnessLifecycle: WakefulnessLifecycle

    @Mock lateinit var windowToken: IBinder

    @Mock lateinit var interactionJankMonitor: InteractionJankMonitor

    @Mock lateinit var vibrator: VibratorHelper

    @Mock lateinit var udfpsUtils: UdfpsUtils

    @Mock lateinit var authController: AuthController

    @Mock lateinit var selectedUserInteractor: SelectedUserInteractor

    @Mock private lateinit var packageManager: PackageManager

    @Mock private lateinit var activityTaskManager: ActivityTaskManager

    @Mock private lateinit var accessibilityManager: AccessibilityManager



    private lateinit var displayRepository: FakeDisplayRepository

    private lateinit var displayStateInteractor: DisplayStateInteractor

    private lateinit var udfpsOverlayInteractor: UdfpsOverlayInteractor

    private lateinit var biometricStatusInteractor: BiometricStatusInteractor

    private lateinit var iconProvider: IconProvider



    private val testScope = TestScope(StandardTestDispatcher())

    private val fakeExecutor = FakeExecutor(FakeSystemClock())

    private val biometricPromptRepository = FakePromptRepository()

    private val biometricStatusRepository = FakeBiometricStatusRepository()

    private val fingerprintRepository = FakeFingerprintPropertyRepository()

    private val displayStateRepository = FakeDisplayStateRepository()

    private val credentialInteractor = FakeCredentialInteractor()

    private val bpCredentialInteractor =

        PromptCredentialInteractor(

            Dispatchers.Main.immediate,

            biometricPromptRepository,

            credentialInteractor,

        )

    private val promptSelectorInteractor by lazy {

        PromptSelectorInteractorImpl(

            fingerprintRepository,

            displayStateInteractor,

            credentialInteractor,

            biometricPromptRepository,

            lockPatternUtils,

        )

    }



    private val credentialViewModel = CredentialViewModel(mContext, bpCredentialInteractor)

    private val defaultLogoIcon = context.getDrawable(R.drawable.ic_android)



    private val kosmos = testKosmos()

    private val msdlPlayer = kosmos.msdlPlayer

    private val context = kosmos.testableContext

    private val lockPatternUtils = kosmos.lockPatternUtils

    private val packageManager = kosmos.packageManager

    private val userManager: UserManager = kosmos.userManager



    private val testScope = kosmos.testScope

    private val fakeExecutor = kosmos.fakeExecutor



    private val defaultLogoIcon = context.getDrawable(R.drawable.ic_android)



    private var authContainer: TestAuthContainerView? = null



    @Before

    fun setup() {

        displayRepository = FakeDisplayRepository()



        displayStateInteractor =

            DisplayStateInteractorImpl(

                testScope.backgroundScope,

                mContext,

                fakeExecutor,

                displayStateRepository,

                displayRepository,

            )

        udfpsOverlayInteractor =

            UdfpsOverlayInteractor(

                context,

                authController,

                selectedUserInteractor,

                fingerprintManager,

                testScope.backgroundScope,

            )

        biometricStatusInteractor =

            BiometricStatusInteractorImpl(

                activityTaskManager,

                biometricStatusRepository,

                fingerprintRepository,

            )

        iconProvider = IconProvider(context)

        // Set up default logo icon

        whenever(packageManager.getApplicationIcon(OP_PACKAGE_NAME)).thenReturn(defaultLogoIcon)

        context.setMockPackageManager(packageManager)
@@ -428,12 +348,6 @@ open class AuthContainerViewTest : SysuiTestCase() { 


    @Test

    fun testShowBiometricUI_ContentViewWithMoreOptionsButton() {

        var isButtonClicked = false

        val contentView =

            PromptContentViewWithMoreOptionsButton.Builder()

                .setMoreOptionsButtonListener(fakeExecutor) { _, _ -> isButtonClicked = true }

                .build()



        val container = initializeFingerprintContainer()



        waitForIdleSync()
@@ -483,7 +397,6 @@ open class AuthContainerViewTest : SysuiTestCase() { 


    @Test

    fun testShowCredentialUI_withContentViewWithMoreOptionsButton() {

        val contentView =

        PromptContentViewWithMoreOptionsButton.Builder()

            .setMoreOptionsButtonListener(fakeExecutor) { _, _ -> }

            .build()
@@ -499,6 +412,7 @@ open class AuthContainerViewTest : SysuiTestCase() { 


    @Test

    fun testCredentialViewUsesEffectiveUserId() {

        kosmos.userManager

        whenever(userManager.getCredentialOwnerProfile(anyInt())).thenReturn(200)

        whenever(lockPatternUtils.getCredentialTypeForUser(eq(200)))

            .thenReturn(CREDENTIAL_TYPE_PATTERN)
@@ -667,27 +581,17 @@ open class AuthContainerViewTest : SysuiTestCase() { 
            testScope.backgroundScope,

            fingerprintProps,

            faceProps,

            wakefulnessLifecycle,

            userManager,

            kosmos.wakefulnessLifecycle,

            kosmos.userManager,

            null /* authContextPlugins */,

            lockPatternUtils,

            interactionJankMonitor,

            { promptSelectorInteractor },

            PromptViewModel(

                displayStateInteractor,

                promptSelectorInteractor,

                context,

                udfpsOverlayInteractor,

                biometricStatusInteractor,

                udfpsUtils,

                iconProvider,

                activityTaskManager,

                accessibilityManager,

            ),

            { credentialViewModel },

            fakeExecutor,

            vibrator,

            msdlPlayer,

            kosmos.lockPatternUtils,

            kosmos.interactionJankMonitor,

            { kosmos.promptSelectorInteractor },

            kosmos.promptViewModel,

            { kosmos.credentialViewModel },

            kosmos.fakeExecutor,

            kosmos.vibratorHelper,

            kosmos.msdlPlayer,

        ) {

        override fun postOnAnimation(runnable: Runnable) {

            runnable.run()

packages/SystemUI/tests/utils/src/com/android/systemui/biometrics/domain/interactor/FakeCredentialInteractorKosmos.kt

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

 * Copyright (C) 2025 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.biometrics.domain.interactor



import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.Kosmos.Fixture



val Kosmos.credentialInteractor by Fixture { FakeCredentialInteractor() }

packages/SystemUI/tests/utils/src/com/android/systemui/biometrics/domain/interactor/PromptCredentialInteractorKosmos.kt

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

 * Copyright (C) 2025 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.biometrics.domain.interactor



import com.android.systemui.biometrics.data.repository.promptRepository

import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.Kosmos.Fixture

import kotlinx.coroutines.Dispatchers



val Kosmos.promptCredentialInteractor by Fixture {

    PromptCredentialInteractor(Dispatchers.Main.immediate, promptRepository, credentialInteractor)

}

packages/SystemUI/tests/utils/src/com/android/systemui/biometrics/domain/interactor/PromptSelectorInteractorKosmos.kt

+1 −1
Original line number Diff line number Diff line
@@ -27,7 +27,7 @@ val Kosmos.promptSelectorInteractor by Fixture { 
        fingerprintPropertyRepository = fingerprintPropertyRepository,

        displayStateInteractor = displayStateInteractor,

        promptRepository = promptRepository,

        credentialInteractor = FakeCredentialInteractor(),

        credentialInteractor = credentialInteractor,

        lockPatternUtils = lockPatternUtils,

    )

}

packages/SystemUI/tests/utils/src/com/android/systemui/biometrics/ui/viewmodel/CredentialViewModelKosmos.kt

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

 * Copyright (C) 2025 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.biometrics.ui.viewmodel



import android.content.applicationContext

import com.android.systemui.biometrics.domain.interactor.promptCredentialInteractor

import com.android.systemui.kosmos.Kosmos

import com.android.systemui.kosmos.Kosmos.Fixture



val Kosmos.credentialViewModel by Fixture {

    CredentialViewModel(applicationContext, promptCredentialInteractor)

}