diff --git a/src/com/android/settings/users/MultiUserSwitchBarController.java b/src/com/android/settings/users/MultiUserSwitchBarController.java index 58de14963fa..33651c31fc4 100644 --- a/src/com/android/settings/users/MultiUserSwitchBarController.java +++ b/src/com/android/settings/users/MultiUserSwitchBarController.java @@ -57,6 +57,11 @@ public class MultiUserSwitchBarController implements SwitchWidgetController.OnSw mSwitchBar.setDisabledByAdmin(RestrictedLockUtilsInternal .checkIfRestrictionEnforced(mContext, UserManager.DISALLOW_USER_SWITCH, UserHandle.myUserId())); + + } else if (mUserCapabilities.mDisallowAddUser) { + mSwitchBar.setDisabledByAdmin(RestrictedLockUtilsInternal + .checkIfRestrictionEnforced(mContext, UserManager.DISALLOW_ADD_USER, + UserHandle.myUserId())); } else { mSwitchBar.setEnabled(!mUserCapabilities.mDisallowSwitchUser && !mUserCapabilities.mIsGuest && mUserCapabilities.isAdmin()); diff --git a/src/com/android/settings/users/UserSettings.java b/src/com/android/settings/users/UserSettings.java index cb778c9281e..5f9f2fe93ac 100644 --- a/src/com/android/settings/users/UserSettings.java +++ b/src/com/android/settings/users/UserSettings.java @@ -1489,7 +1489,7 @@ public class UserSettings extends SettingsPreferenceFragment private void updateAddUserCommon(Context context, RestrictedPreference addUser, boolean canAddRestrictedProfile) { - if ((mUserCaps.mCanAddUser || mUserCaps.mDisallowAddUserSetByAdmin) + if ((mUserCaps.mCanAddUser || !mUserCaps.mDisallowAddUserSetByAdmin) && WizardManagerHelper.isDeviceProvisioned(context) && mUserCaps.mUserSwitcherEnabled) { addUser.setVisible(true); diff --git a/tests/robotests/src/com/android/settings/users/UserSettingsTest.java b/tests/robotests/src/com/android/settings/users/UserSettingsTest.java index ddbcea1a741..54f579c4e4a 100644 --- a/tests/robotests/src/com/android/settings/users/UserSettingsTest.java +++ b/tests/robotests/src/com/android/settings/users/UserSettingsTest.java @@ -402,7 +402,7 @@ public class UserSettingsTest { } @Test - public void updateUserList_addUserDisallowedByAdmin_shouldShowDisabledAddUser() { + public void updateUserList_addUserDisallowedByAdmin_shouldNotShowAddUser() { RestrictedLockUtils.EnforcedAdmin enforcedAdmin = mock( RestrictedLockUtils.EnforcedAdmin.class); mUserCapabilities.mEnforcedAdmin = enforcedAdmin; @@ -413,20 +413,16 @@ public class UserSettingsTest { mFragment.updateUserList(); - verify(mAddUserPreference).setVisible(true); - ArgumentCaptor captor = ArgumentCaptor.forClass( - RestrictedLockUtils.EnforcedAdmin.class); - verify(mAddUserPreference).setDisabledByAdmin(captor.capture()); - assertThat(captor.getValue()).isEqualTo(enforcedAdmin); + verify(mAddUserPreference).setVisible(false); } @Test - public void updateUserList_cannotAddUserButCanSwitchUser_shouldNotShowAddUser() { + public void updateUserList_cannotAddUserButCanSwitchUser_shouldShowDisabledAddUser() { mUserCapabilities.mCanAddUser = false; mFragment.updateUserList(); - verify(mAddUserPreference).setVisible(false); + verify(mAddUserPreference).setEnabled(false); } @Test