From b1ecd3bba8931e30d0f7e27773c69e9c8d42e497 Mon Sep 17 00:00:00 2001 From: Tetiana Meronyk Date: Wed, 29 Nov 2023 13:05:41 +0000 Subject: [PATCH] Fix tests in TimeoutToDockUserPreferenceControllerTest Bug: 313530297 Test: atest TimeoutToDockUserPreferenceControllerTest Change-Id: I3a67744fb7652f5815ab1ba1ba88e13f5436e127 --- .../users/TimeoutToDockUserPreferenceControllerTest.java | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java index 3d8e8930775..fe88148be62 100644 --- a/tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java +++ b/tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java @@ -33,7 +33,6 @@ import android.provider.Settings; import androidx.test.core.app.ApplicationProvider; import com.android.settings.R; -import com.android.settings.Utils; import com.android.settings.core.BasePreferenceController; import com.android.settings.fuelgauge.BatteryBackupHelperTest.ShadowUserHandle; import com.android.settings.testutils.shadow.ShadowSecureSettings; @@ -128,7 +127,8 @@ public class TimeoutToDockUserPreferenceControllerTest { @Test public void getAvailabilityStatus_isCurrentlyMainUser_returnDisabledForUser() { - when(Utils.canCurrentUserDream(mContext)).thenReturn(true); + when(mUserManager.getMainUser()).thenReturn(UserHandle.CURRENT); + when(mUserManager.isUserForeground()).thenReturn(true); assertThat(mController.getAvailabilityStatus()).isEqualTo( BasePreferenceController.DISABLED_FOR_USER); @@ -136,7 +136,7 @@ public class TimeoutToDockUserPreferenceControllerTest { @Test public void getAvailabilityStatus_featureAndMultiUserEnabledAndNonMainUser_returnAvailable() { - when(Utils.canCurrentUserDream(mContext)).thenReturn(false); + when(mUserManager.isUserForeground()).thenReturn(true); assertThat(mController.getAvailabilityStatus()).isEqualTo( BasePreferenceController.AVAILABLE);