Merge "Remove jank on DND schedule page" into udc-dev am: a0c9ce5b2f am: f0bb4e8ce7

Original change: https://googleplex-android-review.googlesource.com/c/platform/packages/apps/Settings/+/21650757

Change-Id: I863c9046230f6f0b27d4bfe37d1d0cd61dce2cc9
Signed-off-by: Automerger Merge Worker <android-build-automerger-merge-worker@system.gserviceaccount.com>
This commit is contained in:
Julia Reynolds
2023-03-01 19:48:17 +00:00
committed by Automerger Merge Worker
5 changed files with 162 additions and 161 deletions

View File

@@ -17,13 +17,15 @@
package com.android.settings.notification.zen;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.fail;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyInt;
import static org.mockito.ArgumentMatchers.anyString;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.never;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
@@ -31,13 +33,19 @@ import static org.mockito.Mockito.when;
import android.app.AutomaticZenRule;
import android.content.Context;
import android.content.SharedPreferences;
import android.content.pm.PackageManager;
import android.content.pm.ResolveInfo;
import android.provider.Settings;
import androidx.fragment.app.Fragment;
import androidx.preference.PreferenceCategory;
import androidx.preference.PreferenceManager;
import androidx.preference.PreferenceScreen;
import androidx.test.core.app.ApplicationProvider;
import com.google.common.collect.ImmutableList;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
@@ -47,6 +55,7 @@ import org.robolectric.RobolectricTestRunner;
import org.robolectric.util.ReflectionHelpers;
import java.lang.reflect.Field;
import java.util.AbstractMap;
import java.util.HashMap;
import java.util.Map;
@@ -56,39 +65,81 @@ public class ZenModeAutomaticRulesPreferenceControllerTest {
private ZenModeAutomaticRulesPreferenceController mController;
@Mock
private ZenModeBackend mBackend;
@Mock
private PreferenceCategory mockPref;
@Mock
private PreferenceCategory mPreferenceCategory;
private PreferenceScreen mPreferenceScreen;
@Mock
private ZenRulePreference mZenRulePreference;
private Context mContext;
@Mock
PackageManager mPm;
@Before
public void setup() {
MockitoAnnotations.initMocks(this);
mContext = ApplicationProvider.getApplicationContext();
mController = spy(new ZenModeAutomaticRulesPreferenceController(mContext, mock(Fragment.class),
null));
ReflectionHelpers.setField(mController, "mBackend", mBackend);
when(mPreferenceScreen.findPreference(mController.getPreferenceKey())).thenReturn(
mockPref);
mController.displayPreference(mPreferenceScreen);
doReturn(mZenRulePreference).when(mController).createZenRulePreference(any());
mContext = spy(ApplicationProvider.getApplicationContext());
when(mContext.getPackageManager()).thenReturn(mPm);
when(mPm.queryIntentActivities(any(), any())).thenReturn(
ImmutableList.of(mock(ResolveInfo.class)));
when(mBackend.getAutomaticZenRules()).thenReturn(new Map.Entry[0]);
mController = spy(new ZenModeAutomaticRulesPreferenceController(
mContext, mock(Fragment.class), null, mBackend));
final PreferenceManager preferenceManager = new PreferenceManager(mContext);
mPreferenceScreen = preferenceManager.createPreferenceScreen(mContext);
mPreferenceCategory = spy(new PreferenceCategory(mContext));
mPreferenceCategory.setKey(mController.getPreferenceKey());
mPreferenceScreen.addPreference(mPreferenceCategory);
}
@Test
public void testDisplayPreference_resetsPreferencesWhenCategoryEmpty() {
// when the PreferenceCategory is empty (no preferences), make sure we clear out any
// stale state in the cached set of zen rule preferences
mController.mZenRulePreferences.put("test1_id", mZenRulePreference);
when(mockPref.getPreferenceCount()).thenReturn(0);
public void testDisplayPreference_notPersistent() {
mController.displayPreference(mPreferenceScreen);
assertTrue(mController.mZenRulePreferences.isEmpty());
assertFalse(mPreferenceCategory.isPersistent());
}
@Test
public void testUpdateState_clearsPreferencesWhenAddingNewPreferences() {
public void testDisplayThenUpdateState_onlyAddsOnceRulesUnchanged() {
final int NUM_RULES = 1;
Map<String, AutomaticZenRule> rMap = new HashMap<>();
String ruleId1 = "test1_id";
AutomaticZenRule autoRule1 = new AutomaticZenRule("test_rule_1", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10);
rMap.put(ruleId1, autoRule1);
// should add 1 new preferences to mockPref
mockGetAutomaticZenRules(NUM_RULES, rMap);
mController.displayPreference(mPreferenceScreen);
mController.updateState(mPreferenceCategory);
assertEquals(NUM_RULES, mPreferenceCategory.getPreferenceCount());
verify(mPreferenceCategory, times(1)).addPreference(any());
}
@Test
public void testDisplayThenUpdateState_addsIfRulesChange() {
Map<String, AutomaticZenRule> rMap = new HashMap<>();
String ruleId1 = "test1_id";
AutomaticZenRule autoRule1 = new AutomaticZenRule("test_rule_1", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10);
rMap.put(ruleId1, autoRule1);
mockGetAutomaticZenRules(1, rMap);
// adds one
mController.displayPreference(mPreferenceScreen);
String ruleId2 = "test2_id";
AutomaticZenRule autoRule2 = new AutomaticZenRule("test_rule_2", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 20);
rMap.put(ruleId2, autoRule2);
mockGetAutomaticZenRules(2, rMap);
mController.updateState(mPreferenceCategory);
assertEquals(2, mPreferenceCategory.getPreferenceCount());
verify(mPreferenceCategory, times(2)).addPreference(any());
}
@Test
public void testUpdateState_addingNewPreferences() {
mController.displayPreference(mPreferenceScreen);
final int NUM_RULES = 3;
Map<String, AutomaticZenRule> rMap = new HashMap<>();
@@ -109,17 +160,27 @@ public class ZenModeAutomaticRulesPreferenceControllerTest {
// should add 3 new preferences to mockPref
mockGetAutomaticZenRules(NUM_RULES, rMap);
mController.updateState(mockPref);
verify(mockPref, times(1)).removeAll();
verify(mockPref, times(NUM_RULES)).addPreference(any());
assertEquals(NUM_RULES, mController.mZenRulePreferences.size());
mController.updateState(mPreferenceCategory);
assertEquals(NUM_RULES, mPreferenceCategory.getPreferenceCount());
}
@Test
public void testUpdateState_clearsPreferencesWhenRemovingPreferences(){
public void testUpdateState_addsAndRemoves(){
mController.displayPreference(mPreferenceScreen);
final int NUM_RULES = 2;
Map<String, AutomaticZenRule> rMap = new HashMap<>();
String FAKE_1 = "fake key 1";
String FAKE_2 = "fake 2";
mPreferenceCategory.addPreference(new ZenRulePreference(mContext,
new AbstractMap.SimpleEntry<>(FAKE_1, mock(AutomaticZenRule.class)),
null, null, mBackend));
mPreferenceCategory.addPreference(new ZenRulePreference(mContext,
new AbstractMap.SimpleEntry<>(FAKE_2, mock(AutomaticZenRule.class)),
null, null, mBackend));
assertNotNull(mPreferenceCategory.findPreference(FAKE_1));
assertNotNull(mPreferenceCategory.findPreference(FAKE_2));
String ruleId1 = "test1_id";
String ruleId2 = "test2_id";
@@ -131,83 +192,37 @@ public class ZenModeAutomaticRulesPreferenceControllerTest {
rMap.put(ruleId1, autoRule1);
rMap.put(ruleId2, autoRule2);
// Add three preferences to the set of previously-known-about ZenRulePreferences; in this
// case, test3_id is "deleted"
mController.mZenRulePreferences.put("test1_id", mZenRulePreference);
mController.mZenRulePreferences.put("test2_id", mZenRulePreference);
mController.mZenRulePreferences.put("test3_id", mZenRulePreference);
// update state should re-add all preferences since a preference was deleted
when(mockPref.getPreferenceCount()).thenReturn(NUM_RULES + 1);
mockGetAutomaticZenRules(NUM_RULES, rMap);
mController.updateState(mockPref);
verify(mockPref, times(1)).removeAll();
verify(mockPref, times(NUM_RULES)).addPreference(any());
assertEquals(NUM_RULES, mController.mZenRulePreferences.size());
mController.updateState(mPreferenceCategory);
assertNull(mPreferenceCategory.findPreference(FAKE_1));
assertNull(mPreferenceCategory.findPreference(FAKE_2));
assertNotNull(mPreferenceCategory.findPreference(ruleId1));
assertNotNull(mPreferenceCategory.findPreference(ruleId2));
}
@Test
public void testUpdateState_clearsPreferencesWhenSameNumberButDifferentPrefs() {
final int NUM_RULES = 2;
Map<String, AutomaticZenRule> rMap = new HashMap<>();
String ruleId1 = "test1_id";
String ruleId2 = "test2_id";
AutomaticZenRule autoRule1 = new AutomaticZenRule("test_rule_1", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10);
AutomaticZenRule autoRule2 = new AutomaticZenRule("test_rule_2", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 20);
rMap.put(ruleId1, autoRule1);
rMap.put(ruleId2, autoRule2);
// Add two preferences to the set of previously-known-about ZenRulePreferences; in this
// case, test3_id is "deleted" but test2_id is "added"
mController.mZenRulePreferences.put("test1_id", mZenRulePreference);
mController.mZenRulePreferences.put("test3_id", mZenRulePreference);
// update state should re-add all preferences since a preference was deleted
when(mockPref.getPreferenceCount()).thenReturn(NUM_RULES);
mockGetAutomaticZenRules(NUM_RULES, rMap);
mController.updateState(mockPref);
verify(mockPref, times(1)).removeAll();
verify(mockPref, times(NUM_RULES)).addPreference(any());
assertEquals(NUM_RULES, mController.mZenRulePreferences.size());
}
@Test
public void testUpdateState_updateEnableState() throws NoSuchFieldException {
final int NUM_RULES = 1;
Map<String, AutomaticZenRule> rMap = new HashMap<>();
public void testUpdateState_updateEnableState() {
mController.displayPreference(mPreferenceScreen);
String testId = "test1_id";
AutomaticZenRule rule = new AutomaticZenRule("rule_name", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10);
rMap.put(testId, rule);
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10);
when(mockPref.getPreferenceCount()).thenReturn(NUM_RULES);
when(mockPref.getPreference(anyInt())).thenReturn(mZenRulePreference);
mController.mZenRulePreferences.put("test1_id", mZenRulePreference);
mPreferenceCategory.addPreference(new ZenRulePreference(mContext,
new AbstractMap.SimpleEntry<>(testId, rule),
null, null, mBackend));
// update state should NOT re-add all the preferences, should only update enable state
rule.setEnabled(false);
rMap.put(testId, rule);
final int NUM_RULES = 1;
Map<String, AutomaticZenRule> rMap = new HashMap<>();
AutomaticZenRule ruleUpdated = new AutomaticZenRule("rule_name", null, null,
null, null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10);
ruleUpdated.setEnabled(false);
rMap.put(testId, ruleUpdated);
mockGetAutomaticZenRules(NUM_RULES, rMap);
setZenRulePreferenceField("mId", testId);
mController.updateState(mockPref);
verify(mZenRulePreference, times(1)).updatePreference(any());
verify(mockPref, never()).removeAll();
assertEquals(NUM_RULES, mController.mZenRulePreferences.size());
}
private void setZenRulePreferenceField(String name, Object value) {
try {
Field field = ZenRulePreference.class.getDeclaredField("mId");
field.setAccessible(true);
field.set(mZenRulePreference, value);
} catch (ReflectiveOperationException e) {
fail("Unable to set mZenRulePreference field: " + name);
}
mController.updateState(mPreferenceCategory);
assertFalse(mPreferenceCategory.findPreference(testId).isEnabled());
assertEquals(NUM_RULES, mPreferenceCategory.getPreferenceCount());
}
private void mockGetAutomaticZenRules(int numRules, Map<String, AutomaticZenRule> rules) {