PawletOS/app_Settings
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add the controller for the top switch.

Browse Source 
Bug: 302189945
Change-Id: Ie43ac181e643a8b215ca830afc6ef670d91a5762
This commit is contained in:
Wa Gao
2023-10-27 18:52:52 +00:00
parent 198ae8901b
commit aa5d7420db
4 changed files with 364 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

118
tests/robotests/src/com/android/settings/security/ContentProtectionPreferenceFragmentTest.java
View File

@@ -17,16 +17,29 @@
package com.android.settings.security;
import static android.app.settings.SettingsEnums.CONTENT_PROTECTION_PREFERENCE;
import static com.android.settings.security.ContentProtectionPreferenceFragment.KEY_WORK_PROFILE_SWITCH;
import static com.google.common.truth.Truth.assertThat;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.spy;
import android.content.ComponentName;
import android.content.Context;
import android.content.pm.UserInfo;
import android.os.UserHandle;
import android.os.UserManager;
import android.provider.SearchIndexableResource;
import androidx.preference.PreferenceScreen;
import androidx.preference.SwitchPreference;
import com.android.settings.R;
import com.android.settings.Utils;
import com.android.settings.testutils.XmlTestUtils;
import com.android.settings.testutils.shadow.ShadowDashboardFragment;
import com.android.settings.testutils.shadow.ShadowUtils;
import org.junit.Before;
import org.junit.Test;
@@ -37,44 +50,116 @@ import org.robolectric.RobolectricTestRunner;
import org.robolectric.RuntimeEnvironment;
import org.robolectric.annotation.Config;
import java.util.Arrays;
import java.util.List;
@RunWith(RobolectricTestRunner.class)
@Config(shadows = ShadowDashboardFragment.class)
@Config(
shadows = {
ShadowDashboardFragment.class,
ShadowUtils.class,
})
public class ContentProtectionPreferenceFragmentTest {
private static final int TEST_PRIMARY_USER_ID = 10;
private static final int TEST_MANAGED_PROFILE_ID = 11;
@Mock
private ContentProtectionPreferenceFragment mMockFragment;
private ContentProtectionPreferenceFragment mFragment;
@Mock private UserManager mMockUserManager;
private Context mContext;
private PreferenceScreen mScreen;
private SwitchPreference mWorkProfileSwitch;
@Before
public void setUp() throws Exception {
MockitoAnnotations.initMocks(this);
mContext = spy(RuntimeEnvironment.application);
mMockFragment = spy(new ContentProtectionPreferenceFragment());
doReturn(mContext).when(mMockFragment).getContext();
mContext = spy(RuntimeEnvironment.application);
mFragment = spy(new ContentProtectionPreferenceFragment());
mScreen = spy(new PreferenceScreen(mContext, /* attrs= */ null));
doReturn(mContext).when(mFragment).getContext();
doReturn(mScreen).when(mFragment).getPreferenceScreen();
mWorkProfileSwitch = new SwitchPreference(mContext);
mWorkProfileSwitch.setVisible(false);
doReturn(mWorkProfileSwitch).when(mScreen).findPreference(KEY_WORK_PROFILE_SWITCH);
doReturn(mMockUserManager).when(mContext).getSystemService(UserManager.class);
doReturn(TEST_PRIMARY_USER_ID).when(mMockUserManager).getUserHandle();
UserInfo primaryUser =
new UserInfo(
TEST_PRIMARY_USER_ID,
null,
UserInfo.FLAG_INITIALIZED | UserInfo.FLAG_PRIMARY);
doReturn(primaryUser).when(mMockUserManager).getUserInfo(TEST_PRIMARY_USER_ID);
UserInfo managedProfile =
new UserInfo(
TEST_MANAGED_PROFILE_ID,
null,
UserInfo.FLAG_INITIALIZED | UserInfo.FLAG_MANAGED_PROFILE);
doReturn(managedProfile).when(mMockUserManager).getUserInfo(TEST_MANAGED_PROFILE_ID);
}
@Test
public void onActivityCreated_workProfileDisplayWorkSwitch() {
UserHandle[] userHandles =
new UserHandle[] {
new UserHandle(TEST_PRIMARY_USER_ID), new UserHandle(TEST_MANAGED_PROFILE_ID)
};
doReturn(Arrays.asList(userHandles)).when(mMockUserManager).getUserProfiles();
assertThat(Utils.getManagedProfile(mMockUserManager).getIdentifier())
.isEqualTo(TEST_MANAGED_PROFILE_ID);
mFragment.onActivityCreated(null);
assertThat(mWorkProfileSwitch.isVisible()).isTrue();
assertThat(mWorkProfileSwitch.isChecked()).isFalse();
assertThat(mWorkProfileSwitch.isEnabled()).isFalse();
}
@Test
public void onActivityCreated_fullyManagedMode_bottomSwitchInvisible() {
final ComponentName componentName =
ComponentName.unflattenFromString("com.android.test/.DeviceAdminReceiver");
ShadowUtils.setDeviceOwnerComponent(componentName);
mFragment.onActivityCreated(null);
assertThat(mWorkProfileSwitch.isVisible()).isFalse();
}
@Test
public void onActivityCreated_personalProfileHideWorkSwitch() {
UserHandle[] userHandles = new UserHandle[] {new UserHandle(TEST_PRIMARY_USER_ID)};
doReturn(Arrays.asList(userHandles)).when(mMockUserManager).getUserProfiles();
assertThat(Utils.getManagedProfile(mMockUserManager)).isNull();
mFragment.onActivityCreated(null);
assertThat(mWorkProfileSwitch.isVisible()).isFalse();
}
@Test
public void getMetricsCategory() {
assertThat(mMockFragment.getMetricsCategory()).isEqualTo(CONTENT_PROTECTION_PREFERENCE);
assertThat(mFragment.getMetricsCategory()).isEqualTo(CONTENT_PROTECTION_PREFERENCE);
}
@Test
public void getPreferenceScreenResId(){
assertThat(mMockFragment.getPreferenceScreenResId())
.isEqualTo(R.layout.content_protection_preference_fragment);
public void getPreferenceScreenResId() {
assertThat(mFragment.getPreferenceScreenResId())
.isEqualTo(R.layout.content_protection_preference_fragment);
}
@Test
public void getNonIndexableKeys_existInXmlLayout() {
final List<String> nonIndexableKeys =
ContentProtectionPreferenceFragment.SEARCH_INDEX_DATA_PROVIDER
.getNonIndexableKeys(mContext);
ContentProtectionPreferenceFragment.SEARCH_INDEX_DATA_PROVIDER.getNonIndexableKeys(
mContext);
final List<String> allKeys =
XmlTestUtils.getKeysFromPreferenceXml(mContext,
R.layout.content_protection_preference_fragment);
XmlTestUtils.getKeysFromPreferenceXml(
mContext, R.layout.content_protection_preference_fragment);
assertThat(allKeys).containsAtLeastElementsIn(nonIndexableKeys);
}
@@ -83,10 +168,11 @@ public class ContentProtectionPreferenceFragmentTest {
public void searchIndexProvider_shouldIndexResource() {
final List<SearchIndexableResource> indexRes =
ContentProtectionPreferenceFragment.SEARCH_INDEX_DATA_PROVIDER
.getXmlResourcesToIndex(mContext, /* enabled = */ true);
.getXmlResourcesToIndex(mContext, /* enabled= */ true);
assertThat(indexRes).isNotNull();
assertThat(indexRes).isNotEmpty();
assertThat(indexRes.get(0).xmlResId).isEqualTo(mMockFragment.getPreferenceScreenResId());
assertThat(indexRes.get(0).xmlResId).isEqualTo(mFragment.getPreferenceScreenResId());
}
}