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.eq; 24 import static org.mockito.Mockito.verify; 25 import static org.mockito.Mockito.when; 26 27 import android.content.ComponentName; 28 import android.content.ContentResolver; 29 import android.content.Context; 30 import android.content.SharedPreferences; 31 import android.content.pm.PackageManager; 32 import android.database.Cursor; 33 import android.net.Uri; 34 import android.test.suitebuilder.annotation.SmallTest; 35 36 import com.android.emergency.ContactTestUtils; 37 import com.android.emergency.PreferenceKeys; 38 import com.android.emergency.TestConfig; 39 40 import org.junit.Before; 41 import org.junit.Test; 42 import org.junit.runner.RunWith; 43 import org.mockito.Mock; 44 import org.mockito.MockitoAnnotations; 45 import org.robolectric.RobolectricTestRunner; 46 import org.robolectric.RuntimeEnvironment; 47 import org.robolectric.annotation.Config; 48 49 /** Unit tests for {@link PreferenceUtils}. */ 50 @SmallTest 51 @RunWith(RobolectricTestRunner.class) 52 @Config(manifest = TestConfig.MANIFEST_PATH, sdk = TestConfig.SDK_VERSION) 53 public final class PreferenceUtilsTest { 54 private static final String PACKAGE_NAME = "com.android.emergency"; 55 private static final String NAME = "Jane"; 56 private static final String PHONE_NUMBER = "5150"; 57 private static final ComponentName COMPONENT_NAME = new ComponentName( 58 PACKAGE_NAME, 59 PACKAGE_NAME + PreferenceUtils.SETTINGS_SUGGESTION_ACTIVITY_ALIAS); 60 61 @Mock ContentResolver mContentResolver; 62 @Mock Context mContext; 63 @Mock Cursor mCursor; 64 @Mock PackageManager mPackageManager; 65 @Mock SharedPreferences mSharedPreferences; 66 @Mock SharedPreferences.Editor mSharedPreferencesEditor; 67 68 @Before 69 public void setUp() { 70 MockitoAnnotations.initMocks(this); 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 82 public void testHasAtLeastOnePreferenceSet_notSet() { 83 when(mSharedPreferences.getString(anyString(), anyString())).thenReturn(""); 84 85 assertThat(PreferenceUtils.hasAtLeastOnePreferenceSet(mContext)).isFalse(); 86 } 87 88 @Test 89 public void testHasAtLeastOnePreferenceSet_set() { 90 when(mSharedPreferences.getString(anyString(), anyString())).thenReturn("mxyzptlk"); 91 92 assertThat(PreferenceUtils.hasAtLeastOnePreferenceSet(mContext)).isTrue(); 93 } 94 95 @Test 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 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 117 public void testEnableSettingsSuggestion() { 118 PreferenceUtils.enableSettingsSuggestion(mContext); 119 120 verify(mPackageManager).setComponentEnabledSetting( 121 eq(COMPONENT_NAME), 122 eq(PackageManager.COMPONENT_ENABLED_STATE_ENABLED), 123 eq(PackageManager.DONT_KILL_APP)); 124 } 125 126 @Test 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(COMPONENT_NAME), 140 eq(PackageManager.COMPONENT_ENABLED_STATE_DISABLED), 141 eq(PackageManager.DONT_KILL_APP)); 142 } 143 144 @Test 145 public void testUpdateSettingsSuggestionState_noPreferencesOrContactSet() { 146 when(mSharedPreferences.getString(anyString(), anyString())).thenReturn(""); 147 148 PreferenceUtils.updateSettingsSuggestionState(mContext); 149 150 verify(mPackageManager).setComponentEnabledSetting( 151 eq(COMPONENT_NAME), 152 eq(PackageManager.COMPONENT_ENABLED_STATE_ENABLED), 153 eq(PackageManager.DONT_KILL_APP)); 154 } 155 156 @Test 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(COMPONENT_NAME), 165 eq(PackageManager.COMPONENT_ENABLED_STATE_DISABLED), 166 eq(PackageManager.DONT_KILL_APP)); 167 } 168 } 169