1 /*
2  * Copyright (C) 2021 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 
17 package com.android.settings.safetycenter;
18 
19 import static com.google.common.truth.Truth.assertThat;
20 
21 import static org.mockito.ArgumentMatchers.any;
22 import static org.mockito.Mockito.doNothing;
23 import static org.mockito.Mockito.doReturn;
24 import static org.mockito.Mockito.doThrow;
25 import static org.mockito.Mockito.spy;
26 import static org.mockito.Mockito.verify;
27 import static org.mockito.Mockito.when;
28 
29 import android.content.ActivityNotFoundException;
30 import android.content.Context;
31 import android.content.Intent;
32 import android.content.pm.PackageManager;
33 
34 import androidx.preference.Preference;
35 import androidx.test.core.app.ApplicationProvider;
36 import androidx.test.ext.junit.runners.AndroidJUnit4;
37 
38 import org.junit.After;
39 import org.junit.Before;
40 import org.junit.Test;
41 import org.junit.runner.RunWith;
42 import org.mockito.ArgumentCaptor;
43 import org.mockito.Mock;
44 import org.mockito.MockitoAnnotations;
45 
46 @RunWith(AndroidJUnit4.class)
47 public class TopLevelSafetyCenterEntryPreferenceControllerTest {
48 
49     private static final String PREFERENCE_KEY = "top_level_safety_center";
50 
51     private TopLevelSafetyCenterEntryPreferenceController
52             mTopLevelSafetyCenterEntryPreferenceController;
53     private Preference mPreference;
54 
55     @Mock
56     private SafetyCenterManagerWrapper mSafetyCenterManagerWrapper;
57 
58     private Context mContext;
59 
60     @Before
setUp()61     public void setUp() {
62         MockitoAnnotations.initMocks(this);
63         SafetyCenterManagerWrapper.sInstance = mSafetyCenterManagerWrapper;
64 
65         mContext = spy(ApplicationProvider.getApplicationContext());
66         PackageManager pm = spy(mContext.getPackageManager());
67         doReturn(pm).when(mContext).getPackageManager();
68         doReturn("com.android.permissioncontroller").when(pm).getPermissionControllerPackageName();
69 
70         mPreference = new Preference(mContext);
71         mPreference.setKey(PREFERENCE_KEY);
72 
73         doNothing().when(mContext).startActivity(any(Intent.class));
74         mTopLevelSafetyCenterEntryPreferenceController =
75                 new TopLevelSafetyCenterEntryPreferenceController(mContext, PREFERENCE_KEY);
76     }
77 
78     @After
tearDown()79     public void tearDown() {
80         SafetyCenterManagerWrapper.sInstance = null;
81     }
82 
83     @Test
handlePreferenceTreeClick_forDifferentPreferenceKey_isNotHandled()84     public void handlePreferenceTreeClick_forDifferentPreferenceKey_isNotHandled() {
85         Preference preference = new Preference(mContext);
86         preference.setKey("some_other_preference");
87 
88         boolean preferenceHandled =
89                 mTopLevelSafetyCenterEntryPreferenceController.handlePreferenceTreeClick(
90                         preference);
91 
92         assertThat(preferenceHandled).isFalse();
93     }
94 
95     @Test
handlePreferenceTreeClick_launchesIntendedIntent()96     public void handlePreferenceTreeClick_launchesIntendedIntent() {
97         final ArgumentCaptor<Intent> intentCaptor = ArgumentCaptor.forClass(Intent.class);
98 
99         boolean preferenceHandled = mTopLevelSafetyCenterEntryPreferenceController
100                 .handlePreferenceTreeClick(mPreference);
101 
102         assertThat(preferenceHandled).isTrue();
103         verify(mContext).startActivity(intentCaptor.capture());
104         assertThat(intentCaptor.getValue().getAction()).isEqualTo(Intent.ACTION_SAFETY_CENTER);
105     }
106 
107     @Test
handlePreferenceTreeClick_onStartActivityThrows_returnsFalse()108     public void handlePreferenceTreeClick_onStartActivityThrows_returnsFalse() {
109         doThrow(ActivityNotFoundException.class)
110                 .when(mContext).startActivity(any(Intent.class));
111 
112         boolean preferenceHandled = mTopLevelSafetyCenterEntryPreferenceController
113                 .handlePreferenceTreeClick(mPreference);
114 
115         assertThat(preferenceHandled).isFalse();
116     }
117 
118     @Test
getAvailabilityStatus_whenSafetyCenterDisabled_returnsUnavailable()119     public void getAvailabilityStatus_whenSafetyCenterDisabled_returnsUnavailable() {
120         when(mSafetyCenterManagerWrapper.isEnabled(any(Context.class))).thenReturn(false);
121 
122         assertThat(mTopLevelSafetyCenterEntryPreferenceController.getAvailabilityStatus())
123                 .isEqualTo(TopLevelSafetyCenterEntryPreferenceController.CONDITIONALLY_UNAVAILABLE);
124     }
125 
126     @Test
getAvailabilityStatus_whenSafetyCenterEnabled_returnsAvailable()127     public void getAvailabilityStatus_whenSafetyCenterEnabled_returnsAvailable() {
128         when(mSafetyCenterManagerWrapper.isEnabled(any(Context.class))).thenReturn(true);
129 
130         assertThat(mTopLevelSafetyCenterEntryPreferenceController.getAvailabilityStatus())
131                 .isEqualTo(TopLevelSafetyCenterEntryPreferenceController.AVAILABLE);
132     }
133 }
134