commit b1ecd3bba8931e30d0f7e27773c69e9c8d42e497
author Tetiana Meronyk <tetianameronyk@google.com> Wed Nov 29 13:05:41 2023 +0000
committer Tetiana Meronyk <tetianameronyk@google.com> Wed Nov 29 14:10:07 2023 +0000
tree 06d25dd8e54432ce0c319e26b0c915e14ae4a7af
parent bc69be0ff69f219400f3cf5b6e8835f1c4d1cefd

Fix tests in TimeoutToDockUserPreferenceControllerTest

Bug: 313530297
Test: atest TimeoutToDockUserPreferenceControllerTest
Change-Id: I3a67744fb7652f5815ab1ba1ba88e13f5436e127

tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java

diff --git a/tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java b/tests/robotests/src/com/android/settings/users/TimeoutToDockUserPreferenceControllerTest.java
index 3d8e893..fe88148 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 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 @@
 
     @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 @@
 
     @Test
     public void getAvailabilityStatus_featureAndMultiUserEnabledAndNonMainUser_returnAvailable() {
-        when(Utils.canCurrentUserDream(mContext)).thenReturn(false);
+        when(mUserManager.isUserForeground()).thenReturn(true);
 
         assertThat(mController.getAvailabilityStatus()).isEqualTo(
                 BasePreferenceController.AVAILABLE);