1323522171d5b4ac0d95edc105aaf849661763c84Beverly/* 2323522171d5b4ac0d95edc105aaf849661763c84Beverly * Copyright (C) 2017 The Android Open Source Project 3323522171d5b4ac0d95edc105aaf849661763c84Beverly * 4323522171d5b4ac0d95edc105aaf849661763c84Beverly * Licensed under the Apache License, Version 2.0 (the "License"); 5323522171d5b4ac0d95edc105aaf849661763c84Beverly * you may not use this file except in compliance with the License. 6323522171d5b4ac0d95edc105aaf849661763c84Beverly * You may obtain a copy of the License at 7323522171d5b4ac0d95edc105aaf849661763c84Beverly * 8323522171d5b4ac0d95edc105aaf849661763c84Beverly * http://www.apache.org/licenses/LICENSE-2.0 9323522171d5b4ac0d95edc105aaf849661763c84Beverly * 10323522171d5b4ac0d95edc105aaf849661763c84Beverly * Unless required by applicable law or agreed to in writing, software 11323522171d5b4ac0d95edc105aaf849661763c84Beverly * distributed under the License is distributed on an "AS IS" BASIS, 12323522171d5b4ac0d95edc105aaf849661763c84Beverly * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13323522171d5b4ac0d95edc105aaf849661763c84Beverly * See the License for the specific language governing permissions and 14323522171d5b4ac0d95edc105aaf849661763c84Beverly * limitations under the License. 15323522171d5b4ac0d95edc105aaf849661763c84Beverly */ 16323522171d5b4ac0d95edc105aaf849661763c84Beverly 17323522171d5b4ac0d95edc105aaf849661763c84Beverlypackage com.android.settings.notification; 18323522171d5b4ac0d95edc105aaf849661763c84Beverly 19323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport static junit.framework.Assert.assertEquals; 20323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport static org.mockito.Mockito.mock; 21323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport static org.mockito.Mockito.when; 22323522171d5b4ac0d95edc105aaf849661763c84Beverly 23323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.app.AutomaticZenRule; 24323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.app.Fragment; 25323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.app.NotificationManager; 26323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.content.Context; 27323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.provider.Settings; 28323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.support.v7.preference.PreferenceCategory; 29323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport android.support.v7.preference.PreferenceScreen; 30323522171d5b4ac0d95edc105aaf849661763c84Beverly 31323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport com.android.settings.testutils.SettingsRobolectricTestRunner; 32323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport com.android.settingslib.core.lifecycle.Lifecycle; 33323522171d5b4ac0d95edc105aaf849661763c84Beverly 34323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.junit.Before; 35323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.junit.Test; 36323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.junit.runner.RunWith; 37323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.mockito.Mock; 38323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.mockito.MockitoAnnotations; 39323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.robolectric.RuntimeEnvironment; 40323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.robolectric.shadows.ShadowApplication; 41323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport org.robolectric.util.ReflectionHelpers; 42323522171d5b4ac0d95edc105aaf849661763c84Beverly 432c5895d615d4ed8f379965656f62ae5310a9167eBeverlyimport java.util.Arrays; 44323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport java.util.HashMap; 452c5895d615d4ed8f379965656f62ae5310a9167eBeverlyimport java.util.List; 46323522171d5b4ac0d95edc105aaf849661763c84Beverlyimport java.util.Map; 47323522171d5b4ac0d95edc105aaf849661763c84Beverly 48323522171d5b4ac0d95edc105aaf849661763c84Beverly@RunWith(SettingsRobolectricTestRunner.class) 49323522171d5b4ac0d95edc105aaf849661763c84Beverlypublic class ZenModeAutomaticRulesPreferenceControllerTest { 5022a39c2b93bc66db71238274a7683d329232d124James Lemieux 5122a39c2b93bc66db71238274a7683d329232d124James Lemieux private static final String GENERIC_RULE_NAME = "test"; 5222a39c2b93bc66db71238274a7683d329232d124James Lemieux private static final String DEFAULT_ID_1 = "DEFAULT_1"; 5322a39c2b93bc66db71238274a7683d329232d124James Lemieux private static final String DEFAULT_ID_2 = "DEFAULT_2"; 5422a39c2b93bc66db71238274a7683d329232d124James Lemieux 55323522171d5b4ac0d95edc105aaf849661763c84Beverly private ZenModeAutomaticRulesPreferenceController mController; 562c5895d615d4ed8f379965656f62ae5310a9167eBeverly private final List<String> mDefaultIds = Arrays.asList(DEFAULT_ID_1, DEFAULT_ID_2); 57323522171d5b4ac0d95edc105aaf849661763c84Beverly 58323522171d5b4ac0d95edc105aaf849661763c84Beverly @Mock 59323522171d5b4ac0d95edc105aaf849661763c84Beverly private ZenModeBackend mBackend; 60323522171d5b4ac0d95edc105aaf849661763c84Beverly @Mock 61323522171d5b4ac0d95edc105aaf849661763c84Beverly private NotificationManager mNotificationManager; 62323522171d5b4ac0d95edc105aaf849661763c84Beverly @Mock 63323522171d5b4ac0d95edc105aaf849661763c84Beverly private PreferenceCategory mockPref; 64323522171d5b4ac0d95edc105aaf849661763c84Beverly @Mock 65323522171d5b4ac0d95edc105aaf849661763c84Beverly private NotificationManager.Policy mPolicy; 66323522171d5b4ac0d95edc105aaf849661763c84Beverly @Mock 67323522171d5b4ac0d95edc105aaf849661763c84Beverly private PreferenceScreen mPreferenceScreen; 68323522171d5b4ac0d95edc105aaf849661763c84Beverly 69323522171d5b4ac0d95edc105aaf849661763c84Beverly private Context mContext; 70323522171d5b4ac0d95edc105aaf849661763c84Beverly 71323522171d5b4ac0d95edc105aaf849661763c84Beverly @Before 72323522171d5b4ac0d95edc105aaf849661763c84Beverly public void setup() { 73323522171d5b4ac0d95edc105aaf849661763c84Beverly MockitoAnnotations.initMocks(this); 74323522171d5b4ac0d95edc105aaf849661763c84Beverly ShadowApplication shadowApplication = ShadowApplication.getInstance(); 75323522171d5b4ac0d95edc105aaf849661763c84Beverly shadowApplication.setSystemService(Context.NOTIFICATION_SERVICE, mNotificationManager); 76323522171d5b4ac0d95edc105aaf849661763c84Beverly 7722a39c2b93bc66db71238274a7683d329232d124James Lemieux mContext = RuntimeEnvironment.application; 78323522171d5b4ac0d95edc105aaf849661763c84Beverly when(mNotificationManager.getNotificationPolicy()).thenReturn(mPolicy); 79323522171d5b4ac0d95edc105aaf849661763c84Beverly mController = new ZenModeAutomaticRulesPreferenceController(mContext, mock(Fragment.class), 80323522171d5b4ac0d95edc105aaf849661763c84Beverly mock(Lifecycle.class)); 812c5895d615d4ed8f379965656f62ae5310a9167eBeverly 82323522171d5b4ac0d95edc105aaf849661763c84Beverly ReflectionHelpers.setField(mController, "mBackend", mBackend); 83b000ace0eb50c32f290cf31f14a52c3f3c3906e0Jaekyun Seok ReflectionHelpers.setField(mController, "mDefaultRuleIds", mDefaultIds); 84323522171d5b4ac0d95edc105aaf849661763c84Beverly 85323522171d5b4ac0d95edc105aaf849661763c84Beverly when(mPreferenceScreen.findPreference(mController.getPreferenceKey())).thenReturn( 86323522171d5b4ac0d95edc105aaf849661763c84Beverly mockPref); 87323522171d5b4ac0d95edc105aaf849661763c84Beverly mController.displayPreference(mPreferenceScreen); 88323522171d5b4ac0d95edc105aaf849661763c84Beverly } 89323522171d5b4ac0d95edc105aaf849661763c84Beverly 90323522171d5b4ac0d95edc105aaf849661763c84Beverly @Test 91323522171d5b4ac0d95edc105aaf849661763c84Beverly public void updateState_checkRuleOrderingDescending() { 92323522171d5b4ac0d95edc105aaf849661763c84Beverly final int NUM_RULES = 4; 93323522171d5b4ac0d95edc105aaf849661763c84Beverly when(mNotificationManager.getAutomaticZenRules()).thenReturn( 94323522171d5b4ac0d95edc105aaf849661763c84Beverly mockAutoZenRulesDecreasingCreationTime(NUM_RULES)); 95323522171d5b4ac0d95edc105aaf849661763c84Beverly 96323522171d5b4ac0d95edc105aaf849661763c84Beverly Map.Entry<String, AutomaticZenRule>[] rules = mController.sortedRules(); 97323522171d5b4ac0d95edc105aaf849661763c84Beverly assertEquals(NUM_RULES, rules.length); 98323522171d5b4ac0d95edc105aaf849661763c84Beverly 99323522171d5b4ac0d95edc105aaf849661763c84Beverly // check ordering, most recent should be at the bottom/end (ie higher creation time) 100323522171d5b4ac0d95edc105aaf849661763c84Beverly for (int i = 0; i < NUM_RULES; i++) { 1012c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(GENERIC_RULE_NAME + (NUM_RULES - 1 - i), rules[i].getKey()); 102323522171d5b4ac0d95edc105aaf849661763c84Beverly } 103323522171d5b4ac0d95edc105aaf849661763c84Beverly } 104323522171d5b4ac0d95edc105aaf849661763c84Beverly 105323522171d5b4ac0d95edc105aaf849661763c84Beverly @Test 106323522171d5b4ac0d95edc105aaf849661763c84Beverly public void updateState_checkRuleOrderingAscending() { 107323522171d5b4ac0d95edc105aaf849661763c84Beverly final int NUM_RULES = 4; 108323522171d5b4ac0d95edc105aaf849661763c84Beverly when(mNotificationManager.getAutomaticZenRules()).thenReturn( 109323522171d5b4ac0d95edc105aaf849661763c84Beverly mockAutoZenRulesAscendingCreationTime(NUM_RULES)); 110323522171d5b4ac0d95edc105aaf849661763c84Beverly 111323522171d5b4ac0d95edc105aaf849661763c84Beverly Map.Entry<String, AutomaticZenRule>[] rules = mController.sortedRules(); 112323522171d5b4ac0d95edc105aaf849661763c84Beverly assertEquals(NUM_RULES, rules.length); 113323522171d5b4ac0d95edc105aaf849661763c84Beverly 114323522171d5b4ac0d95edc105aaf849661763c84Beverly // check ordering, most recent should be at the bottom/end (ie higher creation time) 115323522171d5b4ac0d95edc105aaf849661763c84Beverly for (int i = 0; i < NUM_RULES; i++) { 1162c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(GENERIC_RULE_NAME + i, rules[i].getKey()); 1172c5895d615d4ed8f379965656f62ae5310a9167eBeverly } 1182c5895d615d4ed8f379965656f62ae5310a9167eBeverly } 1192c5895d615d4ed8f379965656f62ae5310a9167eBeverly 1202c5895d615d4ed8f379965656f62ae5310a9167eBeverly @Test 1212c5895d615d4ed8f379965656f62ae5310a9167eBeverly public void updateState_checkRuleOrderingDescending_withDefaultRules() { 1222c5895d615d4ed8f379965656f62ae5310a9167eBeverly final int NUM_RULES = 4; 1232c5895d615d4ed8f379965656f62ae5310a9167eBeverly 1242c5895d615d4ed8f379965656f62ae5310a9167eBeverly Map<String, AutomaticZenRule> ruleMap = mockAutoZenRulesDecreasingCreationTime(NUM_RULES); 1252c5895d615d4ed8f379965656f62ae5310a9167eBeverly ruleMap.put(DEFAULT_ID_2, new AutomaticZenRule("DEFAULT_1_NAME", null, 1262c5895d615d4ed8f379965656f62ae5310a9167eBeverly null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 20)); 1272c5895d615d4ed8f379965656f62ae5310a9167eBeverly ruleMap.put(DEFAULT_ID_1, new AutomaticZenRule("DEFAULT_1_NAME", null, 1282c5895d615d4ed8f379965656f62ae5310a9167eBeverly null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 10)); 1292c5895d615d4ed8f379965656f62ae5310a9167eBeverly when(mNotificationManager.getAutomaticZenRules()).thenReturn(ruleMap); 1302c5895d615d4ed8f379965656f62ae5310a9167eBeverly 1312c5895d615d4ed8f379965656f62ae5310a9167eBeverly Map.Entry<String, AutomaticZenRule>[] rules = mController.sortedRules(); 1322c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(NUM_RULES + 2, rules.length); 1332c5895d615d4ed8f379965656f62ae5310a9167eBeverly 1342c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(rules[0].getKey(), DEFAULT_ID_1); 1352c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(rules[1].getKey(), DEFAULT_ID_2); 1362c5895d615d4ed8f379965656f62ae5310a9167eBeverly // NON-DEFAULT RULES check ordering, most recent at the bottom/end 1372c5895d615d4ed8f379965656f62ae5310a9167eBeverly for (int i = 0; i < NUM_RULES; i++) { 1382c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(GENERIC_RULE_NAME + (NUM_RULES - 1 - i), rules[i + 2].getKey()); 139323522171d5b4ac0d95edc105aaf849661763c84Beverly } 140323522171d5b4ac0d95edc105aaf849661763c84Beverly } 141323522171d5b4ac0d95edc105aaf849661763c84Beverly 142323522171d5b4ac0d95edc105aaf849661763c84Beverly @Test 143323522171d5b4ac0d95edc105aaf849661763c84Beverly public void updateState_checkRuleOrderingMix() { 144323522171d5b4ac0d95edc105aaf849661763c84Beverly final int NUM_RULES = 4; 145323522171d5b4ac0d95edc105aaf849661763c84Beverly // map with creation times: 0, 2, 4, 6 146323522171d5b4ac0d95edc105aaf849661763c84Beverly Map<String,AutomaticZenRule> rMap = mockAutoZenRulesAscendingCreationTime(NUM_RULES); 147323522171d5b4ac0d95edc105aaf849661763c84Beverly 148323522171d5b4ac0d95edc105aaf849661763c84Beverly final String insertedRule1 = "insertedRule1"; 149323522171d5b4ac0d95edc105aaf849661763c84Beverly rMap.put(insertedRule1, new AutomaticZenRule(insertedRule1, null, null, 150323522171d5b4ac0d95edc105aaf849661763c84Beverly Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 5)); 151323522171d5b4ac0d95edc105aaf849661763c84Beverly 152323522171d5b4ac0d95edc105aaf849661763c84Beverly final String insertedRule2 = "insertedRule2"; 153323522171d5b4ac0d95edc105aaf849661763c84Beverly rMap.put(insertedRule2, new AutomaticZenRule(insertedRule2, null, null, 154323522171d5b4ac0d95edc105aaf849661763c84Beverly Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, 3)); 155323522171d5b4ac0d95edc105aaf849661763c84Beverly 156323522171d5b4ac0d95edc105aaf849661763c84Beverly // rule map with rule creation times, 0, 2, 4, 6, 5, 3 157323522171d5b4ac0d95edc105aaf849661763c84Beverly // sort should create ordering based on creation times: 0, 2, 3, 4, 5, 6 158323522171d5b4ac0d95edc105aaf849661763c84Beverly when(mNotificationManager.getAutomaticZenRules()).thenReturn(rMap); 159323522171d5b4ac0d95edc105aaf849661763c84Beverly 160323522171d5b4ac0d95edc105aaf849661763c84Beverly Map.Entry<String, AutomaticZenRule>[] rules = mController.sortedRules(); 161323522171d5b4ac0d95edc105aaf849661763c84Beverly assertEquals(NUM_RULES + 2, rules.length); // inserted 2 rules 162323522171d5b4ac0d95edc105aaf849661763c84Beverly 163323522171d5b4ac0d95edc105aaf849661763c84Beverly // check ordering of inserted rules 1642c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(insertedRule1, rules[4].getKey()); 1652c5895d615d4ed8f379965656f62ae5310a9167eBeverly assertEquals(insertedRule2, rules[2].getKey()); 166323522171d5b4ac0d95edc105aaf849661763c84Beverly } 167323522171d5b4ac0d95edc105aaf849661763c84Beverly 168323522171d5b4ac0d95edc105aaf849661763c84Beverly private Map<String, AutomaticZenRule> mockAutoZenRulesAscendingCreationTime(int numRules) { 169323522171d5b4ac0d95edc105aaf849661763c84Beverly Map<String, AutomaticZenRule> ruleMap = new HashMap<>(); 170323522171d5b4ac0d95edc105aaf849661763c84Beverly 171323522171d5b4ac0d95edc105aaf849661763c84Beverly for (int i = 0; i < numRules; i++) { 172323522171d5b4ac0d95edc105aaf849661763c84Beverly ruleMap.put(GENERIC_RULE_NAME + i, new AutomaticZenRule(GENERIC_RULE_NAME + i, null, 173323522171d5b4ac0d95edc105aaf849661763c84Beverly null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, i * 2)); 174323522171d5b4ac0d95edc105aaf849661763c84Beverly } 175323522171d5b4ac0d95edc105aaf849661763c84Beverly 176323522171d5b4ac0d95edc105aaf849661763c84Beverly return ruleMap; 177323522171d5b4ac0d95edc105aaf849661763c84Beverly } 178323522171d5b4ac0d95edc105aaf849661763c84Beverly 179323522171d5b4ac0d95edc105aaf849661763c84Beverly private Map<String, AutomaticZenRule> mockAutoZenRulesDecreasingCreationTime(int numRules) { 180323522171d5b4ac0d95edc105aaf849661763c84Beverly Map<String, AutomaticZenRule> ruleMap = new HashMap<>(); 181323522171d5b4ac0d95edc105aaf849661763c84Beverly 182323522171d5b4ac0d95edc105aaf849661763c84Beverly for (int i = 0; i < numRules; i++) { 183323522171d5b4ac0d95edc105aaf849661763c84Beverly ruleMap.put(GENERIC_RULE_NAME + i, new AutomaticZenRule(GENERIC_RULE_NAME + i, null, 184323522171d5b4ac0d95edc105aaf849661763c84Beverly null, Settings.Global.ZEN_MODE_IMPORTANT_INTERRUPTIONS, true, numRules - i)); 185323522171d5b4ac0d95edc105aaf849661763c84Beverly } 186323522171d5b4ac0d95edc105aaf849661763c84Beverly 187323522171d5b4ac0d95edc105aaf849661763c84Beverly return ruleMap; 188323522171d5b4ac0d95edc105aaf849661763c84Beverly } 189b000ace0eb50c32f290cf31f14a52c3f3c3906e0Jaekyun Seok} 190