1/*
2 * Copyright (C) 2016 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
17package com.android.settings.enterprise;
18
19import static com.google.common.truth.Truth.assertThat;
20import static org.mockito.Mockito.when;
21
22import android.content.Context;
23
24import com.android.internal.logging.nano.MetricsProto.MetricsEvent;
25import com.android.settings.R;
26import com.android.settings.testutils.FakeFeatureFactory;
27import com.android.settings.testutils.SettingsRobolectricTestRunner;
28import com.android.settings.widget.PreferenceCategoryController;
29import com.android.settingslib.core.AbstractPreferenceController;
30
31import org.junit.Before;
32import org.junit.Test;
33import org.junit.runner.RunWith;
34import org.mockito.Answers;
35import org.mockito.Mock;
36import org.mockito.MockitoAnnotations;
37import org.robolectric.RuntimeEnvironment;
38
39import java.util.List;
40
41@RunWith(SettingsRobolectricTestRunner.class)
42public class EnterprisePrivacySettingsTest {
43
44    @Mock(answer = Answers.RETURNS_DEEP_STUBS)
45    private Context mContext;
46    private FakeFeatureFactory mFeatureFactory;
47    private EnterprisePrivacySettings mSettings;
48
49    @Before
50    public void setUp() {
51        MockitoAnnotations.initMocks(this);
52        mFeatureFactory = FakeFeatureFactory.setupForTest();
53        mSettings = new EnterprisePrivacySettings();
54    }
55
56    @Test
57    public void verifyConstants() {
58        assertThat(mSettings.getMetricsCategory())
59                .isEqualTo(MetricsEvent.ENTERPRISE_PRIVACY_SETTINGS);
60        assertThat(mSettings.getLogTag()).isEqualTo("EnterprisePrivacySettings");
61        assertThat(mSettings.getCategoryKey()).isNull();
62        assertThat(mSettings.getPreferenceScreenResId())
63                .isEqualTo(R.xml.enterprise_privacy_settings);
64    }
65
66    @Test
67    public void isPageEnabled_hasDeviceOwner_shouldReturnTrue() {
68        when(mFeatureFactory.enterprisePrivacyFeatureProvider.hasDeviceOwner())
69                .thenReturn(true);
70
71        assertThat(EnterprisePrivacySettings.isPageEnabled(mContext))
72                .isTrue();
73    }
74
75    @Test
76    public void isPageEnabled_noDeviceOwner_shouldReturnFalse() {
77        when(mFeatureFactory.enterprisePrivacyFeatureProvider.hasDeviceOwner())
78                .thenReturn(false);
79
80        assertThat(EnterprisePrivacySettings.isPageEnabled(mContext))
81                .isFalse();
82    }
83
84    @Test
85    public void getPreferenceControllers() {
86        final List<AbstractPreferenceController> controllers =
87            mSettings.createPreferenceControllers(RuntimeEnvironment.application);
88        verifyPreferenceControllers(controllers);
89    }
90
91    @Test
92    public void getSearchIndexProviderPreferenceControllers() {
93        final List<AbstractPreferenceController> controllers =
94            EnterprisePrivacySettings.SEARCH_INDEX_DATA_PROVIDER
95                .getPreferenceControllers(RuntimeEnvironment.application);
96        verifyPreferenceControllers(controllers);
97    }
98
99    private void verifyPreferenceControllers(List<AbstractPreferenceController> controllers) {
100        assertThat(controllers).isNotNull();
101        assertThat(controllers.size()).isEqualTo(18);
102        int position = 0;
103        assertThat(controllers.get(position++)).isInstanceOf(NetworkLogsPreferenceController.class);
104        assertThat(controllers.get(position++)).isInstanceOf(BugReportsPreferenceController.class);
105        assertThat(controllers.get(position++)).isInstanceOf(
106                SecurityLogsPreferenceController.class);
107        assertThat(controllers.get(position++)).isInstanceOf(
108                EnterpriseInstalledPackagesPreferenceController.class);
109        assertThat(controllers.get(position++)).isInstanceOf(
110                AdminGrantedLocationPermissionsPreferenceController.class);
111        assertThat(controllers.get(position++)).isInstanceOf(
112                AdminGrantedMicrophonePermissionPreferenceController.class);
113        assertThat(controllers.get(position++)).isInstanceOf(
114                AdminGrantedCameraPermissionPreferenceController.class);
115        assertThat(controllers.get(position++)).isInstanceOf(
116                EnterpriseSetDefaultAppsPreferenceController.class);
117        assertThat(controllers.get(position++)).isInstanceOf(
118                AlwaysOnVpnCurrentUserPreferenceController.class);
119        assertThat(controllers.get(position++)).isInstanceOf(
120                AlwaysOnVpnManagedProfilePreferenceController.class);
121        assertThat(controllers.get(position++)).isInstanceOf(ImePreferenceController.class);
122        assertThat(controllers.get(position++)).isInstanceOf(
123                GlobalHttpProxyPreferenceController.class);
124        assertThat(controllers.get(position++)).isInstanceOf(
125                CaCertsCurrentUserPreferenceController.class);
126        assertThat(controllers.get(position++)).isInstanceOf(
127                CaCertsManagedProfilePreferenceController.class);
128        assertThat(controllers.get(position++)).isInstanceOf(
129                BackupsEnabledPreferenceController.class);
130        assertThat(controllers.get(position++)).isInstanceOf(
131                PreferenceCategoryController.class);
132        assertThat(controllers.get(position++)).isInstanceOf(
133                FailedPasswordWipeCurrentUserPreferenceController.class);
134        assertThat(controllers.get(position)).isInstanceOf(
135                FailedPasswordWipeManagedProfilePreferenceController.class);
136    }
137}
138