Fix tests in TimeoutToDockUserPreferenceControllerTest

Bug: 313530297
Test: atest TimeoutToDockUserPreferenceControllerTest
Change-Id: I3a67744fb7652f5815ab1ba1ba88e13f5436e127
This commit is contained in:
Tetiana Meronyk
2023-11-29 13:05:41 +00:00
parent bc69be0ff6
commit b1ecd3bba8

View File

@@ -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);