1 /* 2 * Copyright (C) 2017 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 package com.android.emergency.util; 17 18 import static com.google.common.truth.Truth.assertThat; 19 import static org.mockito.Matchers.eq; 20 import static org.mockito.Mockito.any; 21 import static org.mockito.Mockito.anyInt; 22 import static org.mockito.Mockito.anyString; 23 import static org.mockito.Mockito.verify; 24 import static org.mockito.Mockito.when; 25 26 import android.content.ComponentName; 27 import android.content.ContentResolver; 28 import android.content.Context; 29 import android.content.SharedPreferences; 30 import android.content.pm.PackageManager; 31 import android.database.Cursor; 32 import android.net.Uri; 33 34 import com.android.emergency.ContactTestUtils; 35 import com.android.emergency.PreferenceKeys; 36 import com.android.emergency.TestConfig; 37 38 import org.junit.Before; 39 import org.junit.Test; 40 import org.junit.runner.RunWith; 41 import org.mockito.Mock; 42 import org.mockito.MockitoAnnotations; 43 import org.robolectric.RobolectricTestRunner; 44 import org.robolectric.RuntimeEnvironment; 45 import org.robolectric.annotation.Config; 46 47 /** Unit tests for {@link PreferenceUtils}. */ 48 @RunWith(RobolectricTestRunner.class) 49 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION) 50 public final class PreferenceUtilsTest { 51 private static final String PACKAGE_NAME = "com.android.emergency"; 52 private static final String NAME = "Jane"; 53 private static final String PHONE_NUMBER = "5150"; 54 55 @Mock ContentResolver mContentResolver; 56 @Mock Context mContext; 57 @Mock Cursor mCursor; 58 @Mock PackageManager mPackageManager; 59 @Mock SharedPreferences mSharedPreferences; 60 @Mock SharedPreferences.Editor mSharedPreferencesEditor; 61 62 private ComponentName mComponentName; 63 64 @Before setUp()65 public void setUp() { 66 MockitoAnnotations.initMocks(this); 67 68 mComponentName = new ComponentName( 69 PACKAGE_NAME, 70 PACKAGE_NAME + PreferenceUtils.SETTINGS_SUGGESTION_ACTIVITY_ALIAS); 71 72 when(mContext.getContentResolver()).thenReturn(mContentResolver); 73 when(mContext.getPackageManager()).thenReturn(mPackageManager); 74 when(mContext.getPackageName()).thenReturn(PACKAGE_NAME); 75 when(mContext.getSharedPreferences(anyString(), anyInt())).thenReturn(mSharedPreferences); 76 when(mSharedPreferences.edit()).thenReturn(mSharedPreferencesEditor); 77 when(mSharedPreferencesEditor.putString(anyString(), anyString())) 78 .thenReturn(mSharedPreferencesEditor); 79 } 80 81 @Test testHasAtLeastOnePreferenceSet_notSet()82 public void testHasAtLeastOnePreferenceSet_notSet() { 83 when(mSharedPreferences.getString(anyString(), anyString())).thenReturn(""); 84 85 assertThat(PreferenceUtils.hasAtLeastOnePreferenceSet(mContext)).isFalse(); 86 } 87 88 @Test testHasAtLeastOnePreferenceSet_set()89 public void testHasAtLeastOnePreferenceSet_set() { 90 when(mSharedPreferences.getString(anyString(), anyString())).thenReturn("mxyzptlk"); 91 92 assertThat(PreferenceUtils.hasAtLeastOnePreferenceSet(mContext)).isTrue(); 93 } 94 95 @Test testHasAtLeastOneEmergencyContact_notSet()96 public void testHasAtLeastOneEmergencyContact_notSet() { 97 when(mSharedPreferences.getString(eq(PreferenceKeys.KEY_EMERGENCY_CONTACTS), any())) 98 .thenReturn(""); 99 100 assertThat(PreferenceUtils.hasAtLeastOneEmergencyContact(mContext)).isFalse(); 101 } 102 103 @Test testHasAtLeastOneEmergencyContact_set()104 public void testHasAtLeastOneEmergencyContact_set() { 105 final Uri contactUri = ContactTestUtils.createContact( 106 RuntimeEnvironment.application.getContentResolver(), NAME, PHONE_NUMBER); 107 when(mSharedPreferences.getString(eq(PreferenceKeys.KEY_EMERGENCY_CONTACTS), any())) 108 .thenReturn(contactUri.toString()); 109 when(mContentResolver.query(eq(contactUri), any(), any(), any(), any())) 110 .thenReturn(mCursor); 111 when(mCursor.moveToFirst()).thenReturn(true); 112 113 assertThat(PreferenceUtils.hasAtLeastOneEmergencyContact(mContext)).isTrue(); 114 } 115 116 @Test testEnableSettingsSuggestion()117 public void testEnableSettingsSuggestion() { 118 PreferenceUtils.enableSettingsSuggestion(mContext); 119 120 verify(mPackageManager).setComponentEnabledSetting( 121 eq(mComponentName), 122 eq(PackageManager.COMPONENT_ENABLED_STATE_ENABLED), 123 eq(PackageManager.DONT_KILL_APP)); 124 } 125 126 @Test testUpdateSettingsSuggestionState_contactSet()127 public void testUpdateSettingsSuggestionState_contactSet() { 128 final Uri contactUri = ContactTestUtils.createContact( 129 RuntimeEnvironment.application.getContentResolver(), NAME, PHONE_NUMBER); 130 when(mSharedPreferences.getString(eq(PreferenceKeys.KEY_EMERGENCY_CONTACTS), any())) 131 .thenReturn(contactUri.toString()); 132 when(mContentResolver.query(eq(contactUri), any(), any(), any(), any())) 133 .thenReturn(mCursor); 134 when(mCursor.moveToFirst()).thenReturn(true); 135 136 PreferenceUtils.updateSettingsSuggestionState(mContext); 137 138 verify(mPackageManager).setComponentEnabledSetting( 139 eq(mComponentName), 140 eq(PackageManager.COMPONENT_ENABLED_STATE_DISABLED), 141 eq(PackageManager.DONT_KILL_APP)); 142 } 143 144 @Test testUpdateSettingsSuggestionState_noPreferencesOrContactSet()145 public void testUpdateSettingsSuggestionState_noPreferencesOrContactSet() { 146 when(mSharedPreferences.getString(anyString(), anyString())).thenReturn(""); 147 148 PreferenceUtils.updateSettingsSuggestionState(mContext); 149 150 verify(mPackageManager).setComponentEnabledSetting( 151 eq(mComponentName), 152 eq(PackageManager.COMPONENT_ENABLED_STATE_ENABLED), 153 eq(PackageManager.DONT_KILL_APP)); 154 } 155 156 @Test testUpdateSettingsSuggestionState_preferenceSet()157 public void testUpdateSettingsSuggestionState_preferenceSet() { 158 when(mSharedPreferences.getString(eq(PreferenceKeys.KEY_ALLERGIES), any())) 159 .thenReturn("peanuts"); 160 161 PreferenceUtils.updateSettingsSuggestionState(mContext); 162 163 verify(mPackageManager).setComponentEnabledSetting( 164 eq(mComponentName), 165 eq(PackageManager.COMPONENT_ENABLED_STATE_DISABLED), 166 eq(PackageManager.DONT_KILL_APP)); 167 } 168 } 169