From 61aab4893bf5e211bcba176b447ea2afc03af1c3 Mon Sep 17 00:00:00 2001
From: Rob Mickle <rmickle@google.com>
Date: Mon, 19 May 2025 17:50:56 -0700
Subject: [PATCH] Update test to use fake repository instead of mock.

Bug: 378443899
Test: OverviewCommandHelperTest
Flag: EXEMPT test cleanup up
Change-Id: Ie449a659bf6b1b33acbb6f9f6f45b800e31cf447
---
 .../src/com/android/quickstep/OverviewCommandHelperTest.kt  | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/quickstep/tests/multivalentTests/src/com/android/quickstep/OverviewCommandHelperTest.kt b/quickstep/tests/multivalentTests/src/com/android/quickstep/OverviewCommandHelperTest.kt
index 8f3f74357e..d98cb989b5 100644
--- a/quickstep/tests/multivalentTests/src/com/android/quickstep/OverviewCommandHelperTest.kt
+++ b/quickstep/tests/multivalentTests/src/com/android/quickstep/OverviewCommandHelperTest.kt
@@ -20,6 +20,7 @@ import android.content.Intent
 import android.view.Display.DEFAULT_DISPLAY
 import androidx.test.filters.SmallTest
 import com.android.app.displaylib.DisplayRepository
+import com.android.app.displaylib.fakes.FakePerDisplayRepository
 import com.android.app.displaylib.PerDisplayRepository
 import com.android.launcher3.LauncherState
 import com.android.launcher3.statemanager.StateManager
@@ -96,8 +97,6 @@ class OverviewCommandHelperTest {
         whenever(overviewComponentObserver.getHomeIntent(any())).thenReturn(mock<Intent>())
         whenever(recentView.getStateManager()).thenReturn(stateManager)
         whenever(containerInterface.switchToRecentsIfVisible(any())).thenReturn(true)
-        val taskAnimationManagerRepository = mock<PerDisplayRepository<TaskAnimationManager>>()
-        whenever(taskAnimationManagerRepository.get(any())).thenReturn(taskAnimationManager)
         whenever(taskAnimationManager.maybeStartHomeAction(any())).thenAnswer { invocation ->
             invocation.getArgument<Runnable>(0).run()
         }
@@ -110,7 +109,8 @@ class OverviewCommandHelperTest {
                     dispatcherProvider = TestDispatcherProvider(dispatcher),
                     displayRepository = displayRepository,
                     taskbarManager = mock(),
-                    taskAnimationManagerRepository = taskAnimationManagerRepository,
+                    taskAnimationManagerRepository =
+                        FakePerDisplayRepository<TaskAnimationManager> { _ -> taskAnimationManager },
                     elapsedRealtime = ::elapsedRealtime,
                 )
             )