From 7b83486fe46409ad9eaddf008fa6bc63f80d035e Mon Sep 17 00:00:00 2001
From: Andras Kloczl <andraskloczl@google.com>
Date: Wed, 10 Aug 2022 17:32:17 +0200
Subject: [PATCH] Trying to fix flaky launcher tests

Test: run AddWorkspaceItemsTaskTest.kt
Bug: 227650070
Change-Id: Ie093c5b13bdb4143829f82767fdeb208805f388a
---
 .../com/android/launcher3/model/AddWorkspaceItemsTaskTest.kt    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/tests/src/com/android/launcher3/model/AddWorkspaceItemsTaskTest.kt b/tests/src/com/android/launcher3/model/AddWorkspaceItemsTaskTest.kt
index 65d938b91a..2cd21cb145 100644
--- a/tests/src/com/android/launcher3/model/AddWorkspaceItemsTaskTest.kt
+++ b/tests/src/com/android/launcher3/model/AddWorkspaceItemsTaskTest.kt
@@ -23,6 +23,7 @@ import com.android.launcher3.model.data.ItemInfo
 import com.android.launcher3.model.data.WorkspaceItemInfo
 import com.android.launcher3.util.Executors
 import com.android.launcher3.util.IntArray
+import com.android.launcher3.util.IntSet
 import com.android.launcher3.util.same
 import com.android.launcher3.util.eq
 import com.android.launcher3.util.any
@@ -64,6 +65,7 @@ class AddWorkspaceItemsTaskTest : AbstractWorkspaceModelTest() {
     override fun setup() {
         super.setup()
         MockitoAnnotations.initMocks(this)
+        whenever(mDataModelCallbacks.getPagesToBindSynchronously(any())).thenReturn(IntSet())
         Executors.MAIN_EXECUTOR.submit { mModelHelper.model.addCallbacks(mDataModelCallbacks) }
             .get()
     }